arXiv:2502.04392cs.CLcs.AI2025-02被引 40

用小模型+大模型协作,让手机端助手更省力、更快

Division-of-Thoughts: Harnessing Hybrid Language Model Synergy for Efficient On-Device Agents

  • 把任务拆成小块,由小模型和大模型分工处理
  • 推理时间减少66.12%,API成本降低83.57%,准确率不降
  • 适合资源受限设备上的智能助手应用

网络内容激增使本地AI助手成为应对在线任务复杂性的关键。尽管大语言模型具备涌现推理能力,但在资源受限的本地设备上部署全规模大模型仍具挑战。本文提出分脑协同框架Division-of-Thoughts(DoT),通过本地小模型(SLMs)与云端大模型(LLMs)协同工作,实现高效推理。DoT利用任务分解器激发语言模型的规划能力,将用户查询拆分为子任务;通过任务调度器分析子任务间依赖关系,构建依赖图以支持并行推理和关键步骤识别。为根据子任务难度动态分配模型,引入可插拔适配器(Plug-and-Play Adapter),该适配器不修改小模型参数,仅附加于其上。通过仅依赖任务执行反馈的自强化训练方法提升适配器的任务分配能力。在多个基准测试中,DoT显著降低大模型使用成本,平均推理时间减少66.12%,API成本降低83.57%,同时保持与最优基线相当的推理准确率。

原文摘要 · Abstract (English)

The rapid expansion of web content has made on-device AI assistants indispensable for helping users manage the increasing complexity of online tasks. The emergent reasoning ability in large language models offer a promising path for next-generation on-device AI agents. However, deploying full-scale Large Language Models (LLMs) on resource-limited local devices is challenging. In this paper, we propose Division-of-Thoughts (DoT), a collaborative reasoning framework leveraging the synergy between locally deployed Smaller-scale Language Models (SLMs) and cloud-based LLMs. DoT leverages a Task Decomposer to elicit the inherent planning abilities in language models to decompose user queries into smaller sub-tasks, which allows hybrid language models to fully exploit their respective strengths. Besides, DoT employs a Task Scheduler to analyze the pair-wise dependency of sub-tasks and create a dependency graph, facilitating parallel reasoning of sub-tasks and the identification of key steps. To allocate the appropriate model based on the difficulty of sub-tasks, DoT leverages a Plug-and-Play Adapter, which is an additional task head attached to the SLM that does not alter the SLM's parameters. To boost adapter's task allocation capability, we propose a self-reinforced training method that relies solely on task execution feedback. Extensive experiments on various benchmarks demonstrate that our DoT significantly reduces LLM costs while maintaining competitive reasoning accuracy. Specifically, DoT reduces the average reasoning time and API costs by 66.12% and 83.57%, while achieving comparable reasoning accuracy with the best baseline methods.

智能助手模型协同边缘计算任务分解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。