Query-Key Normalization for Transformers
Not on the map. Why?
Cited by 40 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Low-resource language translation is a challenging but socially valuable NLP task. Building on recent work adapting the Transformer's normalization to this setting, we propose QKNorm, a normalization technique that modifies the attention mechanism to make the softmax function less prone to arbitrary saturation without sacrificing expressivity. Specifically, we apply ℓ2 normalization along the head dimension of each query and key matrix prior to multiplying them and then scale up by a learnable parameter instead of dividing by the square root of the embedding dimension. We show improvements averaging 0.928 BLEU over state-of-the-art bilingual benchmarks for 5 low-resource translation pairs from the TED Talks corpus and IWSLT'15.
Key facts. Cited by 266 papers in the map (data through 7 Oct 2026). First posted on arXiv 8 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 1; 2022: 5; 2023: 11; 2024: 39; 2025: 78; 2026: 132.
266 papers.
Sort:
By area:
- 2025VGGT: Visual Geometry Grounded Transformercited by 1,475
- 2023BloombergGPT: A Large Language Model for Financecited by 923
- 2024Genie: Generative Interactive Environmentscited by 706
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2024RDT-1B: a Diffusion Foundation Model for Bimanual Manipulationcited by 670
- 2024Unified Training of Universal Time Series Forecasting Transformerscited by 588
- 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Modelscited by 446
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 294
- 2022Memorizing Transformerscited by 219
- 2026Gemma 4 Technical Reportcited by 187
- 2024LVSM: A Large View Synthesis Model with Minimal 3D Inductive Biascited by 178
- 2024Lumina-mGPT: Illuminate Flexible Photorealistic Text-to-Image Generation with Multimodal Generative Pretrainingcited by 148
38 more
- 2024Lumina-T2X: Transforming Text into Any Modality, Resolution, and Duration via Flow-based Large Diffusion Transformerscited by 137
- 2023Focused Transformer: Contrastive Training for Context Scalingcited by 133
- 2024GameGen-X: Interactive Open-world Game Video Generationcited by 124
- 2025Test-Time Training Done Rightcited by 121
- 2024Scalable High-Resolution Pixel-Space Image Synthesis with Hourglass Diffusion Transformerscited by 116
- 2025GR-3 Technical Reportcited by 110
- 2022Block-Recurrent Transformerscited by 107
- 2026Generative Modeling via Driftingcited by 98
- 2025Latent Diffusion Model without Variational Autoencodercited by 83
- 2025MotionStreamer: Streaming Motion Generation via Diffusion-based Autoregressive Model in Causal Latent Spacecited by 82
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 76
- 2025RayZer: A Self-supervised Large View Synthesis Modelcited by 74
- 2025LongCat-Video Technical Reportcited by 73
- 2024nGPT: Normalized Transformer with Representation Learning on the Hyperspherecited by 68
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPTcited by 66
- 2024A Survey on LLM-based Code Generation for Low-Resource and Domain-Specific Programming Languagescited by 65
- 2025Scalable Vision-Language-Action Model Pretraining for Robotic Manipulation with Real-Life Human Activity Videoscited by 63
- 2023TransNeXt: Robust Foveal Visual Perception for Vision Transformerscited by 61
- 2025FLOWER: Democratizing Generalist Robot Policies with Efficient Vision-Language-Action Flow Policiescited by 56
- 2024Normalization and effective learning rates in reinforcement learningcited by 55
- 2024Megalodon: Efficient LLM Pretraining and Inference with Unlimited Context Lengthcited by 46
- 2025Rope to Nope and Back Again: A New Hybrid Attention Strategycited by 46
- 2024Exploring the Role of Large Language Models in Prompt Encoding for Diffusion Modelscited by 45
- 2026Parcae: Scaling Laws For Stable Looped Language Modelscited by 45
- 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 43
- 2022How to Understand Masked Autoencoderscited by 37
- 2025DeepVerse: 4D Autoregressive Video Generation as a World Modelcited by 36
- 2023MosaicBERT: A Bidirectional Encoder Optimized for Fast Pretrainingcited by 33
- 2024SpaceByte: Towards Deleting Tokenization from Large Language Modelingcited by 31
- 2024MVGenMaster: Scaling Multi-View Generation from Any Image via 3D Priors Enhanced Diffusion Modelcited by 31
- 2024EzAudio: Enhancing Text-to-Audio Generation with Efficient Diffusion Transformercited by 30
- 2026GLM-5V-Turbo: Toward a Native Foundation Model for Multimodal Agentscited by 30
- 2021Attention Approximates Sparse Distributed Memorycited by 29
- 2026Controlled LLM Training on Spectral Spherecited by 28
- 2024SnapGen: Taming High-Resolution Text-to-Image Models for Mobile Devices with Efficient Architectures and Trainingcited by 27
- 2024Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Surveycited by 27
- 2025Absolute Coordinates Make Motion Generation Easycited by 27
- 2026ELF: Embedded Language Flowscited by 27
50 most cited of 266 All 266 (JSON)
- 2026QuadTok: Quadtree Visual Tokenizer for Autoregressive Image Generation7 Oct 2026
- 2026Latent-Foresight: End-to-End Learning Predictable Representations for Latent World Models1 Oct 2026
- 2026ScaffoldM3C: A Multimodal Sequential Monte Carlo Framework for Generative Stable Construction Planning30 Sep 2026
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026NowcastDiT: Diffusion Transformers are Effective Precipitation Nowcasters29 Sep 2026
- 2026One-Step Next-Latent Prediction Is Not a World Model28 Sep 2026
- 2026KiT: A Foundation Model for Financial Time-Series Forecasting using DiffusionTransformers28 Sep 2026
- 2026Query Expansion and Key Specialization in Transformer Attention Geometry28 Sep 2026
- 2026Broken Symmetry in BF16 Attention: Why FlashAttention Gradients Blow Up Late in Training28 Sep 2026
- 2026Natural Image Autoencoder-Based fMRI Representations for Trait and State Prediction28 Sep 2026
- 2026t0: A Time-Series Foundation Model for Forecasting with Context21 Sep 2026
- 2026ufakzeka-1: Building and Evaluating a 151M-Parameter Turkish Language Model from Scratch18 Sep 2026
38 more
- 2026Directions That Don't Drift: Stiefel Manifold Routing for Transformer Attention16 Sep 2026
- 2026Tele360: Real-Time Feed-Forward Human Reconstruction from Sparse Unposed Cameras14 Sep 2026
- 2026Depth and Scale in the Sub-150M Regime: JugnuLM-53M vs JugnuLM-110M13 Sep 2026
- 2026Learning Length-Extrapolatable Recurrent Models8 Sep 2026
- 2026Conditioned Initialization for Attention7 Sep 2026
- 2026DriveZero: End-to-End Driving Beyond Human Demonstrations5 Sep 2026
- 2026RenderFormer-V2: Neural Rendering with Heterogeneous Scene Primitives4 Sep 2026
- 2026UniMate: One Unified Model to Animate Diverse Skeletons4 Sep 2026
- 2026Almost Free State Prediction Separation3 Sep 2026
- 2026Text Capability Loss in Vision-Language Adaptation: An Attention-Sink Diagnosis1 Sep 2026
- 2026Deriving Scaling Laws for OpenEuroLLM Models: Learning Rate, Batch Size and Loss28 Aug 2026
- 2026Squeezing More from Limited Data with Recursive Transformers27 Aug 2026
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining25 Aug 2026
- 2026From Attention Masks to Inert Zero-Vector Tokens: OAttention and O-Closure for Token Dynamics21 Aug 2026
- 2026Listening Forward: Next Patch Embedding Prediction Enables Scalable Audio Learners20 Aug 2026
- 2026Abra: Scaling Diffusion Image Training18 Aug 2026
- 2026FLARE++: Low-rank attention with dynamic attention routing12 Aug 2026
- 2026LoopMTP: A looped transformer guided by latent multi-token prediction4 Aug 2026
- 2026Surface Keypoint Representation for Multi-Object and Articulated Human-Object Interaction Generation4 Aug 2026
- 2026AURORA-LM: Autoencoding Unified Representation for Continuous-Latent Diffusion Language Modeling3 Aug 2026
- 2026Faster-WAM: Do World Action Models Need Deep Action Modules?3 Aug 2026
- 2026One QK Channel, Many Sources: Tracing Low-Precision Attention Collapse3 Aug 2026
- 2026Training nGPT2 Aug 2026
- 2026ZUNA1.1: A more flexible EEG foundation model for Denoising and Super-resolution29 Jul 2026
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routing28 Jul 2026
- 2026A Controlled Study of Attention-Only Transformers20 Jul 2026
- 2026A Shortcut to Statistically Steady-State Turbulence with Flow Matching14 Jul 2026
- 2026Speaker-Disentangled Chunk-Wise Regression for Syllabic Tokenization5 Jul 2026
- 2026Gemma 4 Technical Report2 Jul 2026
- 2026PointSplat: Compact Gaussian Splatting via Human-Centric Prediction30 Jun 2026
- 2026Scaling Storm-Resolving Atmospheric AI Simulation to the Entire Planet30 Jun 2026
- 2026Flow Matching in Feature Space for Stochastic World Modeling27 Jun 2026
- 2026PhysiFormer: Learning to Simulate Mechanics in World Space25 Jun 2026
- 2026SCoPE: Sightline-Coordinate Positional Encoding for Video Diffusion Transformers25 Jun 2026
- 2026Hallucination in World Models is Predictable and Preventable25 Jun 2026
- 2026A Causal Foundation Model for Structure and Outcome Prediction25 Jun 2026
- 2026ZONOS2 Technical Report23 Jun 2026
- 2026Low-Rank Attention Residuals19 Jun 2026
50 newest of 266 All 266 (JSON)
- 2024RDT-1B: a Diffusion Foundation Model for Bimanual Manipulationcited by 670
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 294
- 2024Lumina-T2X: Transforming Text into Any Modality, Resolution, and Duration via Flow-based Large Diffusion Transformerscited by 137
- 2025Test-Time Training Done Rightcited by 121
- 2024Scalable High-Resolution Pixel-Space Image Synthesis with Hourglass Diffusion Transformerscited by 116
- 2025Latent Diffusion Model without Variational Autoencodercited by 83
- 2024Exploring the Role of Large Language Models in Prompt Encoding for Diffusion Modelscited by 45
- 2024MVGenMaster: Scaling Multi-View Generation from Any Image via 3D Priors Enhanced Diffusion Modelcited by 31
- 2024EzAudio: Enhancing Text-to-Audio Generation with Efficient Diffusion Transformercited by 30
- 2024SnapGen: Taming High-Resolution Text-to-Image Models for Mobile Devices with Efficient Architectures and Trainingcited by 27
- 2026ELF: Embedded Language Flowscited by 27
- 2025Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulationcited by 19
12 most cited of 37 All 37 (JSON)
- 2026NowcastDiT: Diffusion Transformers are Effective Precipitation Nowcasters29 Sep 2026
- 2026KiT: A Foundation Model for Financial Time-Series Forecasting using DiffusionTransformers28 Sep 2026
- 2026Abra: Scaling Diffusion Image Training18 Aug 2026
- 2026AURORA-LM: Autoencoding Unified Representation for Continuous-Latent Diffusion Language Modeling3 Aug 2026
- 2026A Shortcut to Statistically Steady-State Turbulence with Flow Matching14 Jul 2026
- 2026SCoPE: Sightline-Coordinate Positional Encoding for Video Diffusion Transformers25 Jun 2026
- 2026VolFill: Single-View Amodal 3D Scene Reconstruction with Volumetric Flow Matching29 May 2026
- 2026SCRIPT: Scalable Diffusion Policy with Multi-stage Training for Language-driven Physics-Based Humanoid Control21 May 2026
- 2026RiT: Vanilla Diffusion Transformers Suffice in Representation Space21 May 2026
- 2026Stable Audio 318 May 2026
- 2026ELF: Embedded Language Flows11 May 2026
- 2026What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusion8 May 2026
12 newest of 37 All 37 (JSON)
- 2024Genie: Generative Interactive Environmentscited by 706
- 2024GameGen-X: Interactive Open-world Game Video Generationcited by 124
- 2025LongCat-Video Technical Reportcited by 73
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPTcited by 66
- 2025DeepVerse: 4D Autoregressive Video Generation as a World Modelcited by 36
- 2026Faster-WAM: Do World Action Models Need Deep Action Modules?cited by 9
- 2025Generative World Modelling for Humanoids: 1X World Model Challenge Technical Reportcited by 8
- 2026Hallucination in World Models is Predictable and Preventablecited by 7
- 2025VFMF: World Modeling by Forecasting Vision Foundation Model Featurescited by 5
- 2026Flow Matching in Feature Space for Stochastic World Modelingcited by 4
- 2026PhysiFormer: Learning to Simulate Mechanics in World Spacecited by 1
- 2026Latent-Foresight: End-to-End Learning Predictable Representations for Latent World Modelscited by 0
- 2026Latent-Foresight: End-to-End Learning Predictable Representations for Latent World Models1 Oct 2026
- 2026Faster-WAM: Do World Action Models Need Deep Action Modules?3 Aug 2026
- 2026Flow Matching in Feature Space for Stochastic World Modeling27 Jun 2026
- 2026PhysiFormer: Learning to Simulate Mechanics in World Space25 Jun 2026
- 2026Hallucination in World Models is Predictable and Preventable25 Jun 2026
- 2025VFMF: World Modeling by Forecasting Vision Foundation Model Features12 Dec 2025
- 2025LongCat-Video Technical Report25 Oct 2025
- 2025Generative World Modelling for Humanoids: 1X World Model Challenge Technical Report8 Oct 2025
- 2025DeepVerse: 4D Autoregressive Video Generation as a World Model1 Jun 2025
- 2024DrivingWorld: Constructing World Model for Autonomous Driving via Video GPT27 Dec 2024
- 2024GameGen-X: Interactive Open-world Game Video Generation1 Nov 2024
- 2024Genie: Generative Interactive Environments23 Feb 2024
- 2026Controlled LLM Training on Spectral Spherecited by 28
- 2026Timer-S1: A Billion-Scale Time Series Foundation Model with Serial Scalingcited by 24
- 2026Arcee Trinity Large Technical Reportcited by 10
- 2026Towards Principled Design of Mixture-of-Experts Language Models under Memory and Inference Constraintscited by 0
- 2026Mellum2 Technical Reportcited by 0
- 2026Reversible Foundations: Training a 120B Sparse MoE through State-Preserving Scalingcited by 0
- 2026ZONOS2 Technical Reportcited by 0
- 2026Training nGPTcited by 0
- 2026Training nGPT2 Aug 2026
- 2026ZONOS2 Technical Report23 Jun 2026
- 2026Reversible Foundations: Training a 120B Sparse MoE through State-Preserving Scaling5 Jun 2026
- 2026Mellum2 Technical Report29 May 2026
- 2026Timer-S1: A Billion-Scale Time Series Foundation Model with Serial Scaling5 Mar 2026
- 2026Arcee Trinity Large Technical Report19 Feb 2026
- 2026Controlled LLM Training on Spectral Sphere13 Jan 2026
- 2026Towards Principled Design of Mixture-of-Experts Language Models under Memory and Inference Constraints13 Jan 2026
- 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Modelscited by 446
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 76
- 2025Dynamic Large Concept Models: Latent Reasoning in an Adaptive Semantic Spacecited by 14
- 2026Stability and Generalization in Looped Transformerscited by 9
- 2026Text Capability Loss in Vision-Language Adaptation: An Attention-Sink Diagnosiscited by 1
- 2026Text Capability Loss in Vision-Language Adaptation: An Attention-Sink Diagnosis1 Sep 2026
- 2026Stability and Generalization in Looped Transformers16 Apr 2026
- 2026Mamba-3: Improved Sequence Modeling using State Space Principles16 Mar 2026
- 2025Dynamic Large Concept Models: Latent Reasoning in an Adaptive Semantic Space31 Dec 2025
- 2025GLM-4.5: Agentic, Reasoning, and Coding (ARC) Foundation Models8 Aug 2025
- 2026Gemma 4 Technical Reportcited by 187
- 2026Penguin-VL: Exploring the Efficiency Limits of VLM with LLM-based Vision Encoderscited by 7
- 2025Shakti-VLMs: Scalable Vision-Language Models for Enterprise AIcited by 0
- 2026Gemma 4 Technical Report2 Jul 2026
- 2026Penguin-VL: Exploring the Efficiency Limits of VLM with LLM-based Vision Encoders6 Mar 2026
- 2025Shakti-VLMs: Scalable Vision-Language Models for Enterprise AI24 Feb 2025
- 2026Attention Smoothing Is All You Need For Unlearningcited by 6
- 2026Attention Smoothing Is All You Need For Unlearning1 Mar 2026
- 2025VGGT: Visual Geometry Grounded Transformercited by 1,475
- 2023BloombergGPT: A Large Language Model for Financecited by 923
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2024Unified Training of Universal Time Series Forecasting Transformerscited by 588
- 2022Memorizing Transformerscited by 219
- 2024LVSM: A Large View Synthesis Model with Minimal 3D Inductive Biascited by 178
- 2024Lumina-mGPT: Illuminate Flexible Photorealistic Text-to-Image Generation with Multimodal Generative Pretrainingcited by 148
- 2023Focused Transformer: Contrastive Training for Context Scalingcited by 133
- 2022Block-Recurrent Transformerscited by 107
- 2026Generative Modeling via Driftingcited by 98
- 2025MotionStreamer: Streaming Motion Generation via Diffusion-based Autoregressive Model in Causal Latent Spacecited by 82
- 2025RayZer: A Self-supervised Large View Synthesis Modelcited by 74
12 most cited of 186 All 186 (JSON)
- 2026QuadTok: Quadtree Visual Tokenizer for Autoregressive Image Generation7 Oct 2026
- 2026ScaffoldM3C: A Multimodal Sequential Monte Carlo Framework for Generative Stable Construction Planning30 Sep 2026
- 2026The Row Normalization Puzzle in Muon30 Sep 2026
- 2026One-Step Next-Latent Prediction Is Not a World Model28 Sep 2026
- 2026Query Expansion and Key Specialization in Transformer Attention Geometry28 Sep 2026
- 2026Broken Symmetry in BF16 Attention: Why FlashAttention Gradients Blow Up Late in Training28 Sep 2026
- 2026Natural Image Autoencoder-Based fMRI Representations for Trait and State Prediction28 Sep 2026
- 2026t0: A Time-Series Foundation Model for Forecasting with Context21 Sep 2026
- 2026ufakzeka-1: Building and Evaluating a 151M-Parameter Turkish Language Model from Scratch18 Sep 2026
- 2026Directions That Don't Drift: Stiefel Manifold Routing for Transformer Attention16 Sep 2026
- 2026Tele360: Real-Time Feed-Forward Human Reconstruction from Sparse Unposed Cameras14 Sep 2026
- 2026Depth and Scale in the Sub-150M Regime: JugnuLM-53M vs JugnuLM-110M13 Sep 2026
12 newest of 186 All 186 (JSON)