用专家混合模型提升不确定切换系统的强化学习控制能力
SAC-MoE: Reinforcement Learning with Mixture-of-Experts for Control of Hybrid Dynamical Systems with Uncertainty
- 将SAC的策略设计为可自适应选择专家的混合模型
- 在未见环境上零样本泛化性能最高提升6倍
- 适合需要应对突发模式切换的机器人控制场景
混合动力系统由连续变量动态与离散事件交互构成,涵盖足式机器人、车辆和飞机等。当系统模式由不可观测(隐含)参数决定且导致模式切换的事件也不可观测时,传统基于模型的控制方法无法处理此类不确定性,而标准无模型强化学习方法难以应对突发模式切换,导致泛化性能差。为此,我们提出SAC-MoE,将软演员-批评家(SAC)框架中的策略建模为带有可学习路由器的专家混合(MoE),该路由器能自适应选择不同专家。为进一步提升鲁棒性,设计了基于课程学习的训练算法,优先在挑战性场景收集数据,从而改善对未见模式和切换位置的泛化能力。在混合自主竞速和足式行走任务的仿真中,SAC-MoE在零样本泛化到未知环境时表现优于基线方法(最高提升6倍)。课程策略在所有评估策略中均一致提升性能。定性分析显示,可解释的MoE路由器能为不同隐含模式激活相应专家。
原文摘要 · Abstract (English)
Hybrid dynamical systems result from the interaction of continuous-variable dynamics with discrete events and encompass various systems such as legged robots, vehicles and aircrafts. Challenges arise when the system's modes are characterized by unobservable (latent) parameters and the events that cause system dynamics to switch between different modes are also unobservable. Model-based control approaches typically do not account for such uncertainty in the hybrid dynamics, while standard model-free RL methods fail to account for abrupt mode switches, leading to poor generalization. To overcome this, we propose SAC-MoE which models the actor of the Soft Actor-Critic (SAC) framework as a Mixture-of-Experts (MoE) with a learned router that adaptively selects among learned experts. To further improve robustness, we develop a curriculum-based training algorithm to prioritize data collection in challenging settings, allowing better generalization to unseen modes and switching locations. Simulation studies in hybrid autonomous racing and legged locomotion tasks show that SAC-MoE outperforms baselines (up to 6x) in zero-shot generalization to unseen environments. Our curriculum strategy consistently improves performance across all evaluated policies. Qualitative analysis shows that the interpretable MoE router activates different experts for distinct latent modes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。