Probabilistic Energy-Based Neural Networks for Uncertainty-Aware Collider Data Analyses
Raymundo Ramos, Ahmed Hammad, Amit Chakraborty +2 more
We introduce a probabilistic energy-based Neural Network (NN) for uncertainty-aware classification in collider analyses. Standard discriminative classifiers can give highly confident predictions even for events that are different from the data used during training. Energy-Based Models provide an alternative by assigning an energy score to each event, which measures how well the event agrees with the learned data distribution. In this framework, a NN maps the input features into a latent space, where the energy is defined with respect to class-specific embeddings. This allows the model to identify noisy, anomalous or shifted events while maintaining good classification performance. We first study the method using controlled synthetic examples and then test it in a realistic collider analysis based on the $h \to τ^+ τ^-$ process at the Large Hadron Collider, where $h$ is the discovered…