Fast Jet Tagging with MLP-Mixers on FPGAs

Abstract

We explore the innovative use of MLP-Mixer models for real-time jet tagging and establish their feasibility on resource-constrained hardware like FPGAs. MLP-Mixers excel in processing sequences of jet constituents, achieving state-of-the-art performance on datasets mimicking Large Hadron Collider conditions. By using advanced optimization techniques such as High-Granularity Quantization and Distributed Arithmetic, we achieve unprecedented efficiency. These models match or surpass the accuracy of previous architectures, reduce hardware resource usage by up to 97%, double the throughput, and half the latency. Additionally, non-permutation-invariant architectures enable smart feature prioritization and efficient FPGA deployment, setting a new benchmark for machine learning in real-time data processing at particle colliders.

References in corpus (14)

• The anti-k_t jet clustering algorithm
• MLP-Mixer: An all-MLP Architecture for Vision
• ParticleNet: Jet Tagging via Particle Clouds
• Automatic heterogeneous quantization of deep neural networks for low-latency inference on the edge for particle detectors
• JEDI-net: a jet identification algorithm based on interaction networks
• Particle Transformer for Jet Tagging
• Explainable Equivariant Neural Networks for Particle Physics: PELICAN
• LL-GNN: Low Latency Graph Neural Networks on FPGAs for High Energy Physics
• The importance of calorimetry for highly-boosted jet substructure
• Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics
• Fast Muon Tracking with Machine Learning Implemented in FPGA
• Ultrafast jet classification on FPGAs for the HL-LHC
• HGQ: High Granularity Quantization for Real-time Neural Networks on FPGAs
• LLPNet: Graph Autoencoder for Triggering Light Long-Lived Particles at HL-LHC

Cited by in corpus (1)

• da4ml: Distributed Arithmetic for Real-time Neural Networks on FPGAs