arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.19917cs.LGcs.RO
收藏

考虑不确定性的秩一多输入多输出(MIMO)Q网络框架用于加速离线强化学习

Uncertainty-Aware Rank-One MIMO Q Network Framework for Accelerated Offline Reinforcement Learning

School of Electrical Engineering, Korea Advanced Institute of Science and Technology(韩国科学技术院电子工程学院);Department of Artificial Intelligence, Korea University(韩国大学人工智能系)

Thanh Nguyen, Tung Luu, Tri Ton, Sungwoong Kim, Chang D. Yoo

AI总结:

本文提出考虑不确定性的秩一MIMO Q网络框架,通过量化数据不确定性提升离线强化学习的效率与性能。

摘要:

离线强化学习(RL)因其安全且易于扩展的范式而受到广泛关注。然而,这种范式下的训练本身也面临挑战:来自分布外(OOD)数据的外推误差。现有方法试图通过惩罚OOD Q值或对学习策略和行为策略施加相似性约束来解决这一问题。然而,这些方法往往受到过度保守地利用OOD数据、OOD数据表征不精确以及计算开销大等限制。为了解决这些挑战,本文提出了一种考虑不确定性的秩一多输入多输出(MIMO)Q网络框架。该框架旨在通过充分利用OOD数据的潜力来增强离线强化学习,同时确保学习过程的效率。具体而言,该框架量化数据不确定性并在训练损失中利用它,旨在训练一个最大化相应Q函数下限置信度的策略。此外,引入了秩一MIMO架构来建模不确定性感知的Q函数,\TP{提供与网络集合相同的能力进行不确定性量化,但成本接近于单个网络}。因此,该框架在精度、速度和内存效率之间达到了平衡,最终实现了整体性能的提升。在D4RL基准上的大量实验表明,该框架在保持计算效率的同时实现了最先进的性能。通过引入不确定性量化的概念,我们的框架为缓解外推误差和提高离线RL的效率提供了一条有前景的途径。

英文摘要:

Offline reinforcement learning (RL) has garnered significant interest due to its safe and easily scalable paradigm. However, training under this paradigm presents its own challenge: the extrapolation error stemming from out-of-distribution (OOD) data. Existing methodologies have endeavored to address this issue through means like penalizing OOD Q-values or imposing similarity constraints on the learned policy and the behavior policy. Nonetheless, these approaches are often beset by limitations such as being overly conservative in utilizing OOD data, imprecise OOD data characterization, and significant computational overhead. To address these challenges, this paper introduces an Uncertainty-Aware Rank-One Multi-Input Multi-Output (MIMO) Q Network framework. The framework aims to enhance Offline Reinforcement Learning by fully leveraging the potential of OOD data while still ensuring efficiency in the learning process. Specifically, the framework quantifies data uncertainty and harnesses it in the training losses, aiming to train a policy that maximizes the lower confidence bound of the corresponding Q-function. Furthermore, a Rank-One MIMO architecture is introduced to model the uncertainty-aware Q-function, \TP{offering the same ability for uncertainty quantification as an ensemble of networks but with a cost nearly equivalent to that of a single network}. Consequently, this framework strikes a harmonious balance between precision, speed, and memory efficiency, culminating in improved overall performance. Extensive experimentation on the D4RL benchmark demonstrates that the framework attains state-of-the-art performance while remaining computationally efficient. By incorporating the concept of uncertainty quantification, our framework offers a promising avenue to alleviate extrapolation errors and enhance the efficiency of offline RL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑