arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.02456cs.LG

有限状态MDP中的离线贝叶斯偶然与认知不确定性量化及后验值优化

Offline Bayesian Aleatoric and Epistemic Uncertainty Quantification and Posterior Value Optimisation in Finite-State MDPs

Imperial College London(伦敦帝国学院);University of Bayreuth(拜罗伊特大学)

Filippo Valdettaro, A. Aldo Faisal

AI总结:

针对动态未知的有限状态MDPs,本文提出分离认知与偶然不确定性并利用闭式表达式和随机梯度优化贝叶斯后验期望值的方法,在网格世界与AI Clinician任务中验证了其有效性与可扩展性。

摘要:

我们解决了在动态未知的有限状态马尔可夫决策过程(MDPs)离线应用场景中量化贝叶斯不确定性并将其纳入决策的挑战。我们的方法提供了一种有原则的方法来分离认知不确定性和偶然不确定性,并提出了一种新技术来寻找优化贝叶斯后验期望值的最优策略,而无需依赖关于MDP后验分布的强假设。首先,我们利用标准贝叶斯强化学习方法,基于可用数据捕获MDP参数中的后验不确定性。然后,我们解析地计算后验样本间回报分布的前两阶矩,并应用全方差定律来分离偶然不确定性和认知不确定性。为了找到最大化后验期望值的策略,我们利用了值作为策略函数的闭式表达式。这使我们能够提出一种基于随机梯度的方法来解决该问题。我们在简单、可解释的网格世界(gridworlds)中展示了我们智能体的不确定性量化和贝叶斯后验值优化性能,并通过在合成MDPs上的真实值评估进行了验证。最后,我们通过将该方法应用于AI Clinician问题,突出了其现实世界影响力和计算可扩展性,该问题为重症监护室患者推荐治疗方案,已成为使用离线数据的有限状态MDPs的关键用例。我们讨论了在更大规模MDPs上进行贝叶斯建模所面临的挑战,同时展示了将我们基于贝叶斯决策理论的方法应用于现实世界的潜力。我们的代码已在 https://github.com/filippovaldettaro/finite-state-mdps 开源。

英文摘要:

We address the challenge of quantifying Bayesian uncertainty and incorporating it in offline use cases of finite-state Markov Decision Processes (MDPs) with unknown dynamics. Our approach provides a principled method to disentangle epistemic and aleatoric uncertainty, and a novel technique to find policies that optimise Bayesian posterior expected value without relying on strong assumptions about the MDP's posterior distribution. First, we utilise standard Bayesian reinforcement learning methods to capture the posterior uncertainty in MDP parameters based on available data. We then analytically compute the first two moments of the return distribution across posterior samples and apply the law of total variance to disentangle aleatoric and epistemic uncertainties. To find policies that maximise posterior expected value, we leverage the closed-form expression for value as a function of policy. This allows us to propose a stochastic gradient-based approach for solving the problem. We illustrate the uncertainty quantification and Bayesian posterior value optimisation performance of our agent in simple, interpretable gridworlds and validate it through ground-truth evaluations on synthetic MDPs. Finally, we highlight the real-world impact and computational scalability of our method by applying it to the AI Clinician problem, which recommends treatment for patients in intensive care units and has emerged as a key use case of finite-state MDPs with offline data. We discuss the challenges that arise with Bayesian modelling of larger scale MDPs while demonstrating the potential to apply our methods rooted in Bayesian decision theory into the real world. We make our code available at https://github.com/filippovaldettaro/finite-state-mdps .

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

↑