Investigating Resource-efficient Neutron/Gamma Classification ML Models Targeting eFPGAs

Abstract

There has been considerable interest and resulting progress in implementing machine learning (ML) models in hardware over the last several years from the particle and nuclear physics communities. A big driver has been the release of the Python package, hls4ml, which has enabled porting models specified and trained using Python ML libraries to register transfer level (RTL) code. So far, the primary end targets have been commercial FPGAs or synthesized custom blocks on ASICs. However, recent developments in open-source embedded FPGA (eFPGA) frameworks now provide an alternate, more flexible pathway for implementing ML models in hardware. These customized eFPGA fabrics can be integrated as part of an overall chip design. In general, the decision between a fully custom, eFPGA, or commercial FPGA ML implementation will depend on the details of the end-use application. In this work, we explored the parameter space for eFPGA implementations of fully-connected neural network (fcNN) and boosted decision tree (BDT) models using the task of neutron/gamma classification with a specific focus on resource efficiency. We used data collected using an AmBe sealed source incident on Stilbene, which was optically coupled to an OnSemi J-series SiPM to generate training and test data for this study. We investigated relevant input features and the effects of bit-resolution and sampling rate as well as trade-offs in hyperparameters for both ML architectures while tracking total resource usage. The performance metric used to track model performance was the calculated neutron efficiency at a gamma leakage of 10${}^{-3}$. The results of the study will be used to aid the specification of an eFPGA fabric, which will be integrated as part of a test chip.

References in corpus (11)

• XGBoost: A Scalable Tree Boosting System
• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems
• Fast inference of deep neural networks in FPGAs for particle physics
• Automatic heterogeneous quantization of deep neural networks for low-latency inference on the edge for particle detectors
• Fast inference of Boosted Decision Trees in FPGAs for particle physics
• A reconfigurable neural network ASIC for detector front-end data compression at the HL-LHC
• Neutron-$\gamma$ discrimination with organic scintillators: Intrinsic pulse shape and light yield contributions
• Design and Calibration of an Optically Segmented Single Volume Scatter Camera for Neutron Imaging
• Embedded FPGA Developments in 130nm and 28nm CMOS for Machine Learning in Particle Detector Readout
• Studies in Pulse Shape Discrimination for an Optimized ASIC Design

Cited by in corpus (1)

• Variational Autoencoders for At-Source Data Reduction and Anomaly Detection in High Energy Particle Detectors