Rings of Light, Speed of AI: YOLO for Cherenkov Reconstruction
Martino Borsato, Giovanni Laganà, Maurizio Martinelli
TL;DR
The paper tackles the computational bottleneck of Cherenkov ring reconstruction for particle identification in RICH detectors by introducing a YOLO-based per-ring classifier. It uses momentum-bin closed-set classification with polar-transform ring images and annular masking to achieve real-time inference. The approach yields kaon identification efficiencies above 95% with pion misidentification around 1% across a broad momentum range and ~20 ms per full event on a high-end GPU, though a bin-dispatch overhead is currently a bottleneck. Compared with a maximum-likelihood fitter, YOLO is competitive in the momentum region most relevant for RICH1 and offers strong potential for online trigger applications, with plans to unify the model to remove dispatch overhead.
Abstract
Cherenkov rings play a crucial role in identifying charged particles in high-energy physics (HEP) experiments. Most Cherenkov ring pattern reconstruction algorithms currently used in HEP experiments rely on a likelihood fit to the photo-detector response, which often consumes a significant portion of the computing budget for event reconstruction. We present a novel approach to Cherenkov ring reconstruction using YOLO, a computer vision algorithm capable of real-time object identification with a single pass through a neural network. We obtain a reconstruction efficiency above 95% and a pion misidentification rate below 5% across a wide momentum range for all particle species.
