GKD: A General Knowledge Distillation Framework for Large-scale Pre-trained Language Model
Not on the map. Why?
Abstract (arXiv)
Currently, the reduction in the parameter scale of large-scale pre-trained language models (PLMs) through knowledge distillation has greatly facilitated their widespread deployment on various devices. However, the deployment of knowledge distillation systems faces great challenges in real-world industrial-strength applications, which require the use of complex distillation methods on even larger-scale PLMs (over 10B), limited by memory on GPUs and the switching of methods. To overcome these challenges, we propose GKD, a general knowledge distillation framework that supports distillation on larger-scale PLMs using various distillation methods. With GKD, developers can build larger distillation models on memory-limited GPUs and easily switch and combine different distillation methods within a single framework. Experimental results show that GKD can support the distillation of at least 100B-scale PLMs and 25 mainstream methods on 8 NVIDIA A100 (40GB) GPUs.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 33 of its 42 references that are in the map. First posted on arXiv 11 Jun 2023.
Builds on
33 of 42 references.
References · 33
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,801
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2019SuperGLUE: A Stickier Benchmark for General-Purpose Language Understanding Systemscited by 1,718
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,041
- 2021GLM: General Language Model Pretraining with Autoregressive Blank Infillingcited by 1,026
- 2019Improved Knowledge Distillation via Teacher Assistantcited by 567
21 more
- 2019Patient Knowledge Distillation for BERT Model Compressioncited by 502
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 476
- 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 339
- 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 240
- 2019Well-Read Students Learn Better: On the Importance of Pre-training Compact Modelscited by 118
- 2020BERT-of-Theseus: Compressing BERT by Progressive Module Replacingcited by 112
- 2019Knowledge Distillation from Internal Representationscited by 87
- 2020Densely Guided Knowledge Distillation using Multiple Teacher Assistantscited by 82
- 2020ALP-KD: Attention-Based Layer Projection for Knowledge Distillationcited by 65
- 2020Reinforced Multi-Teacher Selection for Knowledge Distillationcited by 62
- 2021Annealing Knowledge Distillationcited by 61
- 2019Model Compression with Two-stage Multi-teacher Knowledge Distillation for Web Question Answering Systemcited by 43
- 2021One Teacher is Enough? Pre-trained Language Model Distillation from Multiple Teacherscited by 39
- 2021Dynamic Knowledge Distillation for Pre-trained Language Modelscited by 38
- 2021Distilling Linguistic Context for Language Model Compressioncited by 33
- 2020LRC-BERT: Latent-representation Contrastive Knowledge Distillation for Natural Language Understandingcited by 27
- 2020TextBrewer: An Open-Source Knowledge Distillation Toolkit for Natural Language Processingcited by 25
- 2021Causal Distillation for Language Modelscited by 19
- 2021RAIL-KD: RAndom Intermediate Layer Mapping for Knowledge Distillationcited by 18
- 2021Marginal Utility Diminishes: Exploring the Minimum Knowledge for BERT Knowledge Distillationcited by 6
- 2022Continuation KD: Improved Knowledge Distillation through the Lens of Continuation Optimizationcited by 3
Built on by
Citing papers per year: 2023: 1; 2024: 5; 2025: 1; 2026: 8.
15 papers.
Sort:
By area:
- 2023SiDA-MoE: Sparsity-Inspired Data-Aware Serving for Efficient and Scalable Large Mixture-of-Experts Modelscited by 36
- 2025AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxescited by 28
- 2024Active Data Curation Effectively Distills Large-Scale Multimodal Modelscited by 19
- 2026MAD-OPD: Breaking the Ceiling in On-Policy Distillation via Multi-Agent Debatecited by 12
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignmentcited by 10
- 2024Reliable Model Watermarking: Defending Against Theft without Compromising on Evasioncited by 9
- 2026VLA-OPD: Bridging Offline SFT and Online RL for Vision-Language-Action Models via On-Policy Distillationcited by 9
- 2024Head-wise Shareable Attention for Large Language Modelscited by 4
- 2026Rethinking the Role of Tensor Decompositions in Post-Training LLM Compressioncited by 3
- 2026X3-OPD: Distilling Reasoning into Large Audio-Language Models via On-Policy Alignmentcited by 3
- 2024QPruner: Probabilistic Decision Quantization for Structured Pruning in Large Language Modelscited by 2
- 2026DreOPD: Degraded-Reference Extrapolative On-Policy Distillation for Flow-matching Modelscited by 2
3 more
- 2026LATS: Large Language Model Assisted Teacher-Student Framework for Multi-Agent Reinforcement Learning in Traffic Signal Controlcited by 1
- 2026Optimizing Teacher-Student Partitioning for Scalable Knowledge Distillation on HPC Systemscited by 0
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Updatecited by 0
- 2026DreOPD: Degraded-Reference Extrapolative On-Policy Distillation for Flow-matching Models10 Aug 2026
- 2026X3-OPD: Distilling Reasoning into Large Audio-Language Models via On-Policy Alignment23 Jul 2026
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Update13 Jul 2026
- 2026Optimizing Teacher-Student Partitioning for Scalable Knowledge Distillation on HPC Systems26 Jun 2026
- 2026Rethinking the Role of Tensor Decompositions in Post-Training LLM Compression2 Jun 2026
- 2026MAD-OPD: Breaking the Ceiling in On-Policy Distillation via Multi-Agent Debate2 May 2026
- 2026VLA-OPD: Bridging Offline SFT and Online RL for Vision-Language-Action Models via On-Policy Distillation27 Mar 2026
- 2026LATS: Large Language Model Assisted Teacher-Student Framework for Multi-Agent Reinforcement Learning in Traffic Signal Control25 Mar 2026
- 2025AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxes17 Jun 2025
- 2024QPruner: Probabilistic Decision Quantization for Structured Pruning in Large Language Models16 Dec 2024
- 2024Active Data Curation Effectively Distills Large-Scale Multimodal Models27 Nov 2024
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignment19 Sep 2024
- 2025AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxescited by 28
- 2026MAD-OPD: Breaking the Ceiling in On-Policy Distillation via Multi-Agent Debatecited by 12
- 2026LATS: Large Language Model Assisted Teacher-Student Framework for Multi-Agent Reinforcement Learning in Traffic Signal Controlcited by 1
- 2026MAD-OPD: Breaking the Ceiling in On-Policy Distillation via Multi-Agent Debate2 May 2026
- 2026LATS: Large Language Model Assisted Teacher-Student Framework for Multi-Agent Reinforcement Learning in Traffic Signal Control25 Mar 2026
- 2025AgentDistill: Training-Free Agent Distillation with Generalizable MCP Boxes17 Jun 2025
- 2024Active Data Curation Effectively Distills Large-Scale Multimodal Modelscited by 19
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignmentcited by 10
- 2024Head-wise Shareable Attention for Large Language Modelscited by 4
- 2024QPruner: Probabilistic Decision Quantization for Structured Pruning in Large Language Modelscited by 2
- 2026Optimizing Teacher-Student Partitioning for Scalable Knowledge Distillation on HPC Systemscited by 0
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Updatecited by 0
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Update13 Jul 2026
- 2026Optimizing Teacher-Student Partitioning for Scalable Knowledge Distillation on HPC Systems26 Jun 2026
- 2024QPruner: Probabilistic Decision Quantization for Structured Pruning in Large Language Models16 Dec 2024
- 2024Active Data Curation Effectively Distills Large-Scale Multimodal Models27 Nov 2024
- 2024Enhancing Knowledge Distillation of Large Language Models through Efficient Multi-Modal Distribution Alignment19 Sep 2024
- 2024Head-wise Shareable Attention for Large Language Models19 Feb 2024