Proofread: Fixes All Errors with One Tap
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
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 3,050
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,653
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,470
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 1,006
- 2023FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPUcited by 575
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 471
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 415
20 more
- 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Casescited by 223
- 2018Dear Sir or Madam, May I introduce the GYAFC Dataset: Corpus, Benchmarks and Metrics for Formality Style Transfercited by 211
- 2020GECToR -- Grammatical Error Correction: Tag, Not Rewritecited by 177
- 2021A Recipe For Arbitrary Text Style Transfer with Large Language Modelscited by 107
- 2017JFLEG: A Fluency Corpus and Benchmark for Grammatical Error Correctioncited by 97
- 2023ChatGPT or Grammarly? Evaluating ChatGPT on Grammatical Error Correction Benchmarkcited by 77
- 2023RewriteLM: An Instruction-Tuned Large Language Model for Text Rewritingcited by 67
- 2023An Emulator for Fine-Tuning Large Language Models using Small Language Modelscited by 61
- 2020Seq2Edits: Sequence Transduction Using Span-level Edit Operationscited by 58
- 2023Baby Llama: knowledge distillation from an ensemble of teachers trained on a small dataset with no performance penaltycited by 57
- 2021Heuristic-Guided Reinforcement Learningcited by 42
- 2020TextSETTR: Few-Shot Text Style Extraction and Tunable Targeted Restylingcited by 40
- 2020Parallel Data Augmentation for Formality Style Transfercited by 39
- 2021FRUIT: Faithfully Reflecting Updated Information in Textcited by 39
- 2022EdiT5: Semi-Autoregressive Text-Editing with T5 Warm-Startcited by 35
- 2020Unsupervised Paraphrasing via Deep Reinforcement Learningcited by 25
- 2024Prompting open-source and commercial language models for grammatical error correction of English learner textcited by 24
- 2022Text Generation with Text-Editing Modelscited by 21
- 2017Mobile Keyboard Input Decoding with Finite-State Transducerscited by 8
- 2023Towards an On-device Agent for Text Rewritingcited by 7
Built on by
Citing papers per year: 2024: 1; 2025: 3.
4 papers.
- 2025WRAVAL -- WRiting Assist eVALuation19 Dec 2025
- 2025Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applications24 May 2025
- 2025SPHERE: An Evaluation Card for Human-AI Systems24 Mar 2025
- 2024Rule by Rule: Learning with Confidence through Vocabulary Expansion30 Oct 2024
Similar, not cited
Close in wording, no citation either way
- 2025Towards Alignment-Centric Paradigm: A Survey of Instruction Tuning in Large Language Models
- 2025Harnessing Rule-Based Reinforcement Learning for Enhanced Grammatical Error Correction
- 2025Evaluating Robustness of Large Language Models Against Multilingual Typographical Errors
- 2026Is Human-Readable Text Necessary for Effective LLM Fine-Tuning?
- 2025LookAhead Tuning: Safer Language Models via Partial Answer Previews
- 2025Dr Genre: Reinforcement Learning from Decoupled LLM Feedback for Generic Text Rewriting
- 2023Automatically Correcting Large Language Models: Surveying the landscape of diverse self-correction strategies
- 2024RE-Adapt: Reverse Engineered Adaptation of Large Language Models