cs.LGOct 29, 2026

Where Position Matters: Profiling-Guided Layer-Wise RoPE Allocation in Transformers

Authors: Zhiwei Liu, Boao Kong, Bing Liu

Abstract

Rotary Position Embedding (RoPE) typically uses uniform coverage across Transformer layers despite heterogeneous positional requirements. Phase-off interventions reveal non-uniform, model-specific rotary-phase dependence, while coverage interventions relate it to coverage-upgrade benefits. We propose profiling-guided layer-wise RoPE allocation, assigning high/low coverage under a fixed budget using early coverage-benefit profiles. The framework supports fixed-width query/key representations and physical rotary-dimension allocation in Multi-head Latent Attention (MLA). Normalized-depth mapping transfers profiles within model families, enabling small-model measurements to guide larger-model allocations before pretraining. A brief analysis relates selection and transfer to score calibration and inter-layer interactions. From-scratch experiments demonstrate validation-loss improvements over available budget-matched baselines, while analytical resource accounting identifies favorable quality--storage--compute trade-offs for physical MLA. These results connect positional demand to transferable, efficient architecture design.

Figures & tables

Appendix figures & tables3 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Fractional Rotation, Full Potential? Investigating Performance and Convergence of Partial RoPE

    Mar 12, 2026Mohammad Aflah Khan, Krishna P. Gummadi, Manish Gupta +1Rotary Positional EmbeddingsPositional Encoding

  2. Efficient Matrix Implementation for Rotary Position Embedding

    Apr 10, 2026Chen Minqi, Zhongqi Yue, Shihao Zhang +5Rotary Positional EmbeddingsEfficient Transformer Inference

  3. AdaRoPE: Not All Attention Heads Should Rotate and Scale Equally

    Jun 5, 2026Shaowen Wang, Yuke Zheng, Tansheng Zhu +4Rotary Positional EmbeddingsSelf-Attention