用扩散模型从稀疏投影重建粒子束4维相空间,速度快1万倍且物理正确。
PhaseFlow4D: Physically Constrained 4D Beam Reconstruction via Feedback-Guided Latent Diffusion

- 构建4D VAE+扩散模型,通过解析投影一致性保证物理正确性
- 在FRIB模拟中实现11000倍加速,准确追踪随时间变化的分布
- 适合高能物理、加速器设计等需实时相空间重建的场景
我们解决从稀疏2D投影序列恢复时变4D分布的问题,类似于从稀疏相机视角合成新视图,但应用于带电粒子束的4D横向相空间密度ρ(x, p_x, y, p_y)。真实加速器系统中无法直接测量该高维分布,仅能获取有限的1D或2D投影。本文提出PhaseFlow4D,一种基于反馈引导的潜在扩散模型,仅凭不完整的2D观测即可重建并跟踪完整的4D相空间,且内置硬性物理约束。核心技术是4D VAE,其解码器生成完整4D相空间张量,并从中解析计算2D投影,与实际测量对比。此投影一致性约束作为架构先验而非软惩罚,确保物理正确性。自适应反馈环持续调节潜在扩散模型的条件向量,实现无需重训练的在线动态追踪。我们在重离子束于稀有同位素束设施(FRIB)的多粒子模拟上验证,全物理模拟需约6小时(100核超算),而PhaseFlow4D以11000倍速度完成准确重建,并忠实追踪源条件变化下的分布偏移,证明在不完整观测下基于原理的生成重建可稳健迁移至非视觉领域。
原文摘要 · Abstract (English)
We address the problem of recovering a time-varying 4D distribution from a sparse sequence of 2D projections - analogous to novel-view synthesis from sparse cameras, but applied to the 4D transverse phase space density $ρ(x,p_x,y,p_y)$ of charged particle beams. Direct single shot measurement of this high-dimensional distribution is physically impossible in real particle accelerator systems; only limited 1D or 2D projections are accessible. We propose PhaseFlow4D, a feedback-guided latent diffusion model that reconstructs and tracks the full 4D phase space from incomplete 2D observations alone, with built-in hard physics constraints. Our core technical contribution is a 4D VAE whose decoder generates the full 4D phase space tensor, from which 2D projections are analytically computed and compared against 2D beam measurements. This projection-consistency constraint guarantees physical correctness by construction - not as a soft penalty, but as an architectural prior. An adaptive feedback loop then continuously tunes the conditioning vector of the latent diffusion model to track time-varying distributions online without retraining. We validate on multi-particle simulations of heavy-ion beams at the Facility for Rare Isotope Beams (FRIB), where full physics simulations require $\sim$6 hours on a 100-core HPC system. PhaseFlow4D achieves accurate 4D reconstructions 11000$\times$ faster while faithfully tracking distribution shifts under time-varying source conditions - demonstrating that principled generative reconstruction under incomplete observations transfers robustly beyond visual domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。