English

What exactly did the Transformer learn from our physics data?

Instrumentation and Methods for Astrophysics 2026-04-14 v1 High Energy Physics - Experiment

Abstract

Transformer networks excel in scientific applications. We explore two scenarios in ultra-high-energy cosmic ray simulations to examine what these network architectures learn. First, we investigate the trained positional encodings in air showers which are azimuthally symmetric. Second, we visualize the attention values assigned to cosmic particles originating from a galaxy catalog. In both cases, the Transformers learn plausible, physically meaningful features.

Keywords

Cite

@article{arxiv.2505.21042,
  title  = {What exactly did the Transformer learn from our physics data?},
  author = {Martin Erdmann and Niklas Langner and Josina Schulte and Dominik Wirtz},
  journal= {arXiv preprint arXiv:2505.21042},
  year   = {2026}
}

Comments

9 pages, 7 figures

R2 v1 2026-07-01T02:42:34.599Z