mlsanityData to 7 Oct 2026

Retrieval and competition: how a protein foundation model starts a protein

arXiv 2605.16331 ↗First posted 5 May 2026q-bio.BM cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Protein language models are increasingly used to guide experimental and clinical decisions, yet it is often unclear whether a confident prediction reflects recognition of biological evidence or retrieval of a statistical default. We examine this distinction for a near-universal biological rule, that proteins begin with methionine, by tracing the computational pathway through which ESM2-8M produces this prediction. The model does not detect methionine at the masked position. Instead, it retrieves a methionine-favouring signal from a reference representation at the beginning-of-sequence token via a position-specific query assembled across layers, with the final output emerging through competition with context-dependent circuits. To understand how positional information reaches the readout, we introduce a norm-direction decomposition of attention scores within rotary frequency bands. Positional encoding operates through coupled changes in query norm and angular alignment distributed across these bands. On sequences whose true N-terminus is not methionine, where the biological question matters, the model predicts methionine anyway. This is not a correct prediction produced by an unexpected mechanism, but the output of a positional-prior retrieval circuit that matches the statistical average and fails where biology diverges from it. Distinguishing the two requires resolution at the level of individual circuits, frequency bands, and query composition, suggesting that mechanistic verification will be necessary, and challenging, for predictions where the biological stakes are higher. Even for the simplest biological rule, the model's prediction is mediated by a distributed computational circuit rather than direct recognition, suggesting that increasing task complexity will further obscure the relationship between model confidence and underlying biological evidence.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 10 of its 24 references that are in the map. First posted on arXiv 5 May 2026.

Builds on

10 of 24 references.

References · 10

  1. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  2. 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
  3. 2025Why do LLMs attend to the first token?cited by 122
  4. 2020BERTology Meets Biology: Interpreting Attention in Protein Language Modelscited by 85
  5. 2024InterPLM: Discovering Interpretable Features in Protein Language Models via Sparse Autoencoderscited by 64
  6. 2025What are you sinking? A geometric approach on attention sinkcited by 26
  7. 2025Interpreting and Steering Protein Language Models through Sparse Autoencoderscited by 16
  8. 2025Towards Interpretable Protein Structure Prediction with Sparse Autoencoderscited by 12
  9. 2026Protein Circuit Tracing via Cross-layer Transcoderscited by 3
  10. 2026Two Stages of Folding: Convergent Mechanisms in AI Protein Folding Trunkscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.