Deep neural network uncertainty quantification for LArTPC reconstruction
- Stanford University, CA (United States)
- SLAC National Accelerator Laboratory (SLAC), Menlo Park, CA (United States)
We evaluate uncertainty quantification (UQ) methods for deep learning applied to liquid argon time projection chamber (LArTPC) physics analysis tasks. As deep learning applications enter widespread usage among physics data analysis, neural networks with reliable estimates of prediction uncertainty and robust performance against overconfidence and out-of-distribution (OOD) samples are critical for their full deployment in analyzing experimental data. While numerous UQ methods have been tested on simple datasets, performance evaluations for more complex tasks and datasets are scarce. Here we assess the application of selected deep learning UQ methods on the task of particle classification using the PiLArNet monte carlo 3D LArTPC point cloud dataset. We observe that UQ methods not only allow for better rejection of prediction mistakes and OOD detection, but also generally achieve higher overall accuracy across different task settings. We assess the precision of uncertainty quantification using different evaluation metrics, such as distributional separation of prediction entropy across correctly and incorrectly identified samples, receiver operating characteristic curves (ROCs), and expected calibration error from observed empirical accuracy. We conclude that ensembling methods can obtain well calibrated classification probabilities and generally perform better than other existing methods in deep learning UQ literature.
- Research Organization:
- SLAC National Accelerator Laboratory (SLAC), Menlo Park, CA (United States)
- Sponsoring Organization:
- USDOE Office of Science (SC), High Energy Physics (HEP); Zoox, Inc
- Grant/Contract Number:
- AC02-76SF00515
- OSTI ID:
- 2326991
- Journal Information:
- Journal of Instrumentation, Vol. 18, Issue 12; ISSN 1748-0221
- Publisher:
- Institute of Physics (IOP)Copyright Statement
- Country of Publication:
- United States
- Language:
- English
Similar Records
LTAU-FF: Loss Trajectory Analysis for Uncertainty in atomistic Force Fields
Uncertainty Quantification of Capacitor Switching Transient Location using Machine Learning