用点集神经网络直接处理液氩探测器的稀疏数据,大幅提速并提升精度。
Particle Hit Clustering and Identification Using Point Set Transformers in Liquid Argon Time Projection Chambers
- 采用点集变换器直接处理稀疏矩阵,避免转为密集矩阵
- 分类准确率提升14%,分割性能提升超22%,耗时减少80%
- 适合高能物理中粒子重建任务,尤其适用于稀疏数据场景
液氩时间投影室在中微子物理和暗物质搜索中广泛应用,因其高空间分辨率而备受青睐。然而,探测器生成的图像极为稀疏,绝大多数检测值为0,即尽管分辨率高,多数区域在单次相互作用中未被使用。传统方法通常将相互作用以稀疏矩阵形式存储——仅记录有能量值的检测位置及其对应数值。但诸如卷积神经网络(CNN)等传统机器学习方法无法直接处理此类稀疏结构,必须将矩阵完全展开为密集矩阵,导致内存与计算开销巨大。本文提出一种基于点集神经网络的模型,可直接在稀疏矩阵上运行,显著提升处理速度与准确性。相比现有先进方法,本方法在分类性能上提升14%,分割性能提升超过22%,运行时间减少80%,内存占用降低66%;相较最先进的CNN方法,分类性能提升超86%,分割性能提升超71%,运行时间减少91%,内存使用减少61%。
原文摘要 · Abstract (English)
Liquid argon time projection chambers are often used in neutrino physics and dark-matter searches because of their high spatial resolution. The images generated by these detectors are extremely sparse, as the energy values detected by most of the detector are equal to 0, meaning that despite their high resolution, most of the detector is unused in a particular interaction. Instead of representing all of the empty detections, the interaction is usually stored as a sparse matrix, a list of detection locations paired with their energy values. Traditional machine learning methods that have been applied to particle reconstruction such as convolutional neural networks (CNNs), however, cannot operate over data stored in this way and therefore must have the matrix fully instantiated as a dense matrix. Operating on dense matrices requires a lot of memory and computation time, in contrast to directly operating on the sparse matrix. We propose a machine learning model using a point set neural network that operates over a sparse matrix, greatly improving both processing speed and accuracy over methods that instantiate the dense matrix, as well as over other methods that operate over sparse matrices. Compared to competing state-of-the-art methods, our method improves classification performance by 14%, segmentation performance by more than 22%, while taking 80% less time and using 66% less memory. Compared to state-of-the-art CNN methods, our method improves classification performance by more than 86%, segmentation performance by more than 71%, while reducing runtime by 91% and reducing memory usage by 61%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。