Elbert: Fast Albert with Confidence-Window Based Early Exit
On the map: Efficient LLM inference and serving → Early exit and edge inference
Abstract (arXiv)
Despite the great success in Natural Language Processing (NLP) area, large pre-trained language models like BERT are not well-suited for resource-constrained or real-time applications owing to the large number of parameters and slow inference speed. Recently, compressing and accelerating BERT have become important topics. By incorporating a parameter-sharing strategy, ALBERT greatly reduces the number of parameters while achieving competitive performance. Nevertheless, ALBERT still suffers from a long inference time. In this work, we propose the ELBERT, which significantly improves the average inference speed compared to ALBERT due to the proposed confidence-window based early exit mechanism, without introducing additional parameters or extra training overhead. Experimental results show that ELBERT achieves an adaptive inference speedup varying from 2× to 10× with negligible accuracy degradation compared to ALBERT on various datasets. Besides, ELBERT achieves higher accuracy than existing early exit methods used for accelerating BERT under the same computation cost. Furthermore, to understand the principle of the early exit mechanism, we also visualize the decision-making process of it in ELBERT.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Placed in Early exit and edge inference, one of 12 directions in Efficient LLM inference and serving. Builds on 26 of its 29 references that are in the map. First posted on arXiv 1 Jul 2021.
Builds on
26 of 29 references.
References · 26
- 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
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,801
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2019Once-for-All: Train One Network and Specialize it for Efficient Deploymentcited by 778
14 more
- 2017BranchyNet: Fast Inference via Early Exiting from Deep Neural Networkscited by 591
- 2019Patient Knowledge Distillation for BERT Model Compressioncited by 502
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 434
- 2019Q-BERT: Hessian Based Ultra Low Precision Quantization of BERTcited by 355
- 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 322
- 2019Revealing the Dark Secrets of BERTcited by 300
- 2019Q8BERT: Quantized 8Bit BERTcited by 296
- 2019A Multiscale Visualization of Attention in the Transformer Modelcited by 292
- 2020BERT Loses Patience: Fast and Robust Inference with Early Exitcited by 267
- 2020FastBERT: a Self-distilling BERT with Adaptive Inference Timecited by 224
- 2014Compressing Deep Convolutional Networks using Vector Quantizationcited by 189
- 2020Compressing BERT: Studying the Effects of Weight Pruning on Transfer Learningcited by 184
- 2019Distilling Task-Specific Knowledge from BERT into Simple Neural Networkscited by 152
- 2020The Right Tool for the Job: Matching Model and Instance Complexitiescited by 125
Built on by
Citing papers per year: 2022: 6; 2023: 3; 2024: 1; 2025: 3; 2026: 1.
14 papers.
Sort:
By area:
- 2024Efficient Tuning and Inference for Large Language Models on Textual Graphscited by 54
- 2022Text Revealer: Private Text Reconstruction via Model Inversion Attacks against Transformerscited by 28
- 2023Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Servingcited by 15
- 2022HuBERT-EE: Early Exiting HuBERT for Efficient Speech Recognitioncited by 13
- 2022BEBERT: Efficient and Robust Binary Ensemble BERTcited by 13
- 2023Subnetwork-to-go: Elastic Neural Network with Dynamic Training and Customizable Inferencecited by 9
- 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depthcited by 9
- 2022Avoid Overthinking in Self-Supervised Models for Speech Recognitioncited by 5
- 2023BERT Lost Patience Won't Be Robust to Adversarial Slowdowncited by 4
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devicescited by 3
- 2022Latency Adjustable Transformer Encoder for Language Understandingcited by 1
- 2022Fast and Accurate FSA System Using ELBERT: An Efficient and Lightweight BERTcited by 1
- 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depth2 Mar 2026
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devices21 Sep 2025
- 2025Efficient Large Language Models with Zero-Shot Adjustable Acceleration1 Sep 2025
- 2025EPEE: Towards Efficient and Effective Foundation Models in Biomedicine3 Mar 2025
- 2024Efficient Tuning and Inference for Large Language Models on Textual Graphs28 Jan 2024
- 2023Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving8 Dec 2023
- 2023Subnetwork-to-go: Elastic Neural Network with Dynamic Training and Customizable Inference6 Dec 2023
- 2023BERT Lost Patience Won't Be Robust to Adversarial Slowdown29 Oct 2023
- 2022Fast and Accurate FSA System Using ELBERT: An Efficient and Lightweight BERT16 Nov 2022
- 2022Avoid Overthinking in Self-Supervised Models for Speech Recognition1 Nov 2022
- 2022BEBERT: Efficient and Robust Binary Ensemble BERT28 Oct 2022
- 2022Text Revealer: Private Text Reconstruction via Model Inversion Attacks against Transformers21 Sep 2022
2 more
- 2022Text Revealer: Private Text Reconstruction via Model Inversion Attacks against Transformerscited by 28
- 2022BEBERT: Efficient and Robust Binary Ensemble BERTcited by 13
- 2023Subnetwork-to-go: Elastic Neural Network with Dynamic Training and Customizable Inferencecited by 9
- 2023BERT Lost Patience Won't Be Robust to Adversarial Slowdowncited by 4
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devicescited by 3
- 2022Latency Adjustable Transformer Encoder for Language Understandingcited by 1
- 2022Fast and Accurate FSA System Using ELBERT: An Efficient and Lightweight BERTcited by 1
- 2025EPEE: Towards Efficient and Effective Foundation Models in Biomedicinecited by 1
- 2025Efficient Large Language Models with Zero-Shot Adjustable Accelerationcited by 0
- 2025FedEL: Federated Elastic Learning for Heterogeneous Devices21 Sep 2025
- 2025Efficient Large Language Models with Zero-Shot Adjustable Acceleration1 Sep 2025
- 2025EPEE: Towards Efficient and Effective Foundation Models in Biomedicine3 Mar 2025
- 2023Subnetwork-to-go: Elastic Neural Network with Dynamic Training and Customizable Inference6 Dec 2023
- 2023BERT Lost Patience Won't Be Robust to Adversarial Slowdown29 Oct 2023
- 2022Fast and Accurate FSA System Using ELBERT: An Efficient and Lightweight BERT16 Nov 2022
- 2022BEBERT: Efficient and Robust Binary Ensemble BERT28 Oct 2022
- 2022Text Revealer: Private Text Reconstruction via Model Inversion Attacks against Transformers21 Sep 2022
- 2022Latency Adjustable Transformer Encoder for Language Understanding10 Jan 2022
Similar, not cited
Close in wording, no citation either way
- 2023SmartBERT: A Promotion of Dynamic Early Exiting Mechanism for Accelerating BERT Inference
- 2021RomeBERT: Robust Training of Multi-Exit BERT
- 2024CEEBERT: Cross-Domain Inference in Early Exit BERT
- 2025NeoBERT: A Next-Generation BERT
- 2024DE3-BERT: Distance-Enhanced Early Exiting for BERT based on Prototypical Networks
- 2021Towards Efficient NLP: A Standard Evaluation and A Strong Baseline
- 2022The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Models
- 2021Accelerating BERT Inference for Sequence Labeling via Early-Exit