Skip to content

feat(policies): add a permutation-equivariant deep-set policy - #14

Open
ZacKienzle2 wants to merge 1 commit into
mainfrom
feat/deep-set-policy
Open

feat(policies): add a permutation-equivariant deep-set policy#14
ZacKienzle2 wants to merge 1 commit into
mainfrom
feat/deep-set-policy

Conversation

@ZacKienzle2

Copy link
Copy Markdown
Owner

Summary

A permutation-equivariant multi-asset hedging policy in the Deep Sets family (roadmap Horizon 4, policy frontier; pure machine-learning focus).

  • Shared per-asset encoder, an order-invariant mean-pooled book summary, and a per-asset head reading each position from its own latent plus the summary.
  • Equivariant under asset relabelling, and the parameter count is independent of the asset count, so one trained policy generalises across book sizes.
  • Pairs with MultiAssetFeatures. Exported from deephedging and deephedging.policies.

Tests

Permutation equivariance, asset-count-independent parameter count, output shape, input validation, and a training-integration test showing the policy learns a correlated two-asset hedge that beats not hedging by more than forty percent of tail risk.

Gates

ruff, ruff format, pyright (strict, 0 errors), fast suite green (185 passed), slow training test passes.

Hedge a multi-asset book with a shared per-asset encoder, an order-invariant mean-pooled summary, and a per-asset head, so the map is equivariant under asset relabelling and the parameter count is independent of the asset count. Pairs with MultiAssetFeatures. Tests pin the equivariance, the asset-count-independent parameter count, the output shape, input validation, and that the policy trains a correlated two-asset hedge that beats not hedging.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant