Reinforcement learning (RL) often struggles to effectively balance exploration and exploitation, particularly under uncertainties in both rewards and policy parameters. Consequently, existing methods can suffer from premature convergence or require extensive hyperparameter tuning. To overcome these challenges, we introduce a novel framework based on the principle of negative free-energy maximization. This approach provides a unified treatment of uncertainty by modeling policies as distributions over their parameters and rewards using Bayesian estimation, yielding an objective function that integrates expected returns, their associated variance, and policy entropy. This objective, derived from a free-energy formulation approximated using cumulant-generating functions, inherently promotes exploration through entropy regularization and uncertainty-driven bonuses. We further present an efficient iterative algorithm employing Gaussian policy approximations and an adaptive inverse temperature, regulated by a Kullback-Leibler (KL) divergence constraint, which reduces the need for manual hyperparameter tuning. Our framework not only generalizes and extends prominent prior works but also offers a principled and scalable solution for robust policy optimization in complex, continuous environments.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Uncertainty-Aware Policy Search via Free Energy Optimization

  • Tikara Hosino

摘要

Reinforcement learning (RL) often struggles to effectively balance exploration and exploitation, particularly under uncertainties in both rewards and policy parameters. Consequently, existing methods can suffer from premature convergence or require extensive hyperparameter tuning. To overcome these challenges, we introduce a novel framework based on the principle of negative free-energy maximization. This approach provides a unified treatment of uncertainty by modeling policies as distributions over their parameters and rewards using Bayesian estimation, yielding an objective function that integrates expected returns, their associated variance, and policy entropy. This objective, derived from a free-energy formulation approximated using cumulant-generating functions, inherently promotes exploration through entropy regularization and uncertainty-driven bonuses. We further present an efficient iterative algorithm employing Gaussian policy approximations and an adaptive inverse temperature, regulated by a Kullback-Leibler (KL) divergence constraint, which reduces the need for manual hyperparameter tuning. Our framework not only generalizes and extends prominent prior works but also offers a principled and scalable solution for robust policy optimization in complex, continuous environments.