Accelerating Training of Transformer-Based Language Models with Progressive Layer Dropping
Not on the map. Why?
Abstract (arXiv)
Recently, Transformer-based language models have demonstrated remarkable performance across many NLP domains. However, the unsupervised pre-training step of these models suffers from unbearable overall computational expenses. Current methods for accelerating the pre-training either rely on massive parallelism with advanced hardware or are not applicable to language modeling. In this work, we propose a method based on progressive layer dropping that speeds the training of Transformer-based language models, not at the cost of excessive hardware resources but from model architecture change and training technique boosted efficiency. Extensive experiments on BERT show that the proposed method achieves a 24% time reduction on average per sample and allows the pre-training to be 2.5 times faster than the baseline to get a similar accuracy on downstream tasks. While being faster, our pre-trained models are equipped with strong knowledge transferability, achieving comparable and sometimes higher GLUE score than the baseline when pre-trained with the same number of samples.
Key facts. Cited by 93 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 16; 2023: 19; 2024: 33; 2025: 9; 2026: 5.
93 papers.
Sort:
By area:
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2021Pre-Trained Models: Past, Present and Futurecited by 319
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 219
- 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 138
- 2023DeID-GPT: Zero-shot Medical Text De-Identification by GPT-4cited by 128
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
- 2022Cramming: Training a Language Model on a Single GPU in One Daycited by 81
- 2023Learning to Grow Pretrained Models for Efficient Transformer Trainingcited by 81
- 2021How to Train BERT with an Academic Budgetcited by 80
- 2022Unified Visual Transformer Compressioncited by 79
38 more
- 2021bert2BERT: Towards Reusable Pretrained Language Modelscited by 70
- 2022ELLE: Efficient Lifelong Pre-training for Emerging Datacited by 70
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2021AutoFreeze: Automatically Freezing Model Blocks to Accelerate Fine-tuningcited by 48
- 2024DiP-GO: A Diffusion Pruner via Few-step Gradient Optimizationcited by 43
- 2022Token Dropping for Efficient BERT Pretrainingcited by 42
- 2022EfficientTrain: Exploring Generalized Curriculum Learning for Training Visual Backbonescited by 42
- 2023No Train No Gain: Revisiting Efficient Training Algorithms For Transformer-based Language Modelscited by 41
- 2021Knowledge Inheritance for Pre-trained Language Modelscited by 38
- 2023A Survey on Efficient Training of Transformerscited by 37
- 2022Automated Progressive Learning for Efficient Training of Vision Transformerscited by 36
- 2021Layer-wise Adaptive Model Aggregation for Scalable Federated Learningcited by 35
- 2023Natural Language Generation and Understanding of Big Code for AI-Assisted Programming: A Reviewcited by 32
- 2024Data curation via joint example selection further accelerates multimodal learningcited by 32
- 2021M6-10T: A Sharing-Delinking Paradigm for Efficient Multi-Trillion Parameter Pretrainingcited by 31
- 2022Parameter-Efficient Sparsity for Large Language Models Fine-Tuningcited by 27
- 2023On Efficient Training of Large-Scale Deep Learning Models: A Literature Reviewcited by 26
- 2023Federated Learning-Empowered AI-Generated Content in Wireless Networkscited by 26
- 20211-bit LAMB: Communication Efficient Large-Scale Large-Batch Training with LAMB's Convergence Speedcited by 24
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
- 2021OAG-BERT: Towards A Unified Backbone Language Model For Academic Knowledge Servicescited by 20
- 2021LightSeq2: Accelerated Training for Transformer-based Models on GPUscited by 19
- 2022Extreme Compression for Pre-trained Transformers Made Simple and Efficientcited by 19
- 2023NLU on Data Diets: Dynamic Data Subset Selection for NLP Classification Taskscited by 19
- 2024SLMRec: Distilling Large Language Models into Small for Sequential Recommendationcited by 18
- 2023Contrastive Alignment of Vision to Language Through Parameter-Efficient Transfer Learningcited by 16
- 2023Reusing Pretrained Models by Multi-linear Operators for Efficient Trainingcited by 15
- 2024SmileyLlama: Modifying Large Language Models for Directed Chemical Space Explorationcited by 14
- 2024SmartFRZ: An Efficient Training Framework using Attention-Based Layer Freezingcited by 12
- 2022Layer Freezing & Data Sieving: Missing Pieces of a Generic Framework for Sparse Trainingcited by 11
- 2022Random-LTD: Random and Layerwise Token Dropping Brings Efficient Training for Large-scale Transformerscited by 11
- 2023Revisiting Token Dropping Strategy in Efficient BERT Pretrainingcited by 11
- 2024Persistent Topological Features in Large Language Modelscited by 11
- 2022Progressive Transformation Learning for Leveraging Virtual Images in Trainingcited by 10
- 2024Preparing Lessons for Progressive Training on Language Modelscited by 10
- 2024Efficient Stagewise Pretraining via Progressive Subnetworkscited by 10
- 2022Graph Reasoning Transformer for Image Parsingcited by 9
50 most cited of 93 All 93 (JSON)
- 2026When Can Attention Heads Be Statically Defined?28 Sep 2026
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference4 Sep 2026
- 2026EPAS: Efficient Training with Progressive Activation Sharing27 Jan 2026
- 2026Suppressing Final Layer Hidden State Jumps in Transformer Pretraining26 Jan 2026
- 2026Distillation-based Layer Dropping (DLD): Effective End-to-end Framework for Dynamic Speech Networks22 Jan 2026
- 2025Dynamic Rank Reinforcement Learning for Adaptive Low-Rank Multi-Head Self Attention in Large Language Models17 Dec 2025
- 2025Argus: Quality-Aware High-Throughput Text-to-Image Inference Serving System10 Nov 2025
- 2025NeuronMLP: Efficient LLM Inference via Singular Value Decomposition Compression and Tiling on AWS Trainium29 Oct 2025
- 2025Convergence, design and training of continuous-time dropout as a random batch method15 Oct 2025
- 2025Input Conditioned Layer Dropping in Speech Foundation Models10 Jul 2025
- 2025Curriculum-Guided Layer Scaling for Language Model Pretraining13 Jun 2025
- 2025TaDA: Training-free recipe for Decoding with Adaptive KV Cache Compression and Mean-centering5 Jun 2025
38 more
- 2025CLaSp: In-Context Layer Skip for Self-Speculative Decoding30 May 2025
- 2025Temporal Action Detection Model Compression by Progressive Block Drop21 Mar 2025
- 2024TrimLLM: Progressive Layer Dropping for Domain-Specific LLMs15 Dec 2024
- 2024DiP-GO: A Diffusion Pruner via Few-step Gradient Optimization22 Oct 2024
- 2024Layer-wise Importance Matters: Less Memory for Better Performance in Parameter-efficient Fine-tuning of Large Language Models15 Oct 2024
- 2024Persistent Topological Features in Large Language Models14 Oct 2024
- 2024TOAST: Transformer Optimization using Adaptive and Simple Transformations7 Oct 2024
- 2024Exploring the Benefit of Activation Sparsity in Pre-training4 Oct 2024
- 2024A framework for measuring the training efficiency of a neural architecture12 Sep 2024
- 2024Efficient Training of Large Vision Models via Advanced Automated Progressive Learning6 Sep 2024
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspective4 Sep 2024
- 2024SmileyLlama: Modifying Large Language Models for Directed Chemical Space Exploration3 Sep 2024
- 2024Attention Is All You Need But You Don't Need All Of It For Inference of Large Language Models22 Jul 2024
- 2024The Remarkable Robustness of LLMs: Stages of Inference?27 Jun 2024
- 2024Data curation via joint example selection further accelerates multimodal learning25 Jun 2024
- 2024Federating to Grow Transformers with Constrained Resources without Model Sharing19 Jun 2024
- 2024REP: Resource-Efficient Prompting for Rehearsal-Free Continual Learning7 Jun 2024
- 2024DHA: Learning Decoupled-Head Attention from Transformer Checkpoints via Adaptive Heads Fusion3 Jun 2024
- 2024SLMRec: Distilling Large Language Models into Small for Sequential Recommendation28 May 2024
- 2024OpenBA-V2: Reaching 77.3% High Compression Ratio with Fast Multi-Stage Pruning9 May 2024
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding25 Apr 2024
- 2024Does Transformer Interpretability Transfer to RNNs?9 Apr 2024
- 2024A Multi-Level Framework for Accelerating Training Transformer Models7 Apr 2024
- 2024The Unreasonable Ineffectiveness of the Deeper Layers26 Mar 2024
- 2024The Heuristic Core: Understanding Subnetwork Generalization in Pretrained Language Models6 Mar 2024
- 2024When do Convolutional Neural Networks Stop Learning?4 Mar 2024
- 2024DropBP: Accelerating Fine-Tuning of Large Language Models by Dropping Backward Propagation27 Feb 2024
- 2024Towards Optimal Learning of Language Models27 Feb 2024
- 2024Efficient Backpropagation with Variance-Controlled Adaptive Sampling27 Feb 2024
- 2024Model Compression and Efficient Inference for Large Language Models: A Survey15 Feb 2024
- 2024Efficient Stagewise Pretraining via Progressive Subnetworks8 Feb 2024
- 2024Data Quality Matters: Suicide Intention Detection on Social Media Posts Using RoBERTa-CNN3 Feb 2024
- 2024SmartFRZ: An Efficient Training Framework using Attention-Based Layer Freezing30 Jan 2024
- 2024Preparing Lessons for Progressive Training on Language Models17 Jan 2024
- 2024UPDP: A Unified Progressive Depth Pruner for CNN and Vision Transformer12 Jan 2024
- 2023Local Masking Meets Progressive Freezing: Crafting Efficient Vision Transformers for Self-Supervised Learning2 Dec 2023
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden State8 Nov 2023
- 2023FedPEAT: Convergence of Federated Learning, Parameter-Efficient Fine Tuning, and Emulator Assisted Tuning for Artificial Intelligence Foundation Models with Mobile Edge Computing26 Oct 2023
50 newest of 93 All 93 (JSON)
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
- 2022Token Dropping for Efficient BERT Pretrainingcited by 42
- 2023Revisiting Token Dropping Strategy in Efficient BERT Pretrainingcited by 11
- 2025CLaSp: In-Context Layer Skip for Self-Speculative Decodingcited by 9
- 2025TaDA: Training-free recipe for Decoding with Adaptive KV Cache Compression and Mean-centeringcited by 3
- 2024TrimLLM: Progressive Layer Dropping for Domain-Specific LLMscited by 2
- 2025NeuronMLP: Efficient LLM Inference via Singular Value Decomposition Compression and Tiling on AWS Trainiumcited by 0
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inferencecited by 0
- 2026Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference4 Sep 2026
- 2025NeuronMLP: Efficient LLM Inference via Singular Value Decomposition Compression and Tiling on AWS Trainium29 Oct 2025
- 2025TaDA: Training-free recipe for Decoding with Adaptive KV Cache Compression and Mean-centering5 Jun 2025
- 2025CLaSp: In-Context Layer Skip for Self-Speculative Decoding30 May 2025
- 2024TrimLLM: Progressive Layer Dropping for Domain-Specific LLMs15 Dec 2024
- 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding25 Apr 2024
- 2023Revisiting Token Dropping Strategy in Efficient BERT Pretraining24 May 2023
- 2022Token Dropping for Efficient BERT Pretraining24 Mar 2022
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 219
- 2024The Unreasonable Ineffectiveness of the Deeper Layers26 Mar 2024
- 2024Does Transformer Interpretability Transfer to RNNs?cited by 9
- 2024Does Transformer Interpretability Transfer to RNNs?9 Apr 2024
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2021Pre-Trained Models: Past, Present and Futurecited by 319
- 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 138
- 2023DeID-GPT: Zero-shot Medical Text De-Identification by GPT-4cited by 128
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
- 2022Cramming: Training a Language Model on a Single GPU in One Daycited by 81
- 2023Learning to Grow Pretrained Models for Efficient Transformer Trainingcited by 81
- 2021How to Train BERT with an Academic Budgetcited by 80
- 2022Unified Visual Transformer Compressioncited by 79
- 2021bert2BERT: Towards Reusable Pretrained Language Modelscited by 70
- 2022ELLE: Efficient Lifelong Pre-training for Emerging Datacited by 70
12 most cited of 76 All 76 (JSON)
- 2026When Can Attention Heads Be Statically Defined?28 Sep 2026
- 2026EPAS: Efficient Training with Progressive Activation Sharing27 Jan 2026
- 2026Suppressing Final Layer Hidden State Jumps in Transformer Pretraining26 Jan 2026
- 2026Distillation-based Layer Dropping (DLD): Effective End-to-end Framework for Dynamic Speech Networks22 Jan 2026
- 2025Dynamic Rank Reinforcement Learning for Adaptive Low-Rank Multi-Head Self Attention in Large Language Models17 Dec 2025
- 2025Argus: Quality-Aware High-Throughput Text-to-Image Inference Serving System10 Nov 2025
- 2025Convergence, design and training of continuous-time dropout as a random batch method15 Oct 2025
- 2025Input Conditioned Layer Dropping in Speech Foundation Models10 Jul 2025
- 2025Curriculum-Guided Layer Scaling for Language Model Pretraining13 Jun 2025
- 2025Temporal Action Detection Model Compression by Progressive Block Drop21 Mar 2025
- 2024Layer-wise Importance Matters: Less Memory for Better Performance in Parameter-efficient Fine-tuning of Large Language Models15 Oct 2024
- 2024Persistent Topological Features in Large Language Models14 Oct 2024
12 newest of 76 All 76 (JSON)