When Perplexity Lies: Generation-Focused Distillation of Hybrid Sequence Models
Not on the map. Why?
Abstract (arXiv)
Converting a pretrained Transformer into a more efficient hybrid model through distillation offers a promising approach to reducing inference costs. However, achieving high-quality generation in distilled models requires careful joint design of both the student architecture and the distillation process. Many prior distillation works evaluate downstream multiple-choice benchmarks by ranking candidate answers with log-likelihood rather than requiring autoregressive generation, which can obscure important differences in model quality. For example, on overlapping benchmarks, we show that a 7B distilled model that nearly matches its teacher to within 0.2 pp under log-likelihood scoring falls behind by 20.8 pp when it must generate answers autoregressively. We investigate this phenomenon with GenDistill, a multi-stage pipeline we designed for distilling a pretrained Transformer into an efficient Hybrid Kimi Delta Attention (Hybrid-KDA) student. Using it as a controlled testbed on Qwen3-0.6B, we systematically ablate six design axes (training objective, loss masking, training duration, dataset selection, parameter freezing, and architecture choice) and evaluate every choice under both log-likelihood and generation-based protocols. We find that log-likelihood-based evaluation consistently underestimates the gap between teacher and student, and can in some cases reverse the ranking of design choices, so conclusions drawn from perplexity-only evaluation may be misleading. Among the factors we study, dataset selection, completion-only masking, and freezing attention layers during post-training have the largest impact on generation quality. Our best distillation recipe, using a Hybrid-KDA model as the student, retains 86-90% of teacher accuracy on knowledge benchmarks while reducing KV cache memory by up to 75% and improving time-to-first-token by 2-4x at 128K-token contexts.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 39 of its 54 references that are in the map. First posted on arXiv 27 Mar 2026.
Builds on
39 of 54 references.
References · 39
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,378
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,270
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
27 more
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2023Quantifying Language Models' Sensitivity to Spurious Features in Prompt Design or: How I learned to start worrying about prompt formattingcited by 778
- 2023C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Modelscited by 694
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
- 2023Large Language Models Are Not Robust Multiple Choice Selectorscited by 454
- 2023CMMLU: Measuring massive multitask language understanding in Chinesecited by 436
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 373
- 2025RWKV-7 "Goose" with Expressive Dynamic State Evolutioncited by 134
- 2024The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicrycited by 119
- 2024Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modelscited by 66
- 2025Are Reasoning Models More Prone to Hallucination?cited by 62
- 2024Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Modelscited by 57
- 2024Linearizing Large Language Modelscited by 51
- 2025Jet-Nemotron: Efficient Language Model with Post Neural Architecture Searchcited by 39
- 2025Llamba: Scaling Distilled Recurrent Models for Efficient Language Processingcited by 35
- 2025Not All Correct Answers Are Equal: Why Your Distillation Source Matterscited by 25
- 2025M1: Towards Scalable Test-Time Compute with Mamba Reasoning Modelscited by 21
- 2025Distilling to Hybrid Attention Models via KL-Guided Layer Selectioncited by 17
- 2026Hybrid Linear Attention Done Right: Efficient Distillation and Effective Architectures for Extremely Long Contextscited by 16
- 2024DiJiang: Efficient Large Language Models through Compact Kernelizationcited by 15
- 2025Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanismcited by 15
- 2024Mitigating Copy Bias in In-Context Learning through Neuron Pruningcited by 11
- 2025ARWKV: Pretrain is not what we need, an RNN-Attention-Based Language Model Born from Transformercited by 9
- 2025Debunk the Myth of SFT Generalizationcited by 8
- 2025FlashEVA: Accelerating LLM inference via Efficient Attentioncited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Expressivity-Efficiency Tradeoffs for Hybrid Sequence Models
- 2025Distilled Pretraining: A modern lens of Data, In-Context Learning and Test-Time Scaling
- 2026Sharpen Without Search: On-Policy Distillation of Sequence-Level Power Distribution
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models
- 2025Knowledge Distillation and Dataset Distillation of Large Language Models: Emerging Trends, Challenges, and Future Directions
- 2025Thinking Slow, Fast: Scaling Inference Compute with Distilled Reasoners
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Models
- 2020Autoregressive Knowledge Distillation through Imitation Learning