Raghuram Shankar
← Language Modelling

Part 9 · The Papers & Sources

The complete reference list

workauthors · identifierwhy it is here
Attention Is All You NeedAshish Vaswani et al., 2017 · arXiv:1706.03762The transformer. Removes recurrence; source of the scaled dot-product formula.
Language Models are Few-Shot LearnersTom B. Brown et al., 2020 · arXiv:2005.14165GPT-3: in-context learning emerges from scale alone.
Neural Machine Translation of Rare Words with Subword UnitsRico Sennrich et al., 2015 · arXiv:1508.07909Brings BPE into NLP; ends the unknown-word problem.
SentencePiece: A simple and language independent subword tokenizer and detokenizer for Neural Text ProcessingTaku Kudo et al., 2018 · arXiv:1808.06226Language-agnostic, reversible tokenization.
Byte Latent Transformer: Patches Scale Better Than TokensArtidoro Pagnoni et al., 2024 · arXiv:2412.09871Dynamic byte patches; the strongest tokenizer-free result.
Root Mean Square Layer NormalizationBiao Zhang et al., 2019 · arXiv:1910.07467Drops mean subtraction; cheaper and equally effective.
RoFormer: Enhanced Transformer with Rotary Position EmbeddingJianlin Su et al., 2021 · arXiv:2104.09864Relative position from absolute rotation.
GLU Variants Improve TransformerNoam Shazeer et al., 2020 · arXiv:2002.05202SwiGLU, and the 8/3 hidden-width convention.
Fast Transformer Decoding: One Write-Head is All You NeedNoam Shazeer et al., 2019 · arXiv:1911.02150MQA: one KV head for the whole layer.
GQA: Training Generalized Multi-Query Transformer Models from Multi-Head CheckpointsJoshua Ainslie et al., 2023 · arXiv:2305.13245Interpolates MHA and MQA; the current default.
Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts LayerNoam Shazeer et al., 2017 · arXiv:1701.06538Conditional computation: parameters without proportional FLOPs.
Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient SparsityWilliam Fedus et al., 2021 · arXiv:2101.03961Top-1 routing makes MoE simple and stable.
DeepSeek-V3 Technical ReportDeepSeek-AI et al., 2024 · arXiv:2412.19437MLA plus auxiliary-loss-free balancing at 671B/37B.
Scaling Laws for Neural Language ModelsJared Kaplan et al., 2020 · arXiv:2001.08361Loss follows smooth power laws in N, D and C.
Training Compute-Optimal Large Language ModelsJordan Hoffmann et al., 2022 · arXiv:2203.15556Chinchilla: scale parameters and tokens together.
Chinchilla Scaling: A replication attemptTamay Besiroglu et al., 2024 · arXiv:2404.10102Shows the published Approach-3 constants are internally inconsistent.
Scaling Data-Constrained Language ModelsNiklas Muennighoff et al., 2023 · arXiv:2305.16264What to do when the data, not the compute, runs out.
Decoupled Weight Decay RegularizationIlya Loshchilov et al., 2017 · arXiv:1711.05101AdamW: decay the weights, not the gradient.
Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter TransferGreg Yang et al., 2022 · arXiv:2203.03466muP: tune small, transfer to large.
FlashAttention: Fast and Memory-Efficient Exact Attention with IO-AwarenessTri Dao et al., 2022 · arXiv:2205.14135Exact attention, restructured around memory traffic.
FlashAttention-2: Faster Attention with Better Parallelism and Work PartitioningTri Dao et al., 2023 · arXiv:2307.08691Better work partitioning; roughly doubles throughput.
Online normalizer calculation for softmaxMaxim Milakov et al., 2018 · arXiv:1805.02867The streaming-softmax trick FlashAttention is built on.
Training Deep Nets with Sublinear Memory CostTianqi Chen et al., 2016 · arXiv:1604.06174Gradient checkpointing: √n memory for one extra forward pass.
ZeRO: Memory Optimizations Toward Training Trillion Parameter ModelsSamyam Rajbhandari et al., 2019 · arXiv:1910.02054Shard optimizer state, gradients, then parameters.
Megatron-LM: Training Multi-Billion Parameter Language Models Using Model ParallelismMohammad Shoeybi et al., 2019 · arXiv:1909.08053Tensor parallelism inside the layer.
GPipe: Efficient Training of Giant Neural Networks using Pipeline ParallelismYanping Huang et al., 2018 · arXiv:1811.06965Pipeline parallelism with micro-batches.
Ring Attention with Blockwise Transformers for Near-Infinite ContextHao Liu et al., 2023 · arXiv:2310.01889Context parallelism with communication hidden behind compute.
Training language models to follow instructions with human feedbackLong Ouyang et al., 2022 · arXiv:2203.02155InstructGPT: the RLHF pipeline that made models usable.
Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov et al., 2023 · arXiv:2305.18290Closed-form solution removes the reward model and the RL loop.
DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language ModelsZhihong Shao et al., 2024 · arXiv:2402.03300GRPO: the group replaces the value network.
DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement LearningDeepSeek-AI et al., 2025 · arXiv:2501.12948Reasoning RL with verifiable rewards, at scale.
Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?Yang Yue et al., 2025 · arXiv:2504.13837Argues RLVR sharpens sampling rather than adding capability.
The FineWeb Datasets: Decanting the Web for the Finest Text Data at ScaleGuilherme Penedo et al., 2024 · arXiv:2406.17557Open, documented web curation at 15T tokens.
DataComp-LM: In search of the next generation of training sets for language modelsJeffrey Li et al., 2024 · arXiv:2406.11794Controlled benchmark showing curation beats raw scale.
Deduplicating Training Data Makes Language Models BetterKatherine Lee et al., 2021 · arXiv:2107.06499Deduplication as a first-class training-data operation.
The Pile: An 800GB Dataset of Diverse Text for Language ModelingLeo Gao et al., 2020 · arXiv:2101.00027The first widely-used open diverse pretraining corpus.
Exploring the Limits of Transfer Learning with a Unified Text-to-Text TransformerColin Raffel et al., 2019 · arXiv:1910.10683T5 and C4; the cleaned-Common-Crawl template.
DoReMi: Optimizing Data Mixtures Speeds Up Language Model PretrainingSang Michael Xie et al., 2023 · arXiv:2305.10429Learns domain weights instead of guessing them.