Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applications

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Yanxiang, Xu, Zheng, Wu, Shanshan, Zhang, Yuanbo, Ramage, Daniel
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916756441792512
author Zhang, Yanxiang
Xu, Zheng
Wu, Shanshan
Zhang, Yuanbo
Ramage, Daniel
author_facet Zhang, Yanxiang
Xu, Zheng
Wu, Shanshan
Zhang, Yuanbo
Ramage, Daniel
contents Error correction is an important capability when applying large language models (LLMs) to facilitate user typing on mobile devices. In this paper, we use LLMs to synthesize a high-quality dataset of error correction pairs to evaluate and improve LLMs for mobile applications. We first prompt LLMs with error correction domain knowledge to build a scalable and reliable addition to the existing data synthesis pipeline. We then adapt the synthetic data distribution to match the mobile application domain by reweighting the samples. The reweighting model is learnt by predicting (a handful of) live A/B test metrics when deploying LLMs in production, given the LLM performance on offline evaluation data and scores from a small privacy-preserving on-device language model. Finally, we present best practices for mixing our synthetic data with other data sources to improve model performance on error correction in both offline evaluation and production live A/B testing.
format Preprint
id arxiv_https___arxiv_org_abs_2505_18488
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applications
Zhang, Yanxiang
Xu, Zheng
Wu, Shanshan
Zhang, Yuanbo
Ramage, Daniel
Machine Learning
Artificial Intelligence
Computation and Language
Error correction is an important capability when applying large language models (LLMs) to facilitate user typing on mobile devices. In this paper, we use LLMs to synthesize a high-quality dataset of error correction pairs to evaluate and improve LLMs for mobile applications. We first prompt LLMs with error correction domain knowledge to build a scalable and reliable addition to the existing data synthesis pipeline. We then adapt the synthetic data distribution to match the mobile application domain by reweighting the samples. The reweighting model is learnt by predicting (a handful of) live A/B test metrics when deploying LLMs in production, given the LLM performance on offline evaluation data and scores from a small privacy-preserving on-device language model. Finally, we present best practices for mixing our synthetic data with other data sources to improve model performance on error correction in both offline evaluation and production live A/B testing.
title Synthesizing and Adapting Error Correction Data for Mobile Large Language Model Applications
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2505.18488