DCTracks: An Open Dataset for Machine Learning-Based Drift Chamber Track Reconstruction
Authors: Qian Liyan, Zhang Yao, Yuan Ye, Zhang Zhaoke, Fang Jin, Jiang Shimiao, Zhang Jin, Li Ke, +6 more
Organizations: Institute of High Energy Physics, Chinese Academy of Sciences No.19B Yuquan Road, Shijingshan, Beijing, China · University of Chinese Academy of Sciences No.19A Yuquan Road, Shijingshan, Beijing, China · Sun Yat-sen University School of Science, Shenzhen Campus of Sun Yat-sen University, Shenzhen, China · China Academy of Space Technology No.104 Youyi Road, Haidian, Beijing, China · Institute of Automation Chinese Academy of Sciences No.95 Zhongguancun East Road, Haidian, Beijing, China · Key Laboratory of Particle Physics and Particle Irradiation (MOE), Institute of Frontier and Interdisciplinary Science, Shandong University, Qingdao, Shandong, China
We introduce a Monte Carlo (MC) dataset of single- and two-track drift chamber events to advance Machine Learning (ML)-based track reconstruction. To enable standardized and comparable evaluation, we define track reconstruction specific metrics and report results for traditional track reconstruction algorithms and a Graph Neural Networks (GNNs) method, facilitating rigorous, reproducible validation for future research.
Figures & tables
Figure 1: BESIII MDC structure (left) and 3D view of the event (right).
Superlayer
Type
Nlayer
Nwire/layer
Radius (mm)
Length (mm)
1
U
4
40,44,48,56
∼ 79 – 115
780–816
2
V
4
64,72,80,80
∼ 127 – 162
828 – 864
3
A
4
76,76,88,88
∼ 197 – 246
1092 –1272
4
A
4
100,100,112,112
∼ 262 – 311
1442 – 1612
5
A
4
128,128,140,140
∼ 327 – 375
1782 – 1952
6
U
4
160 × 4
∼ 400 – 448
2174 – 2192
Table 1: MDC layer structure and geometry parameters.
Event Type
pT [GeV/ c ]
cosθ
ϕ [rad]
Particles
Single-track
0.15∼1.5
−0.93∼0.93
0∼2π
e±,μ±,π±,K±,p,pˉ
Conventional two-track
0.15∼1.5
−0.93∼0.93
0∼2π
π+π−
Close-by two-track
0.15∼1.5
−0.93∼0.93
Δϕ=0.2
π+π−
Table 2: Kinematic settings for single-track and two-track event simulation.
Figure 2: Displays of the simulated events in the x-y plane for a single-track event (left), a conventional two-track event (middle) and a close-by two-track event (right).
Figure 3: Displays of reconstructed events in the x-y plane for a single-track event (left), a conventional two-track event (middle) and a close-by two-track event (right). A condensation point on the track provides estimates the track parameters. This concept is closely related to the GNN finding method we use.
in %
Event Type
ϵhit
phit
Baseline Finder
Single-track ( π+ )
92.24+0.12−0.12
98.58+0.05−0.05
GNN Finder
92.20+0.12−0.12
98.91+0.05−0.05
Baseline Finder
Conventional two-track ( π+π− )
90.87+0.14−0.14
97.93+0.07−0.07
GNN Finder
91.62+0.13−0.13
98.83+0.05−0.05
Baseline Finder
Close-by two-track ( π+π− )
91.26+0.16−0.16
97.95+0.08−0.08
GNN Finder
82.68+0.21−0.21
97.89+0.08−0.08
Table 3: Hit efficiency ( ϵhit ) and hit purity ( phit ) for different event categories.
Figure 4: Hit efficiency and hit purity for tracks found by both the GNN Finder and the Baseline Finder . Results are shown as functions of pTMC (left column) and cosθMC (right column) for single-track π+ events.
in %
ϵtrack
ϵtrack,q
Rclone
Rfake
Rwrong,q
Single-track π+ events
Baseline Finder
99.71+0.02−0.02
99.69+0.02−0.02
0.07+0.01−0.01
0.01
0.02+0.01−0.01
GNN Finder
99.81+0.02−0.02
99.55+0.03−0.03
0.00
0.01
0.27+0.02−0.02
Baseline Fitter
99.70+0.02−0.02
99.68+0.03−0.03
0.06+0.01−0.01
0.01
0.02+0.01−0.01
GNN Fitter
99.75+0.02−0.02
99.50+0.03−0.03
0.00
0.01
0.25+0.02−0.02
Conventional two-track π+π− events
Table 4: Track finding and fitting efficiencies for different event categories.
Figure 5: Track efficiency and track charge efficiency for tracks found by both the GNN Finder (orange) and the Baseline Finder (blue) with and without fitting. Results are shown as functions of pTMC (left column) and cosθMC (right column) for single-track π+ events.
Figure 6: Helix parametrization of a particle trajectory.
Figure 7: Relative transverse momentum resolution for tracks found by both the GNN Finder and the Baseline Finder with and without fitting. Results are shown as a function of pTMC for single-track events (left), conventional two-track events (middle) and close-by two-track (right) events.
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 8: Hit efficiency and hit purity for tracks found by both the GNN Finder and the Baseline Finder . Results are shown as functions of pTMC (left column) and cosθMC (right column) for conventional two-track π+π− events.
Figure 9: Hit efficiency and hit purity for tracks found by both the GNN Finder and the Baseline Finder . Results are shown as functions of pTMC (left column) and cosθMC (right column) for close-by two-track π+π− events.
Figure 10: Track efficiency and track charge efficiency for tracks found by both the GNN Finder (orange) and the Baseline Finder (blue) with and without fitting. Results are shown as functions of pTMC (left column) and cosθMC (right column) for conventional two-track π+π− events.
Figure 11: Track efficiency and track charge efficiency for tracks found by both the GNN Finder (orange) and the Baseline Finder (blue) with and without fitting. Results are shown as functions of pTMC (left column) and cosθMC (right column) for close-by two-track π+π− events.
Figure 12: Track parameters of single-track π+ events.
Figure 13: Track parameters of conventional two-track π+π− events.
Figure 14: Track parameters of close-by two-track π+π− events.
High-pileup conditions in CMS during the HL-LHC era make charged-particle tracking increasingly challenging as detector occupancy and combinatorics grow. We present a hybrid approach that exploits Line Segment Tracking (LST) objects rather than individual hits to enable the first CMS ML-based track reconstruction algorithm. The LST objects, namely the triplets used in this study, are built according to geometry- and physics-driven criteria and carry richer local structure, geometrical cues, and physics quantities such as transverse momentum, which helps the model incorporate detector-level information from the start. We apply a Transformer encoder with geometrically masked sparse self-attention to embed these objects in a learned latent space, and use object condensation to assemble full tracks in a single inference step. This object-based representation reduces complexity in dense environments and makes the model less sensitive to ambiguities introduced by high pileup. Evaluated on CMS Phase-2 simulation of ttˉ events at ⟨μ⟩=200, the approach achieves reconstruction efficiency comparable to the LST quintuplet baseline while reducing the fake rate by roughly a factor of three and suppressing duplicates almost entirely, demonstrating the promise of advanced ML architectures for next-generation particle tracking.
Aashay Arora on behalf of the CMS Collaboration
University of California San Diego, La Jolla, CA, USA
The identification and reconstruction of charged particles, such as muons, is a main challenge for the physics program of the ATLAS experiment at the Large Hadron Collider. This task will become increasingly difficult with the start of the High-Luminosity LHC era after 2030, when the number of proton-proton collisions per bunch crossing will increase from 60 to up to 200. This elevated interaction density will also increase the occupancy within the ATLAS Muon Spectrometer, requiring more efficient and robust real-time data processing strategies within the experiment's trigger system, particularly the Event Filter. To address these algorithmic challenges, we present two machine-learning-based approaches. First, we target the problem of background-hit rejection in the Muon Spectrometer using Graph Neural Networks integrated into the non-ML baseline reconstruction chain, demonstrating a 15 % improvement in reconstruction speed (from 255 ms to 217 ms). Second, we present a proof-of-concept for end-to-end muon tracking using state-of-the-art Vision Transformer architectures, achieving ultra-fast approximate muon reconstruction in 2.3 ms on consumer-grade GPUs at 98 % tracking efficiency.
Jonathan Renusch
on behalf of the ATLAS Collaboration · CERN, Geneva, Switzerland
In the forthcoming years the LHC experiments are going to be upgraded to benefit from the substantial increase of the LHC instantaneous luminosity, which will lead to larger, denser events, and, consequently, greater complexity in reconstructing charged particle tracks, motivating frontier research in new technologies. Quantum machine learning models are being investigated as potential new approaches to high energy physics (HEP) tasks. We characterize and upgrade a quantum graph neural network (QGNN) architecture for charged particle track reconstruction on a simulated high luminosity dataset. The model operates on a set of event graphs, each built from the hits generated in tracking detector layers by particles produced in proton collisions, performing a classification of the possible hit connections between adjacent layers. In this approach the QGNN is designed as a hybrid architecture, interleaving classical feedforward networks with parametrized quantum circuits. We characterize the interplay between the classical and quantum components. We report on the principal upgrades to the original design, and present new evidence of improved training behavior, specifically in terms of convergence toward the final trained configuration.
Matteo Argenton, Laura Cappelli, Concezio Bozzi
Department of Physics and Earth Science, University of Ferrara, Via Saragat 1, Ferrara, 44122, Italy · INFN Ferrara, Via Saragat 1, Ferrara, 44122, Italy