mlsanityData to 7 Oct 2026

Bayesian Repetition Penalty: A Principled Adjacent-Conditional Framework for Reversing Attention Collapse in Autoregressive Language Models

arXiv 2607.22694 ↗First posted 17 Jul 2026cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Attention collapse in autoregressive language models -- manifested as repetitive token loops where the model becomes trapped in self-reinforcing attractors -- is a persistent pathology that existing decoding-time heuristics fail to address at its root cause. We present a principled framework that penalises or compensates anomalous confidence arising from collapsed generation patterns, by comparing a token's observed frequency against its corpus prior through an adjacent-conditional probability construction. The resulting self-normalising penalty ratio R=f(m,n,p)/f(np,n,p) requires no ad hoc standardisation and admits a closed-form logit offset with zero approximation error. The correction is isolated from the loss gradient and accumulated into a frozen output-layer bias via exponential moving average, enabling deployment as a repair mechanism for models that have already collapsed without requiring intrusive modifications to standard training pipelines. Experimental validation on a 1.5B-parameter model demonstrates that the frozen-bias mechanism can rescue a model already trapped in a collapsed attractor, reducing 2-gram repetition from 0.073 to near 0 while preserving generation quality.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 4 of its 4 references that are in the map. First posted on arXiv 17 Jul 2026.

Builds on

4 of 4 references.

References · 4

  1. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  2. 2019CTRL: A Conditional Transformer Language Model for Controllable Generationcited by 754
  3. 2019Neural Text Generation with Unlikelihood Trainingcited by 489
  4. 2017A Deep Reinforced Model for Abstractive Summarizationcited by 364

Built on by

Not cited yet.