arXiv AI
Sep 3

Language Models Can Control Their Own Attention

The paper introduces Declarative Attention (DA), a protocol that lets language models explicitly declare which parts of their context to focus on during generation. By partitioning decoding into full-context, region-specific, and recent-output-only modes, the inference engine can skip large portions of the KV cache, dramatically reducing attended tokens. Experiments on 15 long-context tasks with off-the-shelf models show significant savings (52.0% and 31.1% reductions) with only modest accuracy drops that diminish as model size increases.

By Namgyu Ho, Huzama Ahmad, Woosung Koh, Se-Young Yun, Tal Schuster, Cicero Nogueira dos Santos
arXiv Computation and Language
Oct 21

Are Linguistic and Arithmetic Circuits Dissociated in LLMs?

The paper investigates whether linguistic and arithmetic tasks in large language models (LLMs) rely on shared neural circuits. Using mechanistic interpretability techniques on five LLMs, the authors find that circuit similarity aligns primarily with task domain rather than input/output vocabularies or syntactic structure. This suggests that LLMs may employ largely separate neural circuits for language and arithmetic, indicating a modular organization of abilities.

By Riku Kisako, Tatsuki Kuribayashi, Ryohei Sasano