MMODELYST
Papers/Remember to Forget: Gated Adaptive Positional Encoding
PAP

Remember to Forget: Gated Adaptive Positional Encoding

May 11, 2026

arXiv ↗
Abstract

Rotary Positional Encoding (RoPE) is widely used in modern large language models. However, when sequences are extended beyond the range seen during training, rotary phases can enter out-of-distribution regimes, leading to spurious long-range alignments, diffuse attention, and degraded retrieval. Existing remedies only partially address these failures, as they often trade local positional resolution for long-context stability. We propose GAPE (Gated Adaptive Positional Encoding), a drop-in augmentation for positional encodings that introduces a content-aware bias directly into the attention logits while preserving the rotary geometry. GAPE decouples distance-based suppression from token importance through a query-dependent gate that contracts irrelevant context and a key-dependent gate that preserves salient distant tokens. We prove that protected tokens remain accessible, while the attention mass assigned to unprotected distant tokens decays as a function of the query gate. We further show that GAPE can be implemented within standard scaled dot-product attention. We validate these properties empirically, finding that GAPE consistently yields sharper attention and improved long-context robustness over rotary baselines across both synthetic retrieval and long-context benchmarks.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Riccardo Ali, Alessio Borgi, Christopher Irwin, Mario Severino, Pietro Liò
Your notes (browser-local)
saved
arXiv:2605.10414