Length-Adaptive Transformer: Train Once with Length Drop, Use Anytime with Search
Not on the map. Why?
Cited by 26 in Efficient LLM inference and serving, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Despite transformers' impressive accuracy, their computational cost is often prohibitive to use with limited computational resources. Most previous approaches to improve inference efficiency require a separate model for each possible computational budget. In this paper, we extend PoWER-BERT (Goyal et al., 2020) and propose Length-Adaptive Transformer that can be used for various inference scenarios after one-shot training. We train a transformer with LengthDrop, a structural variant of dropout, which stochastically determines a sequence length at each layer. We then conduct a multi-objective evolutionary search to find a length configuration that maximizes the accuracy and minimizes the efficiency metric under any given computational budget. Additionally, we significantly extend the applicability of PoWER-BERT beyond sequence-level classification into token-level classification with Drop-and-Restore process that drops word-vectors temporarily in intermediate layers and restores at the last layer if necessary. We empirically verify the utility of the proposed approach by demonstrating the superior accuracy-efficiency trade-off under various setups, including span-based question answering and text classification. Code is available at https://github.com/clovaai/length-adaptive-transformer.
Key facts. Cited by 76 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 6; 2022: 17; 2023: 21; 2024: 18; 2025: 12; 2026: 2.
76 papers.
Sort:
By area:
- 2022Token Merging: Your ViT But Fastercited by 995
- 2023LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compressioncited by 443
- 2022Confident Adaptive Language Modelingcited by 297
- 2023LLMLingua: Compressing Prompts for Accelerated Inference of Large Language Modelscited by 279
- 2021Learned Token Pruning for Transformerscited by 138
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Mergingcited by 123
- 2024Dynamic Memory Compression: Retrofitting LLMs for Accelerated Inferencecited by 115
- 2024Evolutionary Computation in the Era of Large Language Model: Survey and Roadmapcited by 89
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2023Zero-TPrune: Zero-Shot Token Pruning through Leveraging of the Attention Graph in Pre-Trained Transformerscited by 61
- 2021A Survey on Green Deep Learningcited by 52
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodalitycited by 49
38 more
- 2022A Survey on Model Compression and Acceleration for Pretrained Language Modelscited by 45
- 2023CrossGET: Cross-Guided Ensemble of Tokens for Accelerating Vision-Language Transformerscited by 44
- 2024ElasticTok: Adaptive Tokenization for Image and Videocited by 44
- 2022Model Cascading: Towards Jointly Improving Efficiency and Accuracy of NLP Systemscited by 43
- 2024AIM: Adaptive Inference of Multi-Modal LLMs via Token Merging and Pruningcited by 43
- 2021TR-BERT: Dynamic Token Reduction for Accelerating BERT Inferencecited by 40
- 2024MELTing point: Mobile Evaluation of Language Transformerscited by 35
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2024The What, Why, and How of Context Length Extension Techniques in Large Language Models -- A Detailed Surveycited by 32
- 2025Shifting AI Efficiency From Model-Centric to Data-Centric Compressioncited by 29
- 2023Lifting the Curse of Capacity Gap in Distilling Language Modelscited by 25
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 25
- 2022A Survey on Dynamic Neural Networks for Natural Language Processingcited by 24
- 2024Token-level Correlation-guided Compression for Efficient Multimodal Document Understandingcited by 22
- 2022Transkimmer: Transformer Learns to Layer-wise Skimcited by 21
- 2022AdapLeR: Speeding up Inference by Adaptive Length Reductioncited by 19
- 2024Characterizing Prompt Compression Methods for Long Context Inferencecited by 18
- 2023OrchestraLLM: Efficient Orchestration of Language Models for Dialogue State Trackingcited by 17
- 2021Block-Skim: Efficient Question Answering for Transformercited by 16
- 2021A Study on Token Pruning for ColBERTcited by 14
- 2023Accelerating Transducers through Adjacent Token Mergingcited by 14
- 2022Pyramid-BERT: Reducing Complexity via Successive Core-set based Token Selectioncited by 13
- 2022DeepSpeed Data Efficiency: Improving Deep Learning Model Quality and Training Efficiency via Efficient Data Sampling and Routingcited by 13
- 2023ConsistentEE: A Consistent and Hardness-Guided Early Exiting Method for Accelerating Language Models Inferencecited by 13
- 2022Random-LTD: Random and Layerwise Token Dropping Brings Efficient Training for Large-scale Transformerscited by 11
- 2024iLLaVA: An Image is Worth Fewer Than 1/3 Input Tokens in Large Multimodal Modelscited by 11
- 2022Can Open-Domain QA Reader Utilize External Knowledge Efficiently like Humans?cited by 10
- 2023Optimizing Retrieval-augmented Reader Models via Token Eliminationcited by 10
- 2023Improving Machine Translation with Large Language Models: A Preliminary Study with Cooperative Decodingcited by 10
- 2023Do We Really Need a Large Number of Visual Prompts?cited by 9
- 2024Selection-p: Self-Supervised Task-Agnostic Prompt Compression for Faithfulness and Transferabilitycited by 9
- 2025CipherPrune: Efficient and Scalable Private Transformer Inferencecited by 9
- 2022E-LANG: Energy-Based Joint Inferencing of Super and Swift Language Modelscited by 8
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Frameworkcited by 8
- 2025Rethinking Chain-of-Thought Reasoning for Videoscited by 8
- 2023DEED: Dynamic Early Exit on Decoder for Accelerating Encoder-Decoder Transformer Modelscited by 7
- 2024Focus on the Core: Efficient Attention via Pruned Token Compression for Document Classificationcited by 7
- 2025LightVLM: Acceleraing Large Multimodal Models with Pyramid Token Merging and KV Cache Compressioncited by 7
50 most cited of 76 All 76 (JSON)
- 2026Sparse Token Routing in Efficient Transformers21 Aug 2026
- 2026Structured-Sparse Attention for Entity Tracking with Subquadratic Sequence Complexity21 May 2026
- 2025Rethinking Chain-of-Thought Reasoning for Videos10 Dec 2025
- 2025ResFormer: All-Time Reservoir Memory for Long Sequence Classification28 Sep 2025
- 2025LightVLM: Acceleraing Large Multimodal Models with Pyramid Token Merging and KV Cache Compression30 Aug 2025
- 2025Training-free Token Reduction for Vision Mamba18 Jul 2025
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Framework7 Jul 2025
- 2025Shifting AI Efficiency From Model-Centric to Data-Centric Compression25 May 2025
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality23 May 2025
- 2025Accelerating Adaptive Retrieval Augmented Generation via Instruction-Driven Representation Reduction of Retrieval Overlaps19 May 2025
- 2025Saliency-driven Dynamic Token Pruning for Large Language Models6 Apr 2025
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representation3 Mar 2025
38 more
- 2025CipherPrune: Efficient and Scalable Private Transformer Inference24 Feb 2025
- 2025Top-Theta Attention: Sparsifying Transformers by Compensated Thresholding12 Feb 2025
- 2024ImagePiece: Content-aware Re-tokenization for Efficient Image Recognition21 Dec 2024
- 2024Deploying Foundation Model Powered Agent Services: A Survey18 Dec 2024
- 2024Faster Vision Mamba is Rebuilt in Minutes via Merged Token Re-training17 Dec 2024
- 2024iLLaVA: An Image is Worth Fewer Than 1/3 Input Tokens in Large Multimodal Models9 Dec 2024
- 2024AIM: Adaptive Inference of Multi-Modal LLMs via Token Merging and Pruning4 Dec 2024
- 2024Training Noise Token Pruning27 Nov 2024
- 2024Self-Satisfied: An end-to-end framework for SAT generation and prediction18 Oct 2024
- 2024Selection-p: Self-Supervised Task-Agnostic Prompt Compression for Faithfulness and Transferability15 Oct 2024
- 2024ElasticTok: Adaptive Tokenization for Image and Video10 Oct 2024
- 2024Practical token pruning for foundation models in few-shot conversational virtual assistant systems21 Aug 2024
- 2024AdaCoder: Adaptive Prompt Compression for Programmatic Visual Question Answering28 Jul 2024
- 2024Token-level Correlation-guided Compression for Efficient Multimodal Document Understanding19 Jul 2024
- 2024Characterizing Prompt Compression Methods for Long Context Inference11 Jul 2024
- 2024Focus on the Core: Efficient Attention via Pruned Token Compression for Document Classification3 Jun 2024
- 2024MELTing point: Mobile Evaluation of Language Transformers19 Mar 2024
- 2024Dynamic Memory Compression: Retrofitting LLMs for Accelerated Inference14 Mar 2024
- 2024Evolutionary Computation in the Era of Large Language Model: Survey and Roadmap18 Jan 2024
- 2024The What, Why, and How of Context Length Extension Techniques in Large Language Models -- A Detailed Survey15 Jan 2024
- 2023ConsistentEE: A Consistent and Hardness-Guided Early Exiting Method for Accelerating Language Models Inference19 Dec 2023
- 2023No-Skim: Towards Efficiency Robustness Evaluation on Skimming-based Language Models15 Dec 2023
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Merging2 Dec 2023
- 2023OrchestraLLM: Efficient Orchestration of Language Models for Dialogue State Tracking16 Nov 2023
- 2023DEED: Dynamic Early Exit on Decoder for Accelerating Encoder-Decoder Transformer Models15 Nov 2023
- 2023On Elastic Language Models13 Nov 2023
- 2023Improving Machine Translation with Large Language Models: A Preliminary Study with Cooperative Decoding6 Nov 2023
- 2023Variator: Accelerating Pre-trained Models with Plug-and-Play Compression Modules24 Oct 2023
- 2023Harnessing Attention Mechanisms: Efficient Sequence Reduction using Attention-based Autoencoders23 Oct 2023
- 2023Optimizing Retrieval-augmented Reader Models via Token Elimination20 Oct 2023
- 2023LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compression10 Oct 2023
- 2023LLMLingua: Compressing Prompts for Accelerated Inference of Large Language Models9 Oct 2023
- 2023Accelerating Transducers through Adjacent Token Merging28 Jun 2023
- 2023Constraint-aware and Ranking-distilled Token Pruning for Efficient Transformer Inference26 Jun 2023
- 2023CrossGET: Cross-Guided Ensemble of Tokens for Accelerating Vision-Language Transformers27 May 2023
- 2023Zero-TPrune: Zero-Shot Token Pruning through Leveraging of the Attention Graph in Pre-Trained Transformers27 May 2023
- 2023Do We Really Need a Large Number of Visual Prompts?26 May 2023
- 2023Infor-Coef: Information Bottleneck-based Dynamic Token Downsampling for Compact and Efficient language model21 May 2023
50 newest of 76 All 76 (JSON)
- 2022Token Merging: Your ViT But Fastercited by 995
- 2023LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compressioncited by 443
- 2022Confident Adaptive Language Modelingcited by 297
- 2021Learned Token Pruning for Transformerscited by 138
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Mergingcited by 123
- 2023Zero-TPrune: Zero-Shot Token Pruning through Leveraging of the Attention Graph in Pre-Trained Transformerscited by 61
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodalitycited by 49
- 2021TR-BERT: Dynamic Token Reduction for Accelerating BERT Inferencecited by 40
- 2024Characterizing Prompt Compression Methods for Long Context Inferencecited by 18
- 2021A Study on Token Pruning for ColBERTcited by 14
- 2023Accelerating Transducers through Adjacent Token Mergingcited by 14
- 2023ConsistentEE: A Consistent and Hardness-Guided Early Exiting Method for Accelerating Language Models Inferencecited by 13
12 most cited of 21 All 21 (JSON)
- 2026Sparse Token Routing in Efficient Transformers21 Aug 2026
- 2025Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality23 May 2025
- 2025Saliency-driven Dynamic Token Pruning for Large Language Models6 Apr 2025
- 2024Training Noise Token Pruning27 Nov 2024
- 2024Selection-p: Self-Supervised Task-Agnostic Prompt Compression for Faithfulness and Transferability15 Oct 2024
- 2024Practical token pruning for foundation models in few-shot conversational virtual assistant systems21 Aug 2024
- 2024AdaCoder: Adaptive Prompt Compression for Programmatic Visual Question Answering28 Jul 2024
- 2024Characterizing Prompt Compression Methods for Long Context Inference11 Jul 2024
- 2024Focus on the Core: Efficient Attention via Pruned Token Compression for Document Classification3 Jun 2024
- 2023ConsistentEE: A Consistent and Hardness-Guided Early Exiting Method for Accelerating Language Models Inference19 Dec 2023
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Merging2 Dec 2023
- 2023DEED: Dynamic Early Exit on Decoder for Accelerating Encoder-Decoder Transformer Models15 Nov 2023
12 newest of 21 All 21 (JSON)
- 2023CrossGET: Cross-Guided Ensemble of Tokens for Accelerating Vision-Language Transformerscited by 44
- 2024AIM: Adaptive Inference of Multi-Modal LLMs via Token Merging and Pruningcited by 43
- 2025Shifting AI Efficiency From Model-Centric to Data-Centric Compressioncited by 29
- 2024Token-level Correlation-guided Compression for Efficient Multimodal Document Understandingcited by 22
- 2024iLLaVA: An Image is Worth Fewer Than 1/3 Input Tokens in Large Multimodal Modelscited by 11
- 2025LightVLM: Acceleraing Large Multimodal Models with Pyramid Token Merging and KV Cache Compressioncited by 7
- 2025LightVLM: Acceleraing Large Multimodal Models with Pyramid Token Merging and KV Cache Compression30 Aug 2025
- 2025Shifting AI Efficiency From Model-Centric to Data-Centric Compression25 May 2025
- 2024iLLaVA: An Image is Worth Fewer Than 1/3 Input Tokens in Large Multimodal Models9 Dec 2024
- 2024AIM: Adaptive Inference of Multi-Modal LLMs via Token Merging and Pruning4 Dec 2024
- 2024Token-level Correlation-guided Compression for Efficient Multimodal Document Understanding19 Jul 2024
- 2023CrossGET: Cross-Guided Ensemble of Tokens for Accelerating Vision-Language Transformers27 May 2023
- 2025Training-free Token Reduction for Vision Mamba18 Jul 2025
- 2024Faster Vision Mamba is Rebuilt in Minutes via Merged Token Re-training17 Dec 2024
- 2024Deploying Foundation Model Powered Agent Services: A Survey18 Dec 2024
- 2025Rethinking Chain-of-Thought Reasoning for Videoscited by 8
- 2025Rethinking Chain-of-Thought Reasoning for Videos10 Dec 2025
- 2023LLMLingua: Compressing Prompts for Accelerated Inference of Large Language Modelscited by 279
- 2024Dynamic Memory Compression: Retrofitting LLMs for Accelerated Inferencecited by 115
- 2024Evolutionary Computation in the Era of Large Language Model: Survey and Roadmapcited by 89
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2021A Survey on Green Deep Learningcited by 52
- 2022A Survey on Model Compression and Acceleration for Pretrained Language Modelscited by 45
- 2024ElasticTok: Adaptive Tokenization for Image and Videocited by 44
- 2022Model Cascading: Towards Jointly Improving Efficiency and Accuracy of NLP Systemscited by 43
- 2024MELTing point: Mobile Evaluation of Language Transformerscited by 35
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2024The What, Why, and How of Context Length Extension Techniques in Large Language Models -- A Detailed Surveycited by 32
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 25
12 most cited of 42 All 42 (JSON)
- 2026Structured-Sparse Attention for Entity Tracking with Subquadratic Sequence Complexity21 May 2026
- 2025ResFormer: All-Time Reservoir Memory for Long Sequence Classification28 Sep 2025
- 2025Put Teacher in Student's Shoes: Cross-Distillation for Ultra-compact Model Compression Framework7 Jul 2025
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representation3 Mar 2025
- 2025CipherPrune: Efficient and Scalable Private Transformer Inference24 Feb 2025
- 2025Top-Theta Attention: Sparsifying Transformers by Compensated Thresholding12 Feb 2025
- 2024ImagePiece: Content-aware Re-tokenization for Efficient Image Recognition21 Dec 2024
- 2024Self-Satisfied: An end-to-end framework for SAT generation and prediction18 Oct 2024
- 2024ElasticTok: Adaptive Tokenization for Image and Video10 Oct 2024
- 2024MELTing point: Mobile Evaluation of Language Transformers19 Mar 2024
- 2024Dynamic Memory Compression: Retrofitting LLMs for Accelerated Inference14 Mar 2024
- 2024Evolutionary Computation in the Era of Large Language Model: Survey and Roadmap18 Jan 2024
12 newest of 42 All 42 (JSON)