Flow-based surrogate models for particle tracking
To capture distribution-dependent dynamics that vanilla CFM cannot represent, the model is extended with cross-attention over the initial particle ensemble (Cross-Attention-CFM) and it is demonstrated that this extension recovers performance on a space-charge benchmark in the PS, where vanilla CFM degrades.