用两阶段强化学习动态组合小模型,提升大模型集成的性能与鲁棒性。
Dynamic Optimizations of LLM Ensembles with Two-Stage Reinforcement Learning Agents
- 两阶段强化学习:先选小模型集,再融合推理结果
- 小规模集成提升8.48%性能,优于单个最优模型
- 新提出的焦点多样性指标有效降低模型错误相关性
大语言模型的发展及其可及性重新激发了多智能体强化学习在动态环境中的应用兴趣。本文提出RL-Focal,一个两阶段强化学习代理框架,用于路由和集成大语言模型。首先,设计决策强化学习代理(Decider RL-agent),通过任务自适应奖励与策略迭代更新,在 $N$ 个大语言模型中动态选择小规模集合 ($m_i \\< N$) 来响应用户定义的下游任务 $i$,以最大化所选集合的误差多样性与推理性能。其次,为实现动态选择模型的有效融合,设计第二阶段融合强化学习代理(Fusion RL-agent),学习解决不同模型间的推理冲突,并自适应不同任务的模型组合。第三,引入焦点多样性度量,更精准建模多个大模型间的错误相关性,进一步提升决策代理的泛化能力,主动剪枝无效组合。通过该机制,显著增强跨任务的奖励感知与策略自适应的集成选择与推理融合。在五个基准测试上的广泛评估表明,与池中最佳单一模型相比,使用小规模集成的RL-Focal实现了8.48%的性能提升,并展现出更强鲁棒性。代码已开源。
原文摘要 · Abstract (English)
The advancement of LLMs and their accessibility have triggered renewed interest in multi-agent reinforcement learning as robust and adaptive frameworks for dynamically changing environments. This paper introduces RL-Focal, a two-stage RL agent framework that routes and ensembles LLMs. First, we develop the Decider RL-agent, which learns to dynamically select an ensemble of small size ($m_i$) among $N$ LLMs ($m_i \ll N$) for incoming queries from a user-defined downstream task $i$, by maximizing both error-diversity and reasoning-performance of the selected ensemble through iterative updates of task-adaptive rewards and policy. Second, to enable effective fusion of dynamically selected LLMs, we develop the stage-2 Fusion RL-agent, which learns to resolve reasoning conflicts from different LLMs and dynamically adapts to different ensemble teams composed by the Decider Agent for different downstream tasks. Third, we introduce the focal diversity metric to better model the error correlations among multiple LLMs, further improving the generalization performance of the Decider Agent, which actively prunes the ensemble combinations. By focal diversity, we enhance performance across tasks by effectively promoting reward-aware and policy-adaptive ensemble selection and inference fusion. Extensive evaluations on five benchmarks show that RL-Focal achieves the performance improvement of 8.48\% with an ensemble of small size compared to the best individual LLM in a pool and offers stronger robustness. Code is available at https://github.com/sftekin/rl-focal
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。