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.
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