arXiv:2606.30192cs.AI2026-06

在目标数据极少时,用智能生成的可靠模拟数据提升视觉强化学习的现实迁移效果。

Domain Adaptation with Adaptive Imagination for Visual Reinforcement Learning under Limited Target Data

论文配图:Domain Adaptation with Adaptive Imagination for Visual Reinforcement Learning under Limited Target Data
图 1 · 摘自论文原文
  • 通过自适应想象生成高质量模拟轨迹,只保留可信部分用于数据增强。
  • 在5个MuJoCo和2个Gymnasium-Robotics任务中超越基线,尤其在小样本下表现突出。
  • 适合缺乏真实交互数据的机器人视觉控制场景,如工业部署或安全敏感环境。

视觉强化学习中的仿真到现实迁移仍面临重大挑战,尤其在图像观测下,仿真与真实世界的状态分布差异显著。领域自适应(DA)是缓解此问题的有前景方法。现有仿真实现通常依赖大量目标数据,但在实际中难以获得,且数据稀缺时性能显著下降。为此,我们提出AIDA(自适应想象领域自适应),一种针对视觉强化学习的领域自适应框架,可在不需额外目标环境交互的前提下应对有限目标数据。核心思想是自适应想象:生成可靠且语义合理的模拟轨迹以扩充稀缺的目标数据。具体而言,AIDA采用分布偏移感知判别器,在想象过渡进入低置信区域时截断轨迹,仅保留可信过渡参与增强。在这些可信过渡上,引入自一致性损失,实现状态→图像观测→状态的循环重建,惩罚原始与重构状态间的差异,提供超出稀缺目标数据的额外适应信号。实验表明,自适应想象能有效剔除不可靠轨迹;通过在可信轨迹上施加自一致性损失,AIDA学习到语义有意义的状态表示,并在五个MuJoCo任务和两个Gymnasium-Robotics任务中优于基线方法。

原文摘要 · Abstract (English)

Sim-to-real transfer remains a major obstacle for reinforcement learning (RL), especially for vision-based control where image observations exacerbate the state-distribution shift between simulation and the real world. Domain adaptation (DA) is a promising remedy for this challenge. Prior sim-to-real DA works have demonstrated encouraging results, yet these approaches typically assume substantially more target data, which is not available in practice. Indeed, their performance degrades significantly when the target data budget is reduced. To address this challenge, we propose AIDA (Adaptive Imagination for Domain Adaptation), a domain adaptation framework for visual reinforcement learning that addresses sim-to-real transfer under scarce target data without requiring additional interaction with the target environment. Our key idea is adaptive imagination: generating reliable and semantic imagination rollouts to augment limited target data. Specifically, AIDA employs a distribution-shift-aware discriminator that truncates rollouts when imagined transitions drift into low-confidence regions, so that only reliable transitions contribute to the augmentation. On these reliable transitions, AIDA introduces a self-consistency loss that cycles through state -> image observation -> state, penalizing discrepancies between the original and reconstructed states. This provides additional adaptation signals beyond the scarce target data. Our experiments demonstrate that adaptive imagination effectively truncates unreliable rollouts. By enforcing a self-consistency loss on the resulting reliable transitions, AIDA learns semantically meaningful state representations and outperforms baselines across five MuJoCo tasks and two Gymnasium-Robotics tasks.

领域自适应强化学习少样本视觉控制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。