Nakyung Lee

Mitigating Attention Localization in Small Scale: Self-Attention Refinement via One-step Belief Propagation

Transformer-based self-attention mechanism serves as the core of modern language models, yet it often suffers from *localization*, where attentions collapse onto a limited subset of tokens and fail to capture long-range dependencies. To address this issue, we propose **Self-Attention One-step Belief