DistRL提升手机端智能体训练效率,让设备控制更快速准确。
DistRL: An Asynchronous Distributed Reinforcement Learning Framework for On-Device Control Agents
- 采用集中训练、分散采样的异步框架,适应动态交互场景。
- 训练效率提升3倍,数据收集速度加快2.4倍,成功率提高20%。
- 适合移动设备上需要实时学习的智能体系统,尤其适合资源受限场景。
手机端控制智能体需响应用户请求,实现无缝交互。将多模态大语言模型(MLLM)集成至这些智能体可增强其理解与执行复杂指令的能力。然而,由于数据有限且在线训练效率低,对MLLM进行端侧微调面临挑战。本文提出DistRL框架,通过集中训练与去中心化数据采集,提升移动端控制智能体在线强化学习微调的效率。框架结合自研的强化学习算法,有效平衡探索与数据优先利用,保障训练稳定可靠。实验表明,相较于领先的同步多机方法,DistRL平均训练效率提升3倍,数据收集速度加快2.4倍;在公开基准测试中,成功率达20%相对提升,且训练时间不变。结果验证了DistRL在真实设备控制任务中的高效性与可扩展性。
原文摘要 · Abstract (English)
On-device control agents, especially on mobile devices, are responsible for operating mobile devices to fulfill users' requests, enabling seamless and intuitive interactions. Integrating Multimodal Large Language Models (MLLMs) into these agents enhances their ability to understand and execute complex commands, thereby improving user experience. However, fine-tuning MLLMs for on-device control presents significant challenges due to limited data availability and inefficient online training processes. This paper introduces DistRL, a novel framework designed to enhance the efficiency of online RL fine-tuning for mobile device control agents. DistRL employs centralized training and decentralized data acquisition to ensure efficient fine-tuning in the context of dynamic online interactions. Additionally, the framework is backed by our tailor-made RL algorithm, which effectively balances exploration with the prioritized utilization of collected data to ensure stable and robust training. Our experiments show that, on average, DistRL delivers a 3X improvement in training efficiency and enables training data collection 2.4X faster than the leading synchronous multi-machine methods. Notably, after training, DistRL achieves a 20% relative improvement in success rate compared to state-of-the-art methods on general Android tasks from an open benchmark, significantly outperforming existing approaches while maintaining the same training time. These results validate DistRL as a scalable and efficient solution, offering substantial improvements in both training efficiency and agent performance for real-world, in-the-wild device control tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。