Instantaneous Grammatical Error Correction with Shallow Aggressive Decoding
On the map: Efficient LLM inference and serving → Speculative decoding
Abstract (arXiv)
In this paper, we propose Shallow Aggressive Decoding (SAD) to improve the online inference efficiency of the Transformer for instantaneous Grammatical Error Correction (GEC). SAD optimizes the online inference efficiency for GEC by two innovations: 1) it aggressively decodes as many tokens as possible in parallel instead of always decoding only one token in each step to improve computational parallelism; 2) it uses a shallow decoder instead of the conventional Transformer architecture with balanced encoder-decoder depth to reduce the computational cost during inference. Experiments in both English and Chinese GEC benchmarks show that aggressive decoding could yield the same predictions as greedy decoding but with a significant speedup for online inference. Its combination with the shallow decoder could offer an even higher online inference speedup over the powerful Transformer baseline without quality loss. Not only does our approach allow a single model to achieve the state-of-the-art results in English GEC benchmarks: 66.4 F0.5 in the CoNLL-14 and 72.9 F0.5 in the BEA-19 test set with an almost 10x online inference speedup over the Transformer-big model, but also it is easily adapted to other languages. Our code is available at https://github.com/AutoTemp/Shallow-Aggressive-Decoding.
Key facts. Cited by 53 papers in the map (data through 7 Oct 2026). Placed in Speculative decoding, one of 12 directions in Efficient LLM inference and serving. Builds on 22 of its 38 references that are in the map. First posted on arXiv 9 Jun 2021.
Builds on
22 of 38 references.
References · 22
- 2017Attention Is All You Needcited by 59,932
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,801
- 2018Blockwise Parallel Decoding for Deep Autoregressive Modelscited by 318
- 2019Levenshtein Transformercited by 197
- 2020GECToR -- Grammatical Error Correction: Tag, Not Rewritecited by 177
- 2019Encode, Tag, Realize: High-Precision Text Editingcited by 99
10 more
- 2020Fully Non-autoregressive Neural Machine Translation: Tricks of the Tradecited by 99
- 2019Parallel Iterative Edit Models for Local Sequence Transductioncited by 81
- 2020Encoder-Decoder Models Can Benefit from Pre-trained Masked Language Models in Grammatical Error Correctioncited by 59
- 2020Seq2Edits: Sequence Transduction Using Span-level Edit Operationscited by 58
- 2019Corpora Generation for Grammatical Error Correctioncited by 53
- 2020Improving the Efficiency of Grammatical Error Correction with Erroneous Span Detection and Correctioncited by 24
- 2018Reaching Human-level Performance in Automatic Grammatical Error Correction: An Empirical Studycited by 22
- 2019Improving Grammatical Error Correction with Machine Translation Pairscited by 17
- 2019Sequence-to-sequence Pre-training with Data Augmentation for Sentence Rewritingcited by 12
- 2021An Efficient Transformer Decoder with Compressed Sub-layerscited by 10
Built on by
Citing papers per year: 2021: 1; 2022: 9; 2023: 15; 2024: 10; 2025: 11; 2026: 7.
53 papers.
Sort:
By area:
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 564
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 362
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 318
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decodingcited by 266
- 2022Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generationcited by 169
- 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillationcited by 160
- 2023Accelerating Transformer Inference for Translation via Parallel Decodingcited by 123
- 2021CPT: A Pre-Trained Unbalanced Transformer for Both Chinese Language Understanding and Generationcited by 81
- 2023Inference with Reference: Lossless Acceleration of Large Language Modelscited by 63
- 2025Judge Decoding: Faster Speculative Sampling Requires Going Beyond Model Alignmentcited by 59
- 2022Grammatical Error Correction: A Survey of the State of the Artcited by 43
38 more
- 2022EdiT5: Semi-Autoregressive Text-Editing with T5 Warm-Startcited by 35
- 2022SynGEC: Syntax-Enhanced Grammatical Error Correction with a Tailored GEC-Oriented Parsercited by 32
- 2025Accelerated Diffusion Models via Speculative Samplingcited by 30
- 2023UniTabE: A Universal Pretraining Protocol for Tabular Foundation Model in Data Sciencecited by 28
- 2022EdgeFormer: A Parameter-Efficient Transformer for On-Device Seq2seq Generationcited by 21
- 2022A Unified Strategy for Multilingual Grammatical Error Correction with Pre-trained Cross-Lingual Language Modelcited by 16
- 2023Disentangled Phonetic Representation for Chinese Spelling Correctioncited by 14
- 2024Speculative Decoding with CTC-based Draft Model for LLM Inference Accelerationcited by 14
- 2025L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Modelscited by 14
- 2023Revisiting Event Argument Extraction: Can EAE Models Learn Better When Being Aware of Event Co-occurrences?cited by 10
- 2022Type-Driven Multi-Turn Corrections for Grammatical Error Correctioncited by 9
- 2023Non-autoregressive Text Editing with Copy-aware Latent Alignmentscited by 8
- 2023Improving Seq2Seq Grammatical Error Correction via Decoding Interventionscited by 8
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUscited by 6
- 2024Improving Grammatical Error Correction via Contextual Data Augmentationcited by 6
- 2025Jakiro: Boosting Speculative Decoding with Decoupled Multi-Head via MoEcited by 6
- 2026Speculative Decoding with a Speculative Vocabularycited by 6
- 2026HSD: Training-Free Acceleration for Document Parsing Vision-Language Models with Hierarchical Speculative Decodingcited by 5
- 2022CSynGEC: Incorporating Constituent-based Syntax for Grammatical Error Correction with a Tailored GEC-Oriented Parsercited by 4
- 2024Context-Aware Assistant Selection for Improved Inference Acceleration with Large Language Modelscited by 4
- 2025SpecVLM: Fast Speculative Decoding in Vision-Language Modelscited by 4
- 2025DynaSpec: Context-aware Dynamic Speculative Sampling for Large-Vocabulary Language Modelscited by 4
- 2025Steering Pretrained Drafters during Speculative Decodingcited by 4
- 2023Does Correction Remain A Problem For Large Language Models?cited by 3
- 2023GEC-DePenD: Non-Autoregressive Grammatical Error Correction with Decoupled Permutation and Decodingcited by 3
- 2023MixEdit: Revisiting Data Augmentation and Beyond for Grammatical Error Correctioncited by 2
- 2023Beyond Hard Samples: Robust and Effective Grammatical Error Correction with Cycle Self-Augmentingcited by 2
- 2024Decoding at the Speed of Thought: Harnessing Parallel Decoding of Lexical Units for LLMscited by 2
- 2025OverFill: Two-Stage Models for Efficient Language Model Decodingcited by 2
- 2023Bidirectional Transformer Reranker for Grammatical Error Correctioncited by 1
- 2023RedPenNet for Grammatical Error Correction: Outputs to Tokens, Attentions to Spanscited by 1
- 2023Grammatical Error Correction via Mixed-Grained Weighted Trainingcited by 1
- 2024Detection-Correction Structure via General Language Model for Grammatical Error Correctioncited by 1
- 2024S2D: Sorted Speculative Decoding For More Efficient Deployment of Nested Large Language Modelscited by 1
- 2025Tutorial Proposal: Speculative Decoding for Efficient LLM Inferencecited by 1
- 2025Cross-Attention Speculative Decodingcited by 0
- 2026Encode Errors: Representational Retrieval of In-Context Demonstrations for Multilingual Grammatical Error Correctioncited by 0
- 2026A Generalized Formalism of Auto-Regressive Decoding for Speech Processingcited by 0
50 most cited of 53 All 53 (JSON)
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Models14 Aug 2026
- 2026CausalGate: Causal Importance Distillation for Transformer Module Pruning21 Jul 2026
- 2026Fully Rotation-Equivariant Spectral-Spatial Learning for Multispectral Object Detection6 Jul 2026
- 2026A Generalized Formalism of Auto-Regressive Decoding for Speech Processing16 Jun 2026
- 2026Encode Errors: Representational Retrieval of In-Context Demonstrations for Multilingual Grammatical Error Correction13 Jun 2026
- 2026Speculative Decoding with a Speculative Vocabulary14 Feb 2026
- 2026HSD: Training-Free Acceleration for Document Parsing Vision-Language Models with Hierarchical Speculative Decoding13 Feb 2026
- 2025Steering Pretrained Drafters during Speculative Decoding13 Nov 2025
- 2025DynaSpec: Context-aware Dynamic Speculative Sampling for Large-Vocabulary Language Models11 Oct 2025
- 2025SpecVLM: Fast Speculative Decoding in Vision-Language Models15 Sep 2025
- 2025OverFill: Two-Stage Models for Efficient Language Model Decoding11 Aug 2025
- 2025Cross-Attention Speculative Decoding30 May 2025
38 more
- 2025L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Models23 May 2025
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Test3 Mar 2025
- 2025Tutorial Proposal: Speculative Decoding for Efficient LLM Inference1 Mar 2025
- 2025Jakiro: Boosting Speculative Decoding with Decoupled Multi-Head via MoE10 Feb 2025
- 2025Judge Decoding: Faster Speculative Sampling Requires Going Beyond Model Alignment31 Jan 2025
- 2025Accelerated Diffusion Models via Speculative Sampling9 Jan 2025
- 2024Speculative Decoding with CTC-based Draft Model for LLM Inference Acceleration25 Nov 2024
- 2024Context-Aware Assistant Selection for Improved Inference Acceleration with Large Language Models16 Aug 2024
- 2024S2D: Sorted Speculative Decoding For More Efficient Deployment of Nested Large Language Models2 Jul 2024
- 2024Improving Grammatical Error Correction via Contextual Data Augmentation25 Jun 2024
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Trees24 Jun 2024
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUs30 May 2024
- 2024Detection-Correction Structure via General Language Model for Grammatical Error Correction28 May 2024
- 2024Decoding at the Speed of Thought: Harnessing Parallel Decoding of Lexical Units for LLMs24 May 2024
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertainty26 Jan 2024
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding15 Jan 2024
- 2023Grammatical Error Correction via Mixed-Grained Weighted Training23 Nov 2023
- 2023GEC-DePenD: Non-Autoregressive Grammatical Error Correction with Decoupled Permutation and Decoding14 Nov 2023
- 2023Improving Seq2Seq Grammatical Error Correction via Decoding Interventions23 Oct 2023
- 2023Beyond Hard Samples: Robust and Effective Grammatical Error Correction with Cycle Self-Augmenting20 Oct 2023
- 2023MixEdit: Revisiting Data Augmentation and Beyond for Grammatical Error Correction18 Oct 2023
- 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillation12 Oct 2023
- 2023Non-autoregressive Text Editing with Copy-aware Latent Alignments11 Oct 2023
- 2023RedPenNet for Grammatical Error Correction: Outputs to Tokens, Attentions to Spans19 Sep 2023
- 2023Does Correction Remain A Problem For Large Language Models?3 Aug 2023
- 2023UniTabE: A Universal Pretraining Protocol for Tabular Foundation Model in Data Science18 Jul 2023
- 2023Revisiting Event Argument Extraction: Can EAE Models Learn Better When Being Aware of Event Co-occurrences?1 Jun 2023
- 2023Disentangled Phonetic Representation for Chinese Spelling Correction24 May 2023
- 2023Bidirectional Transformer Reranker for Grammatical Error Correction22 May 2023
- 2023Accelerating Transformer Inference for Translation via Parallel Decoding17 May 2023
- 2023Inference with Reference: Lossless Acceleration of Large Language Models10 Apr 2023
- 2022Fast Inference from Transformers via Speculative Decoding30 Nov 2022
- 2022CSynGEC: Incorporating Constituent-based Syntax for Grammatical Error Correction with a Tailored GEC-Oriented Parser15 Nov 2022
- 2022Grammatical Error Correction: A Survey of the State of the Art9 Nov 2022
- 2022SynGEC: Syntax-Enhanced Grammatical Error Correction with a Tailored GEC-Oriented Parser22 Oct 2022
- 2022EdiT5: Semi-Autoregressive Text-Editing with T5 Warm-Start24 May 2022
- 2022Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generation30 Mar 2022
- 2022Type-Driven Multi-Turn Corrections for Grammatical Error Correction17 Mar 2022
50 newest of 53 All 53 (JSON)
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 362
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 318
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decodingcited by 266
- 2022Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generationcited by 169
- 2023DistillSpec: Improving Speculative Decoding via Knowledge Distillationcited by 160
- 2023Accelerating Transformer Inference for Translation via Parallel Decodingcited by 123
- 2024Speculative Decoding with CTC-based Draft Model for LLM Inference Accelerationcited by 14
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUscited by 6
- 2026Speculative Decoding with a Speculative Vocabularycited by 6
- 2024Context-Aware Assistant Selection for Improved Inference Acceleration with Large Language Modelscited by 4
- 2025DynaSpec: Context-aware Dynamic Speculative Sampling for Large-Vocabulary Language Modelscited by 4
12 most cited of 16 All 16 (JSON)
- 2026Speculative Decoding with a Speculative Vocabulary14 Feb 2026
- 2025Steering Pretrained Drafters during Speculative Decoding13 Nov 2025
- 2025DynaSpec: Context-aware Dynamic Speculative Sampling for Large-Vocabulary Language Models11 Oct 2025
- 2025Cross-Attention Speculative Decoding30 May 2025
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Test3 Mar 2025
- 2025Tutorial Proposal: Speculative Decoding for Efficient LLM Inference1 Mar 2025
- 2024Speculative Decoding with CTC-based Draft Model for LLM Inference Acceleration25 Nov 2024
- 2024Context-Aware Assistant Selection for Improved Inference Acceleration with Large Language Models16 Aug 2024
- 2024S2D: Sorted Speculative Decoding For More Efficient Deployment of Nested Large Language Models2 Jul 2024
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Trees24 Jun 2024
- 2024S3D: A Simple and Cost-Effective Self-Speculative Decoding Scheme for Low-Memory GPUs30 May 2024
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding15 Jan 2024
12 newest of 16 All 16 (JSON)
- 2026HSD: Training-Free Acceleration for Document Parsing Vision-Language Models with Hierarchical Speculative Decodingcited by 5
- 2025SpecVLM: Fast Speculative Decoding in Vision-Language Modelscited by 4
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Modelscited by 0
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Models14 Aug 2026
- 2026HSD: Training-Free Acceleration for Document Parsing Vision-Language Models with Hierarchical Speculative Decoding13 Feb 2026
- 2025SpecVLM: Fast Speculative Decoding in Vision-Language Models15 Sep 2025
- 2025Accelerated Diffusion Models via Speculative Samplingcited by 30
- 2025Accelerated Diffusion Models via Speculative Sampling9 Jan 2025
- 2021CPT: A Pre-Trained Unbalanced Transformer for Both Chinese Language Understanding and Generationcited by 81
- 2023Inference with Reference: Lossless Acceleration of Large Language Modelscited by 63
- 2022Grammatical Error Correction: A Survey of the State of the Artcited by 43
- 2022EdiT5: Semi-Autoregressive Text-Editing with T5 Warm-Startcited by 35
- 2022SynGEC: Syntax-Enhanced Grammatical Error Correction with a Tailored GEC-Oriented Parsercited by 32
- 2023UniTabE: A Universal Pretraining Protocol for Tabular Foundation Model in Data Sciencecited by 28
- 2022EdgeFormer: A Parameter-Efficient Transformer for On-Device Seq2seq Generationcited by 21
- 2022A Unified Strategy for Multilingual Grammatical Error Correction with Pre-trained Cross-Lingual Language Modelcited by 16
- 2023Disentangled Phonetic Representation for Chinese Spelling Correctioncited by 14
- 2025L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Modelscited by 14
- 2023Revisiting Event Argument Extraction: Can EAE Models Learn Better When Being Aware of Event Co-occurrences?cited by 10
- 2022Type-Driven Multi-Turn Corrections for Grammatical Error Correctioncited by 9
12 most cited of 29 All 29 (JSON)
- 2026CausalGate: Causal Importance Distillation for Transformer Module Pruning21 Jul 2026
- 2026Fully Rotation-Equivariant Spectral-Spatial Learning for Multispectral Object Detection6 Jul 2026
- 2026A Generalized Formalism of Auto-Regressive Decoding for Speech Processing16 Jun 2026
- 2026Encode Errors: Representational Retrieval of In-Context Demonstrations for Multilingual Grammatical Error Correction13 Jun 2026
- 2025OverFill: Two-Stage Models for Efficient Language Model Decoding11 Aug 2025
- 2025L-MTP: Leap Multi-Token Prediction Beyond Adjacent Context for Large Language Models23 May 2025
- 2024Improving Grammatical Error Correction via Contextual Data Augmentation25 Jun 2024
- 2024Detection-Correction Structure via General Language Model for Grammatical Error Correction28 May 2024
- 2023Grammatical Error Correction via Mixed-Grained Weighted Training23 Nov 2023
- 2023GEC-DePenD: Non-Autoregressive Grammatical Error Correction with Decoupled Permutation and Decoding14 Nov 2023
- 2023Improving Seq2Seq Grammatical Error Correction via Decoding Interventions23 Oct 2023
- 2023Beyond Hard Samples: Robust and Effective Grammatical Error Correction with Cycle Self-Augmenting20 Oct 2023
12 newest of 29 All 29 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Generation Meets Verification: Accelerating Large Language Model Inference with Smart Parallel Auto-Correct Decoding
- 2024ChatLang-8: An LLM-Based Synthetic Data Generation Framework for Grammatical Error Correction
- 2024Accelerating Large Language Model Inference with Self-Supervised Early Exits
- 2023Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding
- 2024Closer Look at Efficient Inference Methods: A Survey of Speculative Decoding
- 2022Confident Adaptive Language Modeling
- 2023Is ChatGPT a Highly Fluent Grammatical Error Correction System? A Comprehensive Evaluation
- 2025Accelerating Large Language Model Inference via Early-Exiting Algorithms