REVIEW 2 major objections 2 minor 47 references
An end-to-end point transformer reconstructs charged particle tracks directly from detector hits without graph construction or auxiliary stages.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · grok-4.3
2026-06-26 14:55 UTC pith:2NVO45TM
load-bearing objection HEPTv2 reports a clean end-to-end point transformer that beats prior graph and transformer baselines on TrackML efficiency and latency, but the abstract gives almost no methods or ablation detail to judge whether the LSH encoder actually drives the gains. the 2 major comments →
HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The presented architecture uses a locality-aware point encoder that applies locality-sensitive hashing in detector coordinate space to enable efficient local attention, paired with a track decoder that performs sectorized decoding and direct hit-to-track assignment under joint supervision, yielding an end-to-end model that reconstructs trajectories in one pipeline and reaches 98.6 percent double-majority efficiency at 0.8 percent fake rate with linear scaling in latency and memory.
What carries the argument
Locality-sensitive hashing in detector coordinate space that preserves tracking-relevant geometry to support efficient local attention without explicit graph construction.
Load-bearing premise
Locality-sensitive hashing in detector coordinate space preserves all information needed to resolve track ambiguities without meaningful loss.
What would settle it
On a new set of events with higher hit density or altered geometry, the efficiency falls below that of the strongest graph-based baseline while latency remains comparable.
If this is right
- The model processes events with up to 500000 hits with latency and memory that scale linearly.
- Joint encoder-decoder training removes the need for separate clustering or filtering stages.
- End-to-end optimization produces a better accuracy-latency tradeoff than staged pipelines.
- Direct hit-to-track prediction resolves combinatorial ambiguities without intermediate graph objects.
Where Pith is reading between the lines
- The same hashing-plus-direct-decoding pattern could transfer to other sparse reconstruction tasks that currently rely on graph construction.
- Linear scaling suggests the method remains practical when detector occupancy increases beyond current test conditions.
- Removing graph-building stages may reduce sensitivity to the precise choice of detector coordinate system.
- Sectorized decoding offers a route to parallelization on hardware where memory bandwidth is the bottleneck.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper presents HEPTv2, an end-to-end point-transformer architecture for charged-particle tracking. It combines a locality-aware point encoder that uses locality-sensitive hashing (LSH) in detector coordinate space with a track decoder that performs direct hit-to-track prediction under joint supervision, eliminating graph construction and auxiliary stages. On the TrackML benchmark the model reports 98.6% double-majority efficiency at 0.8% fake rate, ~15 ms inference time and 0.4 GB peak memory per event on an A100 GPU, together with linear scaling up to 5×10^5 hits and gains of 4.5% efficiency and factors of 7–52 in latency over prior transformers and optimized graph pipelines.
Significance. If the reported metrics are reproducible and the architecture generalizes beyond TrackML, the result would be significant: it would demonstrate that a fully end-to-end transformer can match or exceed the accuracy of graph-based methods while delivering substantially lower latency and memory, directly addressing the real-time requirements of HL-LHC tracking. The absence of graph-building and the joint encoder-decoder training are notable technical strengths.
major comments (2)
- [Abstract / encoder section] Abstract and Methods (encoder description): the central efficiency claim (98.6% double-majority at 0.8% fake rate) rests on the assertion that LSH in coordinate space preserves all tracking-relevant geometry. No hash-function parameters, collision-rate statistics, or ablation that isolates the LSH contribution are supplied; without these it is impossible to determine whether the reported gains are architecture-driven or specific to the TrackML hit-density distribution.
- [Abstract] Abstract: the performance numbers are stated without training details, validation splits, error bars, or ablation studies. This absence makes it impossible to assess whether the efficiency and fake-rate figures are statistically supported or sensitive to hyper-parameter choices.
minor comments (2)
- [Abstract] Abstract: the statement that latency and memory “scale approximately linearly” is not accompanied by a supporting figure or table showing the scaling data.
- [Abstract] Abstract: the specific TrackML dataset version, event selection, and double-majority definition used for the quoted metrics are not stated.
Simulated Author's Rebuttal
We thank the referee for the constructive and detailed review. The comments highlight important aspects of reproducibility and technical transparency that we will address through revisions.
read point-by-point responses
-
Referee: [Abstract / encoder section] Abstract and Methods (encoder description): the central efficiency claim (98.6% double-majority at 0.8% fake rate) rests on the assertion that LSH in coordinate space preserves all tracking-relevant geometry. No hash-function parameters, collision-rate statistics, or ablation that isolates the LSH contribution are supplied; without these it is impossible to determine whether the reported gains are architecture-driven or specific to the TrackML hit-density distribution.
Authors: We agree that the current manuscript lacks explicit LSH parameters, collision statistics, and an isolating ablation. In the revision we will add a dedicated subsection to the Methods describing the hash functions (including number of tables, bucket size, and random projection dimensions), measured collision rates on TrackML, and an ablation that replaces LSH with uniform random sampling or full attention. This will allow readers to separate the contribution of coordinate-space locality preservation from the overall architecture. We maintain that the end-to-end training and decoder design are the primary sources of the reported gains, but the requested ablation will make this explicit. revision: yes
-
Referee: [Abstract] Abstract: the performance numbers are stated without training details, validation splits, error bars, or ablation studies. This absence makes it impossible to assess whether the efficiency and fake-rate figures are statistically supported or sensitive to hyper-parameter choices.
Authors: We accept that the abstract and main text currently omit these elements. The revised version will include a new 'Experimental Setup' subsection that specifies the training protocol, the exact TrackML train/validation/test split used, error bars computed over five independent random seeds (mean ± std), and a set of ablations on key hyperparameters (attention heads, sector size, loss coefficients). These additions will substantiate the statistical reliability of the 98.6 % / 0.8 % figures and demonstrate robustness to hyper-parameter variation. revision: yes
Circularity Check
No circularity: empirical performance on external benchmark
full rationale
The paper reports measured tracking efficiency (98.6% double-majority at 0.8% fake rate), latency (~15 ms), and memory (0.4 GB) directly from evaluation on the TrackML dataset. No equations, fitted parameters, or derivation steps are presented that reduce any claimed result to its inputs by construction. The architecture description (LSH encoder + sectorized decoder) is presented as a design choice whose performance is validated empirically rather than derived tautologically. No self-citation load-bearing steps appear in the supplied text.
Axiom & Free-Parameter Ledger
read the original abstract
Charged-particle tracking -- reconstructing trajectories from sparse detector measurements -- is a fundamental high-energy-physics inference problem and a canonical example of learning under extreme combinatorial ambiguity. At the High-Luminosity Large Hadron Collider (HL-LHC), tracking must remain accurate and efficient despite unprecedented collision densities. Graph neural networks perform strongly, but incur substantial costs from graph construction and processing, while transformer-based approaches rely on auxiliary stages that prevent end-to-end optimization. To address this, we present HEPTv2, an end-to-end point-transformer architecture that reconstructs tracks from detector hits in one trainable pipeline. HEPTv2 combines a locality-aware point encoder with a track decoder that predicts complete trajectories without graph-building, clustering, or filtering. The encoder uses locality-sensitive hashing in detector coordinate space to preserve tracking-relevant geometry while enabling efficient local attention. The decoder resolves ambiguities through sectorized decoding and direct hit-to-track prediction under joint encoder-decoder supervision, allowing the full pipeline to be optimized end-to-end. On TrackML, HEPTv2 achieves 98.6% double-majority tracking efficiency at a 0.8% fake rate, while requiring only $\sim$15~ms inference time and 0.4~GB peak memory per event on a NVIDIA A100 GPU. Latency and memory scale approximately linearly for events with up to $5\times10^5$ hits. HEPTv2 establishes a new state of the art in the accuracy-latency trade-off, improving efficiency by 4.5% over the strongest prior transformer and by 1.1--2.2% over optimized graph-based pipelines, while reducing latency by factors of 7 and 38--52, respectively. These results show end-to-end transformers can deliver the accuracy and efficiency required for real-time particle reconstruction at the HL-LHC.
Reference graph
Works this paper leans on
-
[1]
Performance of a geometric deep learning pipeline for hl-lhc particle tracking
X. Ju et al., “Performance of a geometric deep learning pipeline for hl-lhc particle tracking”,The European Physical Journal C81(2021) 1–14
2021
-
[2]
S. Miao et al., “Locality-sensitive hashing-based efficient point transformer with applications in high-energy physics”,arXiv preprint arXiv:2402.12535(2024)
-
[3]
The trackml high-energy physics tracking challenge on kaggle
M. Kiehn et al., “The trackml high-energy physics tracking challenge on kaggle”, inEPJ Web of Conferences, volume 214, p. 06037, EDP Sciences. 2019
2019
-
[4]
Mlpf: efficient machine-learned particle-flow reconstruction using graph neural networks
J. Pata et al., “Mlpf: efficient machine-learned particle-flow reconstruction using graph neural networks”, The European Physical Journal C81(2021), no. 5, 381
2021
-
[5]
Graph neural networks in particle physics
J. Shlomi, P. Battaglia, and J.-R. Vlimant, “Graph neural networks in particle physics”,Machine Learning: Science and Technology2(2021), no. 2, 021001
2021
-
[6]
Graph neural networks for particle tracking and reconstruction
J. Duarte and J.-R. Vlimant, “Graph neural networks for particle tracking and reconstruction”, inArtificial intelligence for high energy physics, pp. 387–436. World Scientific, 2022
2022
-
[7]
Track and vertex reconstruction: From classical to adaptive methods
A. Strandlie and R. Fr ¨uhwirth, “Track and vertex reconstruction: From classical to adaptive methods”, Reviews of modern physics82(2010), no. 2, 1419–1458
2010
-
[8]
Charged particle tracking via edge-classifying interaction networks
G. DeZoort et al., “Charged particle tracking via edge-classifying interaction networks”,Computing and Software for Big Science5(2021), no. 1, 26
2021
-
[9]
High-luminosity large hadron collider (hl-lhc): Technical design report
O. Aberle et al., “High-luminosity large hadron collider (hl-lhc): Technical design report”,
-
[10]
Expected tracking performance of the atlas inner tracker at the hl-lhc
A. Collaboration et al., “Expected tracking performance of the atlas inner tracker at the hl-lhc”,
-
[11]
Software performance of the atlas track reconstruction for lhc run 3
A. C. atlas. publications@ cern. ch et al., “Software performance of the atlas track reconstruction for lhc run 3”,Computing and Software for Big Science8(2024), no. 1, 9
2024
-
[12]
Speeding up particle track reconstruction using a parallel kalman filter algorithm
S. Lantz et al., “Speeding up particle track reconstruction using a parallel kalman filter algorithm”,Journal of Instrumentation15(2020), no. 09, P09030–P09030
2020
-
[13]
Graph neural networks for charged particle tracking on fpgas
A. Elabd et al., “Graph neural networks for charged particle tracking on fpgas”,Frontiers in big Data5 (2022) 828666
2022
-
[14]
traccc: Gpu track reconstruction library for hep experiments
P. Gessinger et al., “traccc: Gpu track reconstruction library for hep experiments”, inEPJ Web of Conferences, volume 337, p. 01187, EDP Sciences. 2025
2025
-
[15]
The tracking machine learning challenge: accuracy phase
S. Amrouche et al., “The tracking machine learning challenge: accuracy phase”, inThe NeurIPS’18 Competition: From Machine Learning to Intelligent Conversations, pp. 231–264. Springer, 2019
2019
-
[16]
Application of kalman filtering to track and vertex fitting
R. Fr ¨uhwirth, “Application of kalman filtering to track and vertex fitting”,Nuclear Instruments and Methods in Physics Research Section A: Accelerators, Spectrometers, Detectors and Associated Equipment262 (1987), no. 2-3, 444–450
1987
-
[17]
The hep. trkx project: deep neural networks for hl-lhc online and offline tracking
S. Farrell et al., “The hep. trkx project: deep neural networks for hl-lhc online and offline tracking”, inEPJ Web of Conferences, volume 150, p. 00003, EDP Sciences. 2017
2017
-
[18]
A pattern recognition algorithm for quantum annealers
F. Bapst et al., “A pattern recognition algorithm for quantum annealers”,Computing and Software for Big Science4(2020), no. 1, 1
2020
-
[19]
Charged particle tracking with machine learning on fpgas
H. Abidi et al., “Charged particle tracking with machine learning on fpgas”,arXiv preprint arXiv:2212.02348(2022). 14
-
[20]
Computational performance of the atlas itk gnn track reconstruction pipeline
A. collaboration et al., “Computational performance of the atlas itk gnn track reconstruction pipeline”, technical report, LHC/ATLAS Experiment, 2024
2024
-
[21]
High pileup particle tracking with object condensation
K. Lieret et al., “High pileup particle tracking with object condensation”,arXiv preprint arXiv:2312.03823 (2023)
-
[22]
Accelerating the inference of the exa. trkx pipeline
A. Lazar et al., “Accelerating the inference of the exa. trkx pipeline”, inJournal of Physics: Conference Series, volume 2438, p. 012008, IOP Publishing. 2023
2023
-
[23]
Attention is all you need
A. Vaswani et al., “Attention is all you need”,Advances in neural information processing systems30(2017)
2017
-
[24]
Big bird: Transformers for longer sequences
M. Zaheer et al., “Big bird: Transformers for longer sequences”,Advances in neural information processing systems33(2020) 17283–17297
2020
-
[25]
Swin transformer: Hierarchical vision transformer using shifted windows
Z. Liu et al., “Swin transformer: Hierarchical vision transformer using shifted windows”, inProceedings of the IEEE/CVF international conference on computer vision, pp. 10012–10022. 2021
2021
-
[26]
Flatformer: Flattened window attention for efficient point cloud transformer
Z. Liu et al., “Flatformer: Flattened window attention for efficient point cloud transformer”, inProceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 1200–1211. 2023
2023
-
[27]
Point transformer v3: Simpler faster stronger
X. Wu et al., “Point transformer v3: Simpler faster stronger”, inProceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 4840–4851. 2024
2024
-
[28]
A density-based algorithm for discovering clusters in large spatial databases with noise
M. Ester et al., “A density-based algorithm for discovering clusters in large spatial databases with noise”, in kdd, volume 96, pp. 226–231. 1996
1996
-
[29]
End-to-end object detection with transformers
N. Carion et al., “End-to-end object detection with transformers”, inEuropean conference on computer vision, pp. 213–229, Springer. 2020
2020
-
[30]
Masked-attention mask transformer for universal image segmentation
B. Cheng et al., “Masked-attention mask transformer for universal image segmentation”, inProceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 1290–1299. 2022
2022
-
[31]
Mask3d: Mask transformer for 3d semantic instance segmentation
J. Schult et al., “Mask3d: Mask transformer for 3d semantic instance segmentation”, in2023 IEEE International Conference on Robotics and Automation (ICRA), pp. 8216–8223, IEEE. 2023
2023
-
[32]
Transformers for charged particle track reconstruction in high-energy physics
S. Van Stroud et al., “Transformers for charged particle track reconstruction in high-energy physics”, Physical Review X15(2025), no. 4, 041046
2025
-
[33]
Representation Learning with Contrastive Predictive Coding
A. v. d. Oord, Y . Li, and O. Vinyals, “Representation learning with contrastive predictive coding”,arXiv preprint arXiv:1807.03748(2018)
work page internal anchor Pith review Pith/arXiv arXiv 2018
-
[34]
Hierarchical graph neural networks for particle track reconstruction
R. Liu et al., “Hierarchical graph neural networks for particle track reconstruction”, inJournal of Physics: Conference Series, volume 3206, p. 012084, IOP Publishing. 2026
2026
-
[35]
FlatQuant: Flatness Matters for LLM Quantization
Y . Sun et al., “Flatquant: Flatness matters for llm quantization”,arXiv preprint arXiv:2410.09426(2024)
work page Pith review arXiv 2024
-
[36]
Qserve: W4a8kv4 quantization and system co-design for efficient llm serving
Y . Lin et al., “Qserve: W4a8kv4 quantization and system co-design for efficient llm serving”,Proceedings of Machine Learning and Systems7(2025)
2025
-
[37]
Bytetransformer: A high-performance transformer boosted for variable-length inputs
Y . Zhai et al., “Bytetransformer: A high-performance transformer boosted for variable-length inputs”, in 2023 IEEE International Parallel and Distributed Processing Symposium (IPDPS), pp. 344–355, IEEE. 2023
2023
-
[38]
Learning representations of irregular particle-detector geometry with distance-weighted graph networks
S. R. Qasim, J. Kieseler, Y . Iiyama, and M. Pierini, “Learning representations of irregular particle-detector geometry with distance-weighted graph networks”,European Physical Journal C79(2019), no. 7, 608
2019
-
[39]
Scalable deep convolutional neural networks for sparse, locally dense liquid argon time projection chamber data
L. Domin ´e and K. Terao, “Scalable deep convolutional neural networks for sparse, locally dense liquid argon time projection chamber data”,Physical Review D102(2020), no. 1, 012005. 15
2020
-
[40]
Graph neural networks for low-energy event classification and reconstruction in icecube
R. Abbasi et al., “Graph neural networks for low-energy event classification and reconstruction in icecube”, Journal of Instrumentation17(2022), no. 11, P11003
2022
-
[41]
GraphNeT: Graph neural networks for neutrino telescope event reconstruction
A. Søgaard et al., “Graphnet: Graph neural networks for neutrino telescope event reconstruction”,arXiv preprint arXiv:2210.12194(2022)
work page Pith review arXiv 2022
-
[42]
ATLAS ITk Track Reconstruction with a GNN-based pipeline
ATLAS Collaboration, “ATLAS ITk Track Reconstruction with a GNN-based pipeline”, technical report, CERN, Geneva, 2022
2022
-
[43]
Locality-sensitive hashing scheme based on p-stable distributions
M. Datar, N. Immorlica, P. Indyk, and V . S. Mirrokni, “Locality-sensitive hashing scheme based on p-stable distributions”, inProceedings of the twentieth annual symposium on Computational geometry, pp. 253–262. 2004
2004
-
[44]
The hungarian method for the assignment problem
H. W. Kuhn, “The hungarian method for the assignment problem”,Naval research logistics quarterly2 (1955), no. 1-2, 83–97
1955
-
[45]
Focal loss for dense object detection
T.-Y . Lin et al., “Focal loss for dense object detection”, inProceedings of the IEEE international conference on computer vision, pp. 2980–2988. 2017
2017
-
[46]
V-net: Fully convolutional neural networks for volumetric med- ical image segmentation
F. Milletari, N. Navab, and S.-A. Ahmadi, “V-net: Fully convolutional neural networks for volumetric med- ical image segmentation”, in2016 fourth international conference on 3D vision (3DV), pp. 565–571, Ieee. 2016
2016
-
[47]
Muon is Scalable for LLM Training
J. Liu et al., “Muon is scalable for llm training”,arXiv preprint arXiv:2502.16982(2025). 16
work page internal anchor Pith review Pith/arXiv arXiv 2025
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.