跳到正文
arXiv cs.AI· Yuan Zhuang, Yuexin Bian, Sihong He, Jie Feng, Qing Su, Songyang Han, Jonathan Petit, Shihao Ji, Yuanyuan Shi, Fei Miao·· 13 小时前AI 评分27

用LoRA做结构正则:离策略强化学习中的Critic学习新方法

Low-Rank Adaptation for Critic Learning in Off-Policy Reinforcement Learning

AI 导读

研究者提出用LoRA作为critic学习的结构正则化手段:冻结随机初始化的基础矩阵,只优化低秩适配器,将权重变化约束为低秩残差。在SAC+SimbaV2、BRC(DMC-Hard)与FastTD3及其SimbaV2变体(IsaacLab)上评测,LoRA降低了训练中的critic损失,多数任务取得最佳或具竞争力表现。该工作入选NeurIPS 2026。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:扩大评论家容量是提升离策略强化学习(RL)的一个有前景的方向。然而,近期工作表明,在基于回放的引导式训练中,更大的评论家容易过拟合且不稳定。本文提出使用低秩适应(LoRA)作为从零开始学习评论家的结构正则化器。我们的方法冻结随机初始化的基础矩阵,仅优化相应的低秩适配器,从而将学习到的权重变化约束为低秩残差。我们在基于状态的连续控制任务上评估了该方法:在 DMC-Hard 上使用 SAC 搭配 SimbaV2 和 BRC,在 IsaacLab 上使用 FastTD3 及其 SimbaV2 变体。实验表明,LoRA 降低了训练过程中的评论家损失并提升了整体策略性能,在大多数任务上取得最佳或具有竞争力的结果。大量实验表明,我们的低秩更新为离策略 RL 中的评论家学习提供了一种简单而有效的结构正则化形式。
评论: 2026 年神经信息处理系统大会(NeurIPS 2026)
主题: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2604.18978 [cs.LG]
  (或 arXiv:2604.18978v3 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2604.18978

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Yuan Zhuang [查看邮箱]
[v1] 2026 年 4 月 21 日星期二 01:59:54 UTC(4,126 KB)
[v2] 2026 年 5 月 7 日星期四 15:47:31 UTC(2,855 KB)
[v3] 2026 年 10 月 29 日星期四 16:01:54 UTC(2,888 KB)

来源:arXiv cs.AI · arxiv.org