Please visit Jefferson Lab Event Policies and Guidance before planning your next event:
May 8 – 12, 2023
Norfolk Waterside Marriott
US/Eastern timezone

Graph Neural Network for 3D Reconstruction in Liquid Argon Time Projection Chambers

May 9, 2023, 12:15 PM
Hampton Roads VII (Norfolk Waterside Marriott)

Hampton Roads VII

Norfolk Waterside Marriott

235 East Main Street Norfolk, VA 23510
Oral Track 9 - Artificial Intelligence and Machine Learning Track 9 - Artificial Intelligence and Machine Learning


Hewes, V (University of Cincinnati)


The Exa.TrkX team has developed a Graph Neural Network (GNN) for reconstruction of liquid argon time projection chamber (LArTPC) data. We discuss the network architecture, a multi-head attention message passing network that classifies detector hits according to the particle type that produced them. By utilizing a heterogeneous graph structure with independent subgraphs for each 2D plane’s hits and for 3D space points, the model achieves a consistent description of the neutrino interaction across all planes.

Performance results will be presented based on publicly available samples from MicroBooNE. These will include both physics performance metrics, achieving ~95% accuracy when integrated over all particle classes, and computational metrics for training on single or distributed GPU systems and for inference on CPU or GPU.

We will discuss applications of the network for additional LArTPC reconstruction tasks, such as event classification, cosmic rejection and particle instance segmentation. Prospects for integration in the data processing chains of experiments will also be presented.

Consider for long presentation Yes

Primary author

Hewes, V (University of Cincinnati)


Cerati, Giuseppe (FNAL) Kowalkowski, Jim (FNAL) Prof. Aurisano, Adam (University of Cincinnati) Prof. Liao, Wei-Keng (Northwestern University) Ms Lee, Claire (Northwestern University) Ms Day, Alexandra (Northwestern University) Mr Gumpula, Kaushal (University of Chicago Data Science Institute) Mr Koloskov, Nikita (University of Chicago Data Science Institute) Ms Koretsky, Samantha (University of Chicago Data Science Institute) Ms Zhang, Haley (University of Chicago Data Science Institute) Ms Yufei, Xiao (University of Chicago Data Science Institute) Dr Grzenda, Daniel (University of Chicago Data Science Institute) Prof. Agrawal, Ankit (Northwestern University) Gray, Lindsey (FNAL) Dr Klijnsma, Thomas (FNAL) Calafiura, Paolo (LBNL) Murnane, Daniel (Lawrence Berkeley National Laboratory) Choma, Nick (LBNL) Conlon, Sean (LBNL) Farrell, Steven (LBNL) Ju, Xiangyang (LBNL) Xu, Yaoyuan (LBNL) Spiropulu, Maria (Caltech) Vlimant, Jean-Roch (Caltech)

Presentation materials