mlsanityData to 14 Oct 2026

Generalizable Foundation Models for Calorimetry via Mixtures-of-Experts and Parameter Efficient Fine Tuning

arXiv 2603.28804 ↗First posted 27 Mar 2026physics.ins-det cs.LG hep-excited by 3

On the map: Mixture-of-experts → MoE theory and gating functions

Abstract (arXiv)

Modern particle physics experiments face an increasing demand for high-fidelity detector simulation as luminosities rise and computational requirements approach the limits of available resources. Deep generative models have emerged as promising surrogates for traditional Monte Carlo simulation, with recent advances drawing inspiration from large language models (LLM) and next-token prediction paradigms. In this work, we introduce a generalizable foundation model for calorimetry built on next-token transformer backbones, designed to support modular adaptation across materials, particle species, and detector configurations. Our approach combines Mixture-of-Experts pre-training with parameter-efficient fine-tuning strategies to enable controlled, additive model expansion without catastrophic forgetting. A pre-trained backbone is trained to generate electromagnetic showers across multiple absorber materials, while new materials are incorporated through the addition and tuning of lightweight expert modules. Extensions to new particle types are achieved via parameter-efficient fine-tuning and modular vocabularies, preserving the integrity of the base model. This design enables efficient, incremental knowledge integration as new simulation datasets become available, a critical requirement in realistic detector-development workflows. In addition, we demonstrate that next-token calorimeter models are computationally competitive with standard generative approaches under established LLM optimization procedures. These results establish next-token architectures as a viable path toward extensible, physics-aware foundation models for calorimetry and future high-energy physics experiments.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in MoE theory and gating functions, one of 12 directions in Mixture-of-experts. Builds on 35 of its 75 references that are in the map. First posted on arXiv 27 Mar 2026.

Builds on

35 of 75 references.

References · 35

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  3. 2020Scaling Laws for Neural Language Modelscited by 6,586
  4. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  6. 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,323
  7. 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,135
  8. 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
  9. 2021Scaling Laws for Transfercited by 272
  10. 2024Round and Round We Go! What makes Rotary Positional Encodings useful?cited by 115
  11. 2017CaloGAN: Simulating 3D High Energy Particle Showers in Multi-Layer Electromagnetic Calorimeters with Generative Adversarial Networkscited by 78
  12. 2021A Living Review of Machine Learning for Particle Physicscited by 78
23 more
  1. 2017Accelerating Science with Generative Adversarial Networks: An Application to 3D Particle Showers in Multi-Layer Calorimeterscited by 55
  2. 2022Score-based Generative Models for Calorimeter Shower Simulationcited by 52
  3. 2019Calorimetry with Deep Learning: Particle Simulation and Reconstruction for Collider Physicscited by 42
  4. 2024CaloChallenge 2022: A Community Challenge for Fast Calorimeter Simulationcited by 35
  5. 2021CaloFlow II: Even Faster and Still Accurate Generation of Calorimeter Showers with Normalizing Flowscited by 34
  6. 2023CaloClouds: Fast Geometry-Independent Highly-Granular Calorimeter Simulationcited by 33
  7. 2022CaloMan: Fast generation of calorimeter showers with density estimation on learned manifoldscited by 28
  8. 2025Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMscited by 27
  9. 2018Generative Models for Fast Calorimeter Simulation.LHCb casecited by 25
  10. 2023Inductive Simulation of Calorimeter Showers with Normalizing Flowscited by 22
  11. 2023CaloClouds II: Ultra-Fast Geometry-Independent Highly-Granular Calorimeter Simulationcited by 20
  12. 2022New directions for surrogate models and differentiable programming for High Energy Physics detector simulationcited by 16
  13. 2019AI-optimized detector design for the future Electron-Ion Collider: the dual-radiator RICH casecited by 15
  14. 2024Calo-VQ: Vector-Quantized Two-Stage Generative Model in Calorimeter Simulationcited by 15
  15. 2025OmniJet-αC: Learning point cloud calorimeter simulations using generative transformerscited by 14
  16. 2024Convolutional L2LFlows: Generating Accurate Showers in Highly Granular Calorimeters Using Convolutional Normalizing Flowscited by 13
  17. 2023CaloQVAE : Simulating high-energy particle-calorimeter interactions using hybrid quantum-classical generative modelscited by 11
  18. 2025CaloHadronic: a diffusion model for the generation of hadronic showerscited by 8
  19. 2022CaloDVAE : Discrete Variational Autoencoders for Fast Calorimeter Shower Simulationcited by 6
  20. 2025Towards Foundation Models for Experimental Readout Systems Combining Discrete and Continuous Datacited by 5
  21. 2024AI-Assisted Detector Design for the EIC (AID(2)E)cited by 3
  22. 2026AllShowers: One model for all calorimeter showerscited by 3
  23. 2025A First Full Physics Benchmark for Highly Granular Calorimeter Surrogatescited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026CaloTrilogy: Toward a Breakthrough in One-Step, End-to-End, Physics-Guided Shower Generation for Modern Calorimeters
  2. 2025Even Faster Simulations with Flow Matching: A Study of Zero Degree Calorimeter Responses
  3. 2025Inverse Autoregressive Flows for Zero Degree Calorimeter fast simulation
  4. 2026Lantern: Conflict-Aware Gradient Blending for Physics-Guided Diffusion Models in Calorimeter Simulation
  5. 2024Generative Diffusion Models for Fast Simulations of Particle Collisions at CERN
  6. 2026Panda Diplomacy: Foundation Model Pre-training across Particle Imaging Detectors for High Energy and Nuclear Physics
  7. 2025Generative Monte Carlo Sampling for Constant-Cost Particle Transport
  8. 2025Towards Overcoming Data Scarcity in Nuclear Energy: A Study on Critical Heat Flux with Physics-consistent Conditional Diffusion Model