核心概念
本文提出了一種基於圖神經網路的 Belle II 實驗中央漂移室端到端多軌跡重建演算法,該演算法在處理未知數量的粒子時表現出顯著的效能提升,特別是在位移軌跡的重建效率方面。
論文資訊
L. Reuter∗, G. De Pietro, S. Stefkova, T. Ferber, V. Bertacchi, G. Casarosa, L. Corona, P. Ecker, A. Glazov, Y. Han, M. Laurenza, T. Lueck, L. Massaccesi, S. Mondal, B. Scavino, S. Spataro, C. Wessel, L. Zani. End-to-End Multi-Track Reconstruction using Graph Neural Networks at Belle II. arXiv:2411.13596v1 [physics.ins-det] 18 Nov 2024.
研究目標
本研究旨在開發一種基於圖神經網路 (GNN) 的新型軌跡重建演算法 CAT Finder (CDC AI Track),用於 Belle II 實驗中央漂移室 (CDC) 中帶電粒子的軌跡重建,特別是針對從交互作用點位移的軌跡。
方法
使用 GEANT4 模擬 Belle II 探測器,並使用 basf2 模擬探測器響應,生成包含不同拓撲結構的模擬事件,包括瞬發軌跡、位移軌跡和頂點事件。
使用圖神經網路 (GNN) 架構,將探測器擊中點作為節點,並根據其空間接近度建立邊緣。
採用物件凝聚損失函數來處理未知數量的軌跡。
使用模擬數據集訓練和驗證 GNN 模型,並針對不同的超參數進行優化。
使用獨立的模擬樣本評估 CAT Finder 的性能,並與 Belle II 中使用的基準軌跡查找算法進行比較。
主要發現
與 Belle II 中使用的基準算法相比,CAT Finder 在各種不同事件拓撲結構中的軌跡查找效率均有顯著提高,特別是在位移軌跡方面。
對於質量在 GeV 範圍內的假設長壽命大質量粒子,其沿其飛行方向均勻衰減成兩個帶電粒子,GNN 的組合軌跡查找和擬合效率達到 85.4%,假陽性率為 2.5%,而基準算法的組合軌跡查找和擬合效率為 52.2%,假陽性率為 4.1%。
CAT Finder 在正向和反向探測器區域的瞬發軌跡效率和假陽性率也優於基準算法。
主要結論
本研究證明了基於 GNN 的端到端多軌跡重建算法在粒子物理學環境中的可行性和有效性。
CAT Finder 在重建位移軌跡方面表現出的改進,對於尋找超越標準模型的新物理現象具有重要意義。
意義
這是第一個在真實粒子物理環境中使用的漂移室探測器端到端多軌跡機器學習算法。
CAT Finder 的開發和實施有可能顯著提高 Belle II 實驗和其他粒子物理實驗的物理靈敏度。
局限性和未來研究
CAT Finder 的性能尚未針對 Belle II 實驗中可能遇到的所有物理過程進行全面優化。
未來的工作將集中於通過共同優化軌跡查找和擬合步驟來進一步提高 CAT Finder 的性能。
還將探索將 CAT Finder 與 Belle II 中使用的其他軌跡重建算法相結合,以進一步提高整體軌跡重建效率。
統計資料
模擬數據集包含超過 112 萬個事件,用於訓練和驗證 GNN 模型。
訓練後的 GNN 模型包含 797,812 個可訓練參數。
CAT Finder 在重建質量在 GeV 範圍內的假設長壽命大質量粒子的位移軌跡時,實現了 85.4% 的組合軌跡查找和擬合效率,假陽性率為 2.5%。
相比之下,基準算法在相同任務上的組合軌跡查找和擬合效率為 52.2%,假陽性率為 4.1%。