Search papers, labs, and topics across Lattice.
This paper investigates how Rotary Position Embeddings (RoPE) frequencies are utilized by transformers, revealing that their non-uniform usage is driven by the relative-distance structure of training data. By formalizing a frequency-matching principle, the authors demonstrate that optimal frequency scales inversely with the width of data-induced dependencies, thereby explaining the prevalence of mid-low frequencies in language models. The findings also link frequency selection to length generalization, highlighting that effective scaling can enhance model performance on longer-context dependencies, contingent on their alignment with training data structures.
Frequency usage in transformers is not random; it鈥檚 intricately tied to the data鈥檚 dependency structure, revealing a data-driven mechanism behind RoPE's emergent behavior.
Rotary Position Embeddings (RoPE) provide transformers with a fixed grid of positional frequencies, yet trained models use these frequencies highly non-uniformly. We study what determines this frequency usage and propose a data-centered explanation: RoPE frequencies are selected to match the relative-distance structure of the training data. Viewing each frequency as a positional lens, we formalize a field-resolution tradeoff and show that, for a data-induced dependency profile of width $W$, the optimal frequency scales as $1/W$. This frequency-matching principle explains controlled observations on synthetic and text-based data, and suggests that the mid-low frequency bands observed in language models arise from the multi-scale dependency structure of natural language. We further connect frequency selection to position-interpolation-based length generalization: scaling frequencies down expands the effective field while reducing resolution. This helps when longer-context dependencies are approximate dilations of those seen during training, but can fail when relevant dependencies do not scale with context length. Empirically, we show that natural language exhibits approximate self-similarity across positional scales, explaining why test-time frequency scaling can support long-context generalization. Overall, our results identify a data-driven mechanism behind emergent RoPE frequency usage and show that long-context generalization depends on two forms of scale matching: between learned frequencies and training-time dependencies, and between frequency scaling and how those dependencies extend to longer contexts.