arXiv:2411.01641cs.LGhep-ex2024-11被引 11

用量子图网络提升高能物理数据处理,更省参数还更准。

Lorentz-Equivariant Quantum Graph Neural Network for High-Energy Physics

  • 用量子电路替代传统模块,保持洛伦兹对称性。
  • 在夸克-胶子喷注识别上达74%准确率,仅用4个量子比特。
  • 适合数据少、噪声大的高能物理任务,如喷注分类。

高亮度大型强子对撞机的数据激增带来了严峻的计算挑战,亟需高效的数据处理新方法。量子机器学习凭借量子硬件庞大的希尔伯特空间,展现出巨大潜力。然而,现有量子图神经网络(GNN)抗噪能力弱,且受限于固定对称群,难以适应复杂粒子相互作用建模。本文将LorentzNet中的洛伦兹等变模块替换为经修饰的量子电路,性能显著提升,参数量减少近5.5倍。量子电路天然保持对称性,可有效替代多层感知机(MLP),并集成洛伦兹对称性以稳健处理相对论不变性。所提出的洛伦兹等变量子图神经网络(Lorentz-EQGNN)在夸克-胶子喷注标签数据集上达到74.00%测试准确率和87.38%的AUC,优于经典与量子基线模型,且仅使用4个量子比特。在电子-光子数据集上取得67.00%准确率和68.20% AUC,仅需800个训练样本即表现优异。在通用MNIST和FashionMNIST数据集上分别实现88.10%和74.80%的准确率。消融实验验证了量子组件的关键作用,背景拒识率显著优于经典方法。结果表明,Lorentz-EQGNN在抗噪喷注标签、事件分类及数据稀缺的高能物理任务中具有直接应用潜力。

原文摘要 · Abstract (English)

The rapid data surge from the high-luminosity Large Hadron Collider introduces critical computational challenges requiring novel approaches for efficient data processing in particle physics. Quantum machine learning, with its capability to leverage the extensive Hilbert space of quantum hardware, offers a promising solution. However, current quantum graph neural networks (GNNs) lack robustness to noise and are often constrained by fixed symmetry groups, limiting adaptability in complex particle interaction modeling. This paper demonstrates that replacing the Lorentz Group Equivariant Block modules in LorentzNet with a dressed quantum circuit significantly enhances performance despite using nearly 5.5 times fewer parameters. Additionally, quantum circuits effectively replace MLPs by inherently preserving symmetries, with Lorentz symmetry integration ensuring robust handling of relativistic invariance. Our Lorentz-Equivariant Quantum Graph Neural Network (Lorentz-EQGNN) achieved $74.00\%$ test accuracy and an AUC of $87.38\%$ on the Quark-Gluon jet tagging dataset, outperforming the classical and quantum GNNs with a reduced architecture using only 4 qubits. On the Electron-Photon dataset, Lorentz-EQGNN reached $67.00\%$ test accuracy and an AUC of $68.20\%$, demonstrating competitive results with just 800 training samples. Evaluation of our model on generic MNIST and FashionMNIST datasets confirmed Lorentz-EQGNN's efficiency, achieving $88.10\%$ and $74.80\%$ test accuracy, respectively. Ablation studies validated the impact of quantum components on performance, with notable improvements in background rejection rates over classical counterparts. These results highlight Lorentz-EQGNN's potential for immediate applications in noise-resilient jet tagging, event classification, and broader data-scarce HEP tasks.

量子神经网络高能物理图神经网络对称性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。