摘要
arXiv:2410.07250v2 通知类型: 替换-交叉
摘要:在高粒子多重性和严重重叠的簇射能量沉积下等极端操作条件下,经典粒子流算法在分辨率、效率和准确性方面遇到了明显的局限。为了解决这一挑战,本文提出并系统评估了一种深度学习重建框架:对于多通道稀疏特征,我们设计了一种结合加权均方误差和结构相似性指数的混合损失函数,有效地平衡了像素级准确性和结构保真度。通过将3D卷积、Squeeze-and-Excitation通道注意机制和偏移自注意力模块集成到基础卷积神经网络中,我们增强了模型捕获跨模态时空相关性和能量-位移非线性的能力。在自构建的模拟数据集和Pythia喷流数据集上验证,该框架的90K参数轻量级变体在性能上接近5M参数的基础模型,而25M参数的3D模型在插值和外推任务上均达到了最先进的结果。全面的实验定量评估了各个组件的贡献,并提供了性能与参数之间的权衡指南。所有核心代码和数据处理脚本均已开源在GitHub仓库中,以促进社区的再现和扩展。