English

LiteGUI: Distilling Compact GUI Agents with Reinforcement Learning

Artificial Intelligence 2026-05-11 v1 Machine Learning

Abstract

Developing lightweight, on-device vision-language GUI agents is essential for efficient cross-platform automated interaction. However, current on-device agents are constrained by limited model capacity, and further performance improvements remain urgently needed. Traditional Supervised Fine-Tuning (SFT) for small-scale models often leads to overfitting, catastrophic forgetting and policy rigidity, and thus fails to fully address these challenges. In this work, we propose a novel SFT-free training paradigm that significantly enhances the performance of small-scale models. We first present the initial systematic integration of generalized knowledge distillation into the GUI agent domain via Guided On-policy Distillation. By incorporating oracle reference trajectories together with a dynamic retrieval mechanism, our method reduces hallucinations and mitigates the cognitive misalignment inherent in multi-solution GUI tasks. Building on this foundation, we further introduce a Multi-solution Dual-level GRPO framework that jointly aligns macro-level subtask planning with micro-level execution matching, thereby improving exploration in long-horizon GUI agent scenarios. In addition, we construct an automated data generation pipeline to synthesize GUI task trajectories with rich multi-solution annotations. Extensive experiments show that our method achieves state-of-the-art performance among lightweight models while remaining competitive with substantially larger-scale models across all benchmarks. Ablation studies further demonstrate that structured on-policy distillation and multi-solution dual-level exploration can fully unlock the capabilities of 2B/3B scale agents, surpassing the performance limits of conventional imitation learning.

Keywords

Cite

@article{arxiv.2605.07505,
  title  = {LiteGUI: Distilling Compact GUI Agents with Reinforcement Learning},
  author = {Yubin Wu and Zicheng Cai and Liping Ning and Hua Wang and Zhi Chen and Yaohua Tang and Hao Chen},
  journal= {arXiv preprint arXiv:2605.07505},
  year   = {2026}
}
R2 v1 2026-07-01T12:57:23.296Z