Implement Phase 2: secondary particle prediction #9

Merged
lbogner merged 21 commits from phase2-secondary-prediction into master 2026-07-17 11:04:22 +02:00
lbogner commented 2026-07-17 11:04:03 +02:00 (Migrated from gitlab.etp.kit.edu)

Two-stage factorisation: Stage 1 predicts 9D primary kinematics + n_sec
classification head (COND_DIM reduced to 8, dropping n_sec/e_sec inputs);
Stage 2 (SecondaryDecoder) generates K_MAX=15 secondary slots via masked
flow matching over (stick_logit, local_dir, type_emb) conditioned on Stage 1
output. Joint training with combined loss L_s1 + λ_nsecL_nsec + λ_s2L_s2.

Co-Authored-By: Claude Sonnet 4.6 noreply@anthropic.com

Two-stage factorisation: Stage 1 predicts 9D primary kinematics + n_sec classification head (COND_DIM reduced to 8, dropping n_sec/e_sec inputs); Stage 2 (SecondaryDecoder) generates K_MAX=15 secondary slots via masked flow matching over (stick_logit, local_dir, type_emb) conditioned on Stage 1 output. Joint training with combined loss L_s1 + λ_nsec*L_nsec + λ_s2*L_s2. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
lbogner commented 2026-07-17 11:04:23 +02:00 (Migrated from gitlab.etp.kit.edu)

mentioned in commit b658f4d40e

mentioned in commit b658f4d40ee0f8d159c9a01f9fba0b5729624406
lbogner (Migrated from gitlab.etp.kit.edu) merged commit b658f4d40e into master 2026-07-17 11:04:23 +02:00
Sign in to join this conversation.
No Reviewers
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: lars/giant#9