Variational Autoencoders for Anomalous Jet Tagging

Abstract

We present a detailed study on Variational Autoencoders (VAEs) for anomalous jet tagging at the Large Hadron Collider. By taking in low-level jet constituents' information, and training with background QCD jets in an unsupervised manner, the VAE is able to encode important information for reconstructing jets, while learning an expressive posterior distribution in the latent space. When using the VAE as an anomaly detector, we present different approaches to detect anomalies: directly comparing in the input space or, instead, working in the latent space. In order to facilitate general search approaches such as bump-hunt, mass-decorrelated VAEs based on distance correlation regularization are also studied. We find that the naive mass-decorrelated VAEs fail at maintaining proper detection performance, by assigning higher probabilities to some anomalous samples. To build a performant mass-decorrelated anomalous jet tagger, we propose the Outlier Exposed VAE (OE-VAE), for which some outlier samples are introduced in the training process to guide the learned information. OE-VAEs are employed to achieve two goals at the same time: increasing sensitivity of outlier detection and decorrelating jet mass from the anomaly score. We succeed in reaching excellent results from both aspects. Code implementation of this work can be found at https://github.com/taolicheng/VAE-Jet

36 pages, 23 figures. To appear in Physical Review D

References in corpus (3)

• Measuring and testing dependence by correlation of distances
• Extending the Bump Hunt with Machine Learning
• A generic anti-QCD jet tagger

Cited by in corpus (29)

• Quantum Computing for High-Energy Physics: State of the Art and Challenges. Summary of the QC4HEP Working Group
• Classifying Anomalies THrough Outer Density Estimation (CATHODE)
• Bump Hunting in Latent Space
• Machine Learning for Anomaly Detection in Particle Physics
• ABCDisCo: Automating the ABCD Method with Machine Learning
• Challenges for Unsupervised Anomaly Detection in Particle Physics
• Quasi Anomalous Knowledge: Searching for new physics with embedded knowledge
• Quantum anomaly detection in the latent space of proton collision events at the LHC
• Linearized Optimal Transport for Collider Events
• Anomaly detection search for new resonances decaying into a Higgs boson and a generic new particle $X$ in hadronic final states using $\sqrt{s}=13$ TeV $pp$ collisions with the ATLAS detector
• Deep Generative Models for Detector Signature Simulation: A Taxonomic Review
• What's Anomalous in LHC Jets?
• Nanosecond anomaly detection with decision trees and real-time application to exotic Higgs decays
• Anomalies, Representations, and Self-Supervision
• Anomaly Awareness
• GAN-AE : An anomaly detection algorithm for New Physics search in LHC data
• Enhancing the hunt for new phenomena in dijet final-states using anomaly detection filters at the High-Luminosity Large Hadron Collider
• Searching for dark jets with displaced vertices using weakly supervised machine learning
• Semi-visible jets, energy-based models, and self-supervision
• Differentiable Earth Mover's Distance for Data Compression at the High-Luminosity LHC
• Non-resonant Anomaly Detection with Background Extrapolation
• High-dimensional Anomaly Detection with Radiative Return in $e^{+}{e}^{-}$ Collisions
• Unsupervised and lightly supervised learning in particle physics
• Simulation-based Anomaly Detection for Multileptons at the LHC
• Triggering Dark Showers with Conditional Dual Auto-Encoders
• Semi-supervised permutation invariant particle-level anomaly detection
• Tensor Network for Anomaly Detection in the Latent Space of Proton Collision Events at the LHC
• Graph theory inspired anomaly detection at the LHC
• Reweighting and Analysing Event Generator Systematics by Neural Networks on High-Level Features