SPADE: Split-and-Delay Embeddings for Autoregressive High-Granularity Calorimeter Simulation

We introduce SPADE (SPlit And Delay Embeddings), an autoregressive transformer for sequences whose tokens carry multiple features. Rather than embedding these features jointly, SPADE embeds them independently. Delaying each feature stream relative to the previous one allows intra-token correlations to be learned by the standard self-attention mechanism. Applied to point-cloud calorimeter shower generation in the highly granular ILD detector, SPADE is competitive with the state of the art AllShowers model on photon showers, and substantially outperforms its VQ-VAE-based predecessor OmniJet-$\alpha_C$. The mechanism is applicable to any generative task with multi-feature tokens, enabling LLM-style pretraining workflows for higher-dimensional data.

Paper

References (32)

07Fast transformer decoding: One writer, en heads ( Preprint 1911.02150 )2019
09JINST 13 C03015 ( Preprint 1801.02024 )2018
10Neural discrete representation learning ( Preprint 1711.00937 )2018
11Layer normalization ( Preprint 1607.06450 )2016
12(GEANT4)2003 · Nucl. Instrum. Meth.

Scroll for more · 20 remaining

Similar papers

© 2026 NYSGPT2525 LLC