In plain words: A phone keyboard that uses a language model to write in your own style, learning from your typing history through layered memory while keeping data on the device. Unlike ordinary keyboards that only suggest the next word, it generates personalized text in real time.
Abstract · HUOZIIME: An On-Device LLM-enhanced Input Method for Deep Personalization
Mobile input method editors (IMEs) are the primary interface for text input, yet they remain constrained to manual typing and struggle to produce personalized text. While lightweight large language models (LLMs) make on-device auxiliary generation feasible, enabling deeply personalized, privacy-preserving, and real-time generative IMEs poses fundamental challenges.To this end, we present HUOZIIME, a personalized on-device IME powered by LLM. We endow HUOZIIME with initial human-like prediction ability by post-training a base LLM on synthesized personalization data. Notably, a hierarchical memory mechanism is designed to continually capture and leverage user-specific input history. Furthermore, we perform systemic optimizations tailored to on-device LLMbased IME deployment, ensuring efficient and responsive operation under mobile constraints.Experiments demonstrate efficient on-device execution and high-fidelity memory-driven personalization. Code and package are available at https://github.com/Shan-HIT/HuoziIME.
Baocai Shan, Yuzhuang Xu, Wanxiang Che
arXiv:2604.14159 · cs.CL, cs.AI · submitted Mar 23, 2026
abstract · pdf · html