自适应低秩结构让强化学习在不确定性下更鲁棒且高效
ADARL: Adaptive Low-Rank Structures for Robust Policy Learning under Uncertainty
- 通过自适应调整策略秩,动态匹配任务内在复杂度
- 在MuJoCo上优于固定秩基线和顶尖鲁棒强化学习方法
- 避免对抗性优化,收敛到任务真实秩,适合复杂环境控制
鲁棒强化学习旨在处理环境动态中的认知不确定性,但现有方法多依赖计算昂贵的嵌套极小极大优化,导致策略过度保守。本文提出自适应秩表示(AdaRL),一种双层优化框架,通过将策略复杂度与任务内在维度对齐来提升鲁棒性。下层在围绕中心模型的Wasserstein球中采样动态,进行固定秩约束下的策略优化;上层自适应调整秩,以平衡偏差-方差权衡,将策略参数投影至低秩流形。该设计无需求解对抗最坏情况动态,即可实现鲁棒性且不过度参数化。在MuJoCo连续控制基准测试中的实验表明,AdaRL不仅持续优于固定秩基线(如SAC)和顶尖鲁棒强化学习方法(如RNAC、Parseval),还收敛至底层任务的真实秩。结果表明,自适应低秩策略表示为模型不确定性下的鲁棒强化学习提供了高效且原理清晰的替代方案。
原文摘要 · Abstract (English)
Robust reinforcement learning (Robust RL) seeks to handle epistemic uncertainty in environment dynamics, but existing approaches often rely on nested min--max optimization, which is computationally expensive and yields overly conservative policies. We propose \textbf{Adaptive Rank Representation (AdaRL)}, a bi-level optimization framework that improves robustness by aligning policy complexity with the intrinsic dimension of the task. At the lower level, AdaRL performs policy optimization under fixed-rank constraints with dynamics sampled from a Wasserstein ball around a centroid model. At the upper level, it adaptively adjusts the rank to balance the bias--variance trade-off, projecting policy parameters onto a low-rank manifold. This design avoids solving adversarial worst-case dynamics while ensuring robustness without over-parameterization. Empirical results on MuJoCo continuous control benchmarks demonstrate that AdaRL not only consistently outperforms fixed-rank baselines (e.g., SAC) and state-of-the-art robust RL methods (e.g., RNAC, Parseval), but also converges toward the intrinsic rank of the underlying tasks. These results highlight that adaptive low-rank policy representations provide an efficient and principled alternative for robust RL under model uncertainty.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。