让手机模型自己判断何时求助云端,提升推理效率与准确率
Bridging On-Device and Cloud LLMs for Collaborative Reasoning: A Unified Methodology for Local Routing and Post-Training
- 用强化学习训练手机模型自主判断是否调用云端
- 在多个推理任务上接近全量云端模型表现
- 适合移动端大模型部署与低延迟推理场景
设备-云协同为部署大语言模型带来可能,利用轻量级本地模型提升效率,依赖强大云端模型实现优质推理。核心挑战在于:对每个输入请求,应决定是本地处理还是卸载至云端。现有方法多依赖外部路由机制,难以仅凭提示词判断任务难度,尤其在复杂推理任务中表现不佳。针对此问题,我们提出让本地 LLM 在推理时自主决策是否调用云端支持,通过基于强化学习的后训练实现。将本地 LLM 后训练建模为奖励最大化问题,设计分层奖励以鼓励本地求解并合理卸载。为解决该问题,我们开发了一种算法,包含群体级策略梯度以稳定优化,并引入自适应提示过滤提供互补学习信号,缓解策略坍塌(即仅本地或仅云端执行)。在多个推理基准上,对设备规模的 LLaMA 与 Qwen 模型进行大量实验表明,本方法持续优于基线,显著缩小与完整云端 LLM 的差距。
原文摘要 · Abstract (English)
Device-cloud collaboration holds promise for deploying large language models (LLMs), leveraging lightweight on-device models for efficiency while relying on powerful cloud models for superior reasoning. A central challenge in this setting is determining, for each incoming query, whether it should be processed locally or offloaded to the cloud. Existing approaches typically rely on external routers, which often struggle to determine difficulty from the prompt itself, especially for tasks involving complex reasoning. Motivated by this limitation, we propose enabling on-device LLMs to decide internally whether to invoke cloud assistance at inference time, with this capability instilled through reinforcement learning based post-training. Casting on-device LLM post-training as a reward maximization problem, we design hierarchical rewards to encourage local problem solving and judicious cloud offloading. To solve the resulting problem, we develop an algorithm featuring a group-level policy gradient that stabilizes optimization, together with adaptive prompt filtering that provides complementary learning signals to mitigate policy collapse (i.e., exclusive local execution or exclusive cloud offloading). Extensive experiments on on-device-scale LLaMA and Qwen models across multiple reasoning benchmarks show that our method consistently outperforms baselines and significantly narrows the gap to full cloud LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。