Search arXivSearch

arXiv subjects

Adam Aurisano

Publications and source records attributed to Adam Aurisano.

15 recordsLinked to original sources

Submanifold Sparse Convolutional Networks for Automated 3D Segmentation of Kidneys and Kidney Tumours in Computed Tomography

Accurate delineation of kidney tumours in Computed Tomography (CT) is essential for downstream quantitative analysis and precision oncology, but manual segmentation is a specialised task, time-consuming and difficult to scale. Automated 3D segmentation remains challenging because CT scans are large volumetric images, making high-resolution dense convolutional networks computationally expensive and often dependent on downsampling or patch-based inference. We propose a two-stage 3D segmentation methodology based on voxel sparsification and submanifold sparse convolutional networks (SSCNs). Stage 1 uses a low-resolution sparse network to identify a region of interest (ROI); Stage 2 applies a high-resolution sparse network for refined segmentation within the cropped ROI. This enables native high-resolution 3D processing while reducing memory use and inference time. We evaluate the method on the KiTS23 renal cancer CT dataset using 5-fold cross-validation. Our method achieved Dice similarity coefficients of 95.8% for kidneys + masses, 85.7% for tumours + cysts, and 80.3% for tumours alone, competitive with top KiTS23 approaches. In direct comparisons on the same cross-validation folds, the proposed sparse method achieves tumour + cyst and tumour-only Dice scores comparable to, and slightly higher than, a patch-based nnU-Net baseline, while consistently requiring less VRAM and shorter inference time across the tested hardware. Across the tested GPUs, our sparse model is markedly faster than both nnU-Net and the zero-shot zoom-out/zoom-in foundation model SegVol, which localises kidneys well but underperforms on small heterogeneous lesions. Compared to an equivalent dense implementation of the same architecture, the proposed sparse approach achieves up to a 60% reduction in inference time and up to a 75% reduction in VRAM usage across both CPU and the GPU configurations tested.

cs.CV

Graph Neural Network for Neutrino Physics Event Reconstruction

Liquid Argon Time Projection Chamber (LArTPC) detector technology offers a wealth of high-resolution information on particle interactions, and leveraging that information to its full potential requires sophisticated automated reconstruction techniques. This article describes NuGraph2, a Graph Neural Network (GNN) for low-level reconstruction of simulated neutrino interactions in a LArTPC detector. Simulated neutrino interactions in the MicroBooNE detector geometry are described as heterogeneous graphs, with energy depositions on each detector plane forming nodes on planar subgraphs. The network utilizes a multi-head attention message-passing mechanism to perform background filtering and semantic labelling on these graph nodes, identifying those associated with the primary physics interaction with 98.0\% efficiency and labelling them according to particle type with 94.9\% efficiency. The network operates directly on detector observables across multiple 2D representations, but utilizes a 3D-context-aware mechanism to encourage consistency between these representations. Model inference takes 0.12~s/event on a CPU, and 0.005s/event batched on a GPU. This architecture is designed to be a general-purpose solution for particle reconstruction in neutrino physics, with the potential for deployment across a broad range of detector technologies, and offers a core convolution engine that can be leveraged for a variety of tasks beyond the two described in this article.

physics.data-an

Automated Segmentation of Computed Tomography Images with Submanifold Sparse Convolutional Networks

Quantitative cancer image analysis relies on the accurate delineation of tumours, a very specialised and time-consuming task. For this reason, methods for automated segmentation of tumours in medical imaging have been extensively developed in recent years, being Computed Tomography one of the most popular imaging modalities explored. However, the large amount of 3D voxels in a typical scan is prohibitive for the entire volume to be analysed at once in conventional hardware. To overcome this issue, the processes of downsampling and/or resampling are generally implemented when using traditional convolutional neural networks in medical imaging. In this paper, we propose a new methodology that introduces a process of sparsification of the input images and submanifold sparse convolutional networks as an alternative to downsampling. As a proof of concept, we applied this new methodology to Computed Tomography images of renal cancer patients, obtaining performances of segmentations of kidneys and tumours competitive with previous methods (~84.6% Dice similarity coefficient), while achieving a significant improvement in computation time (2-3 min per training epoch).

eess.IV

End-to-end analysis using image classification

End-to-end analyses of data from high-energy physics experiments using machine and deep learning techniques have emerged in recent years. These analyses use deep learning algorithms to go directly from low-level detector information directly to high-level quantities that classify the interactions. The most popular class of algorithms for these analyses are convolutional neural networks that operate on experimental data formatted as images. End-to-end analyses skip stages of the traditional workflow that includes the reconstruction of particles produced in the interactions, and as such are not limited by efficiency losses and sources of inaccuracy throughout the event reconstruction process. In many cases, deep learning end-to-end analyses have been shown to have significantly increased performance compared to previous state-of-the-art methods.

physics.data-an

Reconstruction of Large Radius Tracks with the Exa.TrkX pipeline

Particle tracking is a challenging pattern recognition task at the Large Hadron Collider (LHC) and the High Luminosity-LHC. Conventional algorithms, such as those based on the Kalman Filter, achieve excellent performance in reconstructing the prompt tracks from the collision points. However, they require dedicated configuration and additional computing time to efficiently reconstruct the large radius tracks created away from the collision points. We developed an end-to-end machine learning-based track finding algorithm for the HL-LHC, the Exa.TrkX pipeline. The pipeline is designed so as to be agnostic about global track positions. In this work, we study the performance of the Exa.TrkX pipeline for finding large radius tracks. Trained with all tracks in the event, the pipeline simultaneously reconstructs prompt tracks and large radius tracks with high efficiencies. This new capability offered by the Exa.TrkX pipeline may enable us to search for new physics in real time.

physics.ins-det

Tau Neutrinos in the Next Decade: from GeV to EeV

Tau neutrinos are the least studied particle in the Standard Model. This whitepaper discusses the current and expected upcoming status of tau neutrino physics with attention to the broad experimental and theoretical landscape spanning long-baseline, beam-dump, collider, and astrophysical experiments. This whitepaper was prepared as a part of the NuTau2021 Workshop.

hep-ph

Accelerating the Inference of the Exa.TrkX Pipeline

Recently, graph neural networks (GNNs) have been successfully used for a variety of particle reconstruction problems in high energy physics, including particle tracking. The Exa.TrkX pipeline based on GNNs demonstrated promising performance in reconstructing particle tracks in dense environments. It includes five discrete steps: data encoding, graph building, edge filtering, GNN, and track labeling. All steps were written in Python and run on both GPUs and CPUs. In this work, we accelerate the Python implementation of the pipeline through customized and commercial GPU-enabled software libraries, and develop a C++ implementation for inferencing the pipeline. The implementation features an improved, CUDA-enabled fixed-radius nearest neighbor search for graph building and a weakly connected component graph algorithm for track labeling. GNNs and other trained deep learning models are converted to ONNX and inferenced via the ONNX Runtime C++ API. The complete C++ implementation of the pipeline allows integration with existing tracking software. We report the memory usage and average event latency tracking performance of our implementation applied to the TrackML benchmark dataset.

physics.ins-det

Performance of a Geometric Deep Learning Pipeline for HL-LHC Particle Tracking

The Exa.TrkX project has applied geometric learning concepts such as metric learning and graph neural networks to HEP particle tracking. Exa.TrkX's tracking pipeline groups detector measurements to form track candidates and filters them. The pipeline, originally developed using the TrackML dataset (a simulation of an LHC-inspired tracking detector), has been demonstrated on other detectors, including DUNE Liquid Argon TPC and CMS High-Granularity Calorimeter. This paper documents new developments needed to study the physics and computing performance of the Exa.TrkX pipeline on the full TrackML dataset, a first step towards validating the pipeline using ATLAS and CMS data. The pipeline achieves tracking efficiency and purity similar to production tracking algorithms. Crucially for future HEP applications, the pipeline benefits significantly from GPU acceleration, and its computational requirements scale close to linearly with the number of particles in the event.

physics.data-an

Graph Neural Network for Object Reconstruction in Liquid Argon Time Projection Chambers

This paper presents a graph neural network (GNN) technique for low-level reconstruction of neutrino interactions in a Liquid Argon Time Projection Chamber (LArTPC). GNNs are still a relatively novel technique, and have shown great promise for similar reconstruction tasks in the LHC. In this paper, a multihead attention message passing network is used to classify the relationship between detector hits by labelling graph edges, determining whether hits were produced by the same underlying particle, and if so, the particle type. The trained model is 84% accurate overall, and performs best on the EM shower and muon track classes. The model's strengths and weaknesses are discussed, and plans for developing this technique further are summarised.

hep-ex

Track Seeding and Labelling with Embedded-space Graph Neural Networks

To address the unprecedented scale of HL-LHC data, the Exa.TrkX project is investigating a variety of machine learning approaches to particle track reconstruction. The most promising of these solutions, graph neural networks (GNN), process the event as a graph that connects track measurements (detector hits corresponding to nodes) with candidate line segments between the hits (corresponding to edges). Detector information can be associated with nodes and edges, enabling a GNN to propagate the embedded parameters around the graph and predict node-, edge- and graph-level observables. Previously, message-passing GNNs have shown success in predicting doublet likelihood, and we here report updates on the state-of-the-art architectures for this task. In addition, the Exa.TrkX project has investigated innovations in both graph construction, and embedded representations, in an effort to achieve fully learned end-to-end track finding. Hence, we present a suite of extensions to the original model, with encouraging results for hitgraph classification. In addition, we explore increased performance by constructing graphs from learned representations which contain non-linear metric structure, allowing for efficient clustering and neighborhood queries of data points. We demonstrate how this framework fits in with both traditional clustering pipelines, and GNN approaches. The embedded graphs feed into high-accuracy doublet and triplet classifiers, or can be used as an end-to-end track classifier by clustering in an embedded space. A set of post-processing methods improve performance with knowledge of the detector physics. Finally, we present numerical results on the TrackML particle tracking challenge dataset, where our framework shows favorable results in both seeding and track finding.

physics.ins-det

Graph Neural Networks for Particle Reconstruction in High Energy Physics detectors

Pattern recognition problems in high energy physics are notably different from traditional machine learning applications in computer vision. Reconstruction algorithms identify and measure the kinematic properties of particles produced in high energy collisions and recorded with complex detector systems. Two critical applications are the reconstruction of charged particle trajectories in tracking detectors and the reconstruction of particle showers in calorimeters. These two problems have unique challenges and characteristics, but both have high dimensionality, high degree of sparsity, and complex geometric layouts. Graph Neural Networks (GNNs) are a relatively new class of deep learning architectures which can deal with such data effectively, allowing scientists to incorporate domain knowledge in a graph structure and learn powerful representations leveraging that structure to identify patterns of interest. In this work we demonstrate the applicability of GNNs to these two diverse particle reconstruction problems.

physics.ins-det

Machine Learning in High Energy Physics Community White Paper

Machine learning has been applied to several problems in particle physics research, beginning with applications to high-level physics analysis in the 1990s and 2000s, followed by an explosion of applications in particle and event identification and reconstruction in the 2010s. In this document we discuss promising future research and development areas for machine learning in particle physics. We detail a roadmap for their implementation, software and hardware resource requirements, collaborative initiatives with the data science community, academia and industry, and training the particle physics community in data science. The main objective of the document is to connect and motivate these areas of research and development with the physics drivers of the High-Luminosity Large Hadron Collider and future neutrino experiments and identify the resource needs for their implementation. Additionally we identify areas where collaboration with external communities will be of great benefit.

physics.comp-ph

Sterile neutrino search in the NOvA Far Detector

The majority of neutrino oscillation experiments have obtained evidence for neutrino oscillations that are compatible with the three-flavor model. Explaining anomalous results from short-baseline experiments, such as LSND and MiniBooNE, in terms of neutrino oscillations requires the existence of sterile neutrinos. The search for sterile neutrino mixing conducted in NOvA uses a long baseline of 810 km between Near Detector (ND) at Fermilab and Far Detector (FD) in Minnesota. The signal for sterile neutrino oscillations is a deficit of neutral-current neutrino interactions at the FD with respect to the ND prediction. In this document, We will present the analysis improvements that we are implementing for future NC sterile neutrino searches with NOvA. These include: improved modelling of our detector response; the inclusion of NC 2p2h interaction modelling; implementing a better energy reconstruction techniques; and including possible oscillation due to sterile neutrinos in the ND . This improvements enable us to do a simultaneous ND-FD shape fit of the NC energy spectrum covering a wider sterile mass range than previous analyses.

hep-ex

NOvA Short-Baseline Tau Neutrino Appearance Search

Standard three-flavor neutrino oscillations have well explained by a wide range of neutrino experiments. However, the anomalous results, such as electron-antineutrino excess seen by LSND and MiniBooNE do not fit the three-flavor paradigm. This can be explained by an additional fourth flavor sterile neutrino at a larger scale than the existing three flavor neutrinos. The NOvA experiment consists of two finely segmented, liquid scintillator detectors operating 14 .6 mrad off-axis from the NuMI muon-neutrino beam. The Near Detector is located on the Fermilab campus, 1 km from the NuMI target, while the Far Detector is located at Ash River, MN, 810 km from the NuMI target. The NOvA experiment is primarily designed to measure electron-neutrino appearance at the Far Detector using the Near Detector to control systematic uncertainties; however, the Near Detector is well suited for searching for anomalous short-baseline oscillations. This poster will present a novel method for selecting tau neutrino interactions with high purity at the Near Detector using a convolutional neural network. Using this method, the sensitivity to anomalous short-baseline tau-neutrino appearance due to sterile neutrino oscillations will be presented.

hep-ex

Cosmology and Dark Matter at the LHC

We examine the question of whether neutralinos produced at the LHC can be shown to be the particles making up the astronomically observed dark matter. If the WIMP alllowed region lies in the SUGRA coannihilation region, then a strong signal for this would be the unexpected near degeneracy of the stau and neutralino i.e., a mass difference \Delta M\simeq (5-15) GeV. For the mSUGRA model we show such a small mass difference can be measured at the LHC using the signal 3\tau+jet+E_T^{\rm miss}. Two observables, opposite sign minus like sign pairs and the peak of the \tau\tau mass distribution allows the simultaneous determination of \Delta M to 15% and the gluino mass M_{\tilde g} to be 6% at the benchmark point of M_{\tilde g}=850 GeV, A_0=0, \mu>0 with 30 fb^{-1}. With 10 fb^{-1}, \Delta M can be determined to 22% and one can probe the parameter space up to m_{1/2}=700 GeV with 100 fb^{-1}.

hep-ph