A Risk-Sensitive Approach to Policy Optimization

Abstract

Standard deep reinforcement learning (DRL) aims to maximize expected reward, considering collected experiences equally in formulating a policy. This differs from human decision-making, where gains and losses are valued differently and outlying outcomes are given increased consideration. It also fails to capitalize on opportunities to improve safety and/or performance through the incorporation of distributional context. Several approaches to distributional DRL have been investigated, with one popular strategy being to evaluate the projected distribution of returns for possible actions. We propose a more direct approach whereby risk-sensitive objectives, specified in terms of the cumulative distribution function (CDF) of the distribution of full-episode rewards, are optimized. This approach allows for outcomes to be weighed based on relative quality, can be used for both continuous and discrete action spaces, and may naturally be applied in both constrained and unconstrained settings. We show how to compute an asymptotically consistent estimate of the policy gradient for a broad class of risk-sensitive objectives via sampling, subsequently incorporating variance reduction and regularization measures to facilitate effective on-policy learning. We then demonstrate that the use of moderately "pessimistic" risk profiles, which emphasize scenarios where the agent performs poorly, leads to enhanced exploration and a continual focus on addressing deficiencies. We test the approach using different risk profiles in six OpenAI Safety Gym environments, comparing to state of the art on-policy methods. Without cost constraints, we find that pessimistic risk profiles can be used to reduce cost while improving total reward accumulation. With cost constraints, they are seen to provide higher positive rewards than risk-neutral approaches at the prescribed allowable cost.

16 pages, 13 figures. AAAI 2023 (Special Track on Safe and Robust AI)

References in corpus (11)

• Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learning
• Distributional Soft Actor-Critic: Off-Policy Reinforcement Learning for Addressing Value Estimation Errors
• A Distributional Perspective on Reinforcement Learning
• Lyapunov-based Safe Policy Optimization for Continuous Control
• First Order Constrained Optimization in Policy Space
• DSAC: Distributional Soft Actor-Critic for Risk-Sensitive Reinforcement Learning
• Cautious Reinforcement Learning via Distributional Risk in the Dual Domain
• Constrained Reinforcement Learning Has Zero Duality Gap
• Exponential Bellman Equation and Improved Regret Bounds for Risk-Sensitive Reinforcement Learning
• Risk-Sensitive Deep RL: Variance-Constrained Actor-Critic Provably Finds Globally Optimal Policy
• Clipped-Objective Policy Gradients for Pessimistic Policy Optimization