Position-Agnostic Pre-Projection for Transformer Attention: Nonlinear Feature Construction and Content Skip Before Q/K/V
We propose a simple architectural modification to transformer attention blocks: inserting a small nonlinear MLP between the layer norm and Q/K/V projections. This pre-projection operates in a position-agnostic manner—it constructs richer feature representations from token content alone, before any positional encoding (e.g., RoPE) is applied. In frozen-probe experiments on Pythia-160M and Pythia-410M, training only the pre-projection parameters while keeping the base model frozen yields consistent improvements across all benchmarks. At 160M scale, the pre-projection outperforms LoRA on LAMBADA (0.154 vs. 0.126) despite using fewer parameters. Combining pre-projection with a small LoRA achieves the best results overall—matching a 10× larger standalone LoRA on perplexity while preserving comprehensiongains. The pre-projection adds no K/V cache overhead and its position-agnostic design makes it naturally suited to multimodal architectures where different modalities have fundamentally different notions of position.
Authors
- Shinde Chirag (ORCID: https://orcid.org/0009-0006-1926-3793)
Publication Details
- Journal
- Zenodo (CERN European Organization for Nuclear Research)
- Published
- 2026-04-10
- DOI
- https://doi.org/10.5281/zenodo.19498159
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.00