mlsanityData to 14 Oct 2026

Proofread: Fixes All Errors with One Tap

arXiv 2406.04523 ↗First posted 6 Jun 2024cs.CL cs.LGcited by 4

Not on the map. Why?

Abstract (arXiv)

The impressive capabilities in Large Language Models (LLMs) provide a powerful approach to reimagine users' typing experience. This paper demonstrates Proofread, a novel Gboard feature powered by a server-side LLM in Gboard, enabling seamless sentence-level and paragraph-level corrections with a single tap. We describe the complete system in this paper, from data generation, metrics design to model tuning and deployment. To obtain models with sufficient quality, we implement a careful data synthetic pipeline tailored to online use cases, design multifaceted metrics, employ a two-stage tuning approach to acquire the dedicated LLM for the feature: the Supervised Fine Tuning (SFT) for foundational quality, followed by the Reinforcement Learning (RL) tuning approach for targeted refinement. Specifically, we find sequential tuning on Rewrite and proofread tasks yields the best quality in SFT stage, and propose global and direct rewards in the RL tuning stage to seek further improvement. Extensive experiments on a human-labeled golden set showed our tuned PaLM2-XS model achieved 85.56% good ratio. We launched the feature to Pixel 8 devices by serving the model on TPU v5 in Google Cloud, with thousands of daily active users. Serving latency was significantly reduced by quantization, bucket inference, text segmentation, and speculative decoding. Our demo could be seen in hrefhttps://youtu.be/4ZdcuiwFU7IYoutube.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 32 of its 41 references that are in the map. First posted on arXiv 6 Jun 2024.

Builds on

32 of 41 references.

References · 32

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  4. 2022Constitutional AI: Harmlessness from AI Feedbackcited by 3,050
  5. 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  7. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
  8. 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,470
  9. 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
  10. 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 575
  11. 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 471
  12. 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 415
20 more
  1. 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Casescited by 223
  2. 2018Dear Sir or Madam, May I introduce the GYAFC Dataset: Corpus, Benchmarks and Metrics for Formality Style Transfercited by 211
  3. 2020GECToR -- Grammatical Error Correction: Tag, Not Rewritecited by 177
  4. 2021A Recipe For Arbitrary Text Style Transfer with Large Language Modelscited by 107
  5. 2017JFLEG: A Fluency Corpus and Benchmark for Grammatical Error Correctioncited by 97
  6. 2023ChatGPT or Grammarly? Evaluating ChatGPT on Grammatical Error Correction Benchmarkcited by 77
  7. 2023RewriteLM: An Instruction-Tuned Large Language Model for Text Rewritingcited by 67
  8. 2023An Emulator for Fine-Tuning Large Language Models using Small Language Modelscited by 61
  9. 2020Seq2Edits: Sequence Transduction Using Span-level Edit Operationscited by 58
  10. 2023Baby Llama: knowledge distillation from an ensemble of teachers trained on a small dataset with no performance penaltycited by 57
  11. 2021Heuristic-Guided Reinforcement Learningcited by 42
  12. 2020TextSETTR: Few-Shot Text Style Extraction and Tunable Targeted Restylingcited by 40
  13. 2020Parallel Data Augmentation for Formality Style Transfercited by 39
  14. 2021FRUIT: Faithfully Reflecting Updated Information in Textcited by 39
  15. 2022EdiT5: Semi-Autoregressive Text-Editing with T5 Warm-Startcited by 35
  16. 2020Unsupervised Paraphrasing via Deep Reinforcement Learningcited by 25
  17. 2024Prompting open-source and commercial language models for grammatical error correction of English learner textcited by 24
  18. 2022Text Generation with Text-Editing Modelscited by 21
  19. 2017Mobile Keyboard Input Decoding with Finite-State Transducerscited by 8
  20. 2023Towards an On-device Agent for Text Rewritingcited by 7

Built on by

Cited by year

Citing papers per year: 2024: 1; 2025: 3.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Towards Alignment-Centric Paradigm: A Survey of Instruction Tuning in Large Language Models
  2. 2025Harnessing Rule-Based Reinforcement Learning for Enhanced Grammatical Error Correction
  3. 2025Evaluating Robustness of Large Language Models Against Multilingual Typographical Errors
  4. 2026Is Human-Readable Text Necessary for Effective LLM Fine-Tuning?
  5. 2025LookAhead Tuning: Safer Language Models via Partial Answer Previews
  6. 2025Dr Genre: Reinforcement Learning from Decoupled LLM Feedback for Generic Text Rewriting
  7. 2023Automatically Correcting Large Language Models: Surveying the landscape of diverse self-correction strategies
  8. 2024RE-Adapt: Reverse Engineered Adaptation of Large Language Models