Risk-averse decision-making under uncertainty in partially observable domains is a central challenge in artificial intelligence and is essential for developing reliable autonomous agents. The formal framework for such problems is the partially observable Markov decision process (POMDP), where risk sensitivity is introduced through a risk measure applied to the value function, with Conditional Value-at-Risk (CVaR) being a particularly significant criterion. However, solving POMDPs is computationally intractable in general, and approximate methods rely on computationally expensive simulations of future agent trajectories. This work introduces a theoretical framework for accelerating CVaR value function evaluation in POMDPs with formal performance guarantees. We derive new bounds on the CVaR of a random variable X using an auxiliary random variable Y, under assumptions relating their cumulative distribution and density functions; these bounds yield interpretable concentration inequalities and converge as the distributional discrepancy vanishes. Building on this, we establish upper and lower bounds on the CVaR value function computable from a simplified belief-MDP, accommodating general simplifications of the transition dynamics. We develop estimators for these bounds within a particle-belief MDP framework with probabilistic guarantees, and employ them for acceleration via action elimination: actions whose bounds indicate suboptimality under the simplified model are safely discarded while ensuring consistency with the original POMDP. Empirical evaluation across multiple POMDP domains confirms that the bounds reliably separate safe from dangerous policies while achieving substantial computational speedups under the simplified model.
Figures & tables
Figure 1. Overview of the proposed framework. The original POMDP yields the original CVaR value function. The simplified POMDP Ms admits a simplified value function, related to the original via Theorem 6.2 . In practice, the simplified value function and distributional discrepancy are estimated from sample trajectories, yielding probabilistic computable bounds on the original value function (Theorem 7.4 ).
Figure 2. Conceptual overview. Direct evaluation of QMπ under the original model M is computationally expensive. Instead, the simplified model Ms is used to compute bounds on QMπ , where the bound quality is governed by the distributional discrepancy ϵ between their belief transition models.
Figure 3. Problem formulation illustrations. (a) The CVaR value function focuses on the tail of the return distribution, unlike the expected value. (b) The simplified model Ps yields a computable CDF FY , while the original CDF FX is intractable. When ∣FX(r)−FY(r)∣≤ϵ , bounds on \CVaRα(X) can be derived from Y .
Figure 4. Illustrations of bounds on FX(x) . (a) Bounds from Theorem 5.1 . (b) Bounds from Theorem 5.3 , where g depends on x and yields a tighter result than FY(x)+ϵ .
Figure 5. Accumulation of distributional discrepancy over the planning horizon. Each bar represents the expected one-step TV distance Es[Δs(bt,at)∣bk,ak] at time t . The cumulative sum ϵ=∑t=kT−1Δts (dashed) governs the bound quality: when ϵ<α (below the dash-dotted line), the CVaR bounds from Theorem 6.2 remain informative; when ϵ≥α , the bounds reduce to trivial extrema.
Figure 6. Offline–online decoupling for distributional discrepancy estimation. In the offline phase, states are sampled from a proposal distribution Q0 and the TV distances Δs are computed using the original observation model pz . During online planning, the precomputed Δs values are reweighted via importance sampling using only the state-transition model, eliminating the need to evaluate pz in real time.
Figure 7. Illustration of bound-based action selection when ϵ′≥α . (a) The agent chooses between a safe path (cost ≈1×2=2 ) and a dangerous path through an obstacle (cost ≈100×2=200 ). (b) Although the CVaR bounds collapse to [d^min,d^max] , the return estimators remain well-separated, enabling the agent to distinguish between the two paths.
Figure 8. Figure 8a shows the bounds established in Theorem 5.7 , demonstrating how a simple truncated Normal distribution can be employed to bound the CVaR of a truncated GMM. Figure 8b illustrates the sensitivity of the CVaR bounds to the distributional discrepancy ϵ , showing how the bound interval widens as ϵ increases, while the true CVaR remains contained within the bounds. When ϵ exceeds the risk level α , the bounds become trivial, spanning the entire support; this is expected, as a large distributional discrepancy implies that the surrogate random variable Y provides negligible information about the target random variable X . Figure 8c presents a comparison of the sampling times for each distribution, while Figure 8d reports the total computation time required to sample from the truncated GMM and estimate its CVaR, in contrast to sampling from the truncated Normal surrogate and computing the associated concentration bounds. Figures 8a , 8c , and 8d were configured with confidence level α=0.2 and probability of error δ=0.05 . Figure 8b uses α=0.5 . Figures 8c and 8d display 95% confidence intervals around the empirical means.
Figure 9. Figures 9a , 9b , and 9c illustrate pairs of agent trajectories, consisting of a safe (orange) and a dangerous (blue) path. In Figure 9a , the agent moves from the lower-left to the upper-right goal location; the dangerous trajectory incurs higher cost. Figure 9b depicts trajectories in the Laser Tag environment, where the dangerous path traverses a high-cost region indicated in red. Figure 9c shows a similar pair of trajectories, with the dangerous path entering a hazardous region that leads to a terminal state. Across all three environments, the theoretical bounds on the values of the corresponding action sequences successfully discriminate between the safe and dangerous trajectories. Bounds are computed with probability of error δ=0.05 .
Figure 10. Sensitivity analysis of the computational acceleration with respect to the planning horizon and number of return samples, for Light-Dark (left), Laser Tag (center), and Push (right), with α=0.5 . Rows 1–2: computation time and acceleration ratio as a function of the planning horizon. Rows 3–4: computation time and acceleration ratio as a function of the number of return samples. Error bars indicate 95% confidence intervals.
Figure 11. Sensitivity of the CVaR bounds for the Light-Dark (left), Laser Tag (center), and Push (right) environments, using the complete safe (blue) and dangerous (red) action sequences depicted in Figure 9 ; solid and dashed lines indicate the upper and lower bounds, respectively. Rows 1–2: bounds as a function of the planning horizon with α=0.5 and α=0.1 , respectively. Row 3: bounds as a function of the risk level α at a fixed horizon. As the horizon increases, the accumulated distributional discrepancy grows, causing the bounds to widen. As α decreases (more risk-averse), the bounds increase and the bound intervals widen, yet they remain well separated between the safe and dangerous paths across the full range of α . Bounds are computed with probability of error δ=0.05 .
Figure 12. Evaluation of the BetaZero neural network policy against the safe and dangerous action sequences in the Light-Dark POMDP ( α=0.1 , H=9 ), with bounds computed under the simplified observation model. Figure 12a : CVaR bounds for the dangerous sequence (blue), safe sequence (orange), and BetaZero policy (green). The bound intervals of the policy and dangerous sequence do not overlap, demonstrating that the bounds correctly identify the policy as less risky. Bounds are computed with probability of error δ=0.05 . Figure 12b : agent trajectories; the BetaZero policy (green) and safe sequence (orange) navigate through the illuminated safe zones, while the dangerous sequence (blue) traverses the high-cost region.
Figure 13. Sensitivity of the CVaR bounds (computed under the simplified observation model) for the BetaZero policy, safe sequence, and dangerous sequence in the Light-Dark POMDP; solid and dashed lines denote upper and lower bounds, respectively. Figure 13a : bounds as a function of the risk level α at horizon H=9 . Figure 13b : bounds as a function of the planning horizon at α=0.1 . The dangerous sequence remains clearly separated from the policy and safe sequence across all conditions. Bounds are computed with probability of error δ=0.05 .
Figure 14. Computational speedup from evaluating CVaR bounds under the simplified observation model relative to the original model for the BetaZero policy in the Light-Dark POMDP ( α=0.1 ). Row 1: computation time and speedup ratio as a function of the planning horizon. Row 2: computation time and speedup ratio as a function of the number of return samples. The simplified model achieves a consistent ≈5× speedup across all conditions. Error bars indicate 95% confidence intervals.
Appendix figures & tables17 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 15. Illustration of the upper and lower bounds FYU and FYL on the cumulative distribution function FX .
Figure 16. Comparison of concentration inequalities for \CVaRα(X) established by Theorem 5.5 (this work) and by Thomas and Learned-Miller, 2019 , evaluated on Beta(2,2) , Beta(0.5,0.5) , Beta(2,5) , Beta(5,2) , Beta(10,2) , Beta(2,10) , and Laplace(0,1) distributions. The two sets of bounds coincide for all tested distributions.
Aspect
Original
Simplified
Observation noise
Gaussian Mixture Model (GMM)
Isotropic Gaussian
Reward noise
Stochastic (Push, Laser Tag)
Deterministic
Appendix
Table 1. Architectural differences between the original and simplified models across all domains.
Parameter
Value
Grid size
7×7
Start state
(1,1)
Goal state
(6,6) ; radius 1.5
State transition covariance
0.06⋅I2
Obs. noise (far from beacon)
N(0,0.06⋅I2) / GMM( K=2500 )
Obs. noise (near beacon)
N(0,0.03⋅I2) / GMM( K=2500 , scaled)
Appendix
Table 2. Configuration for the 2D Light-Dark POMDP.
Parameter
Value
Grid size
6×6
Initial state
(0.5,0.5,1.0,0.5,5.0,5.0)
Robot transition covariance
diag(0.005,0.005)
Observation noise σ
0.1 (on object position)
Push threshold
1.0
Friction coefficient
0.3
Appendix
Table 3. Configuration for the Continuous Push POMDP.
Parameter
Value
Arena
11×7 continuous grid
Walls
8 axis-aligned wall cells
Initial state
(2.0,1.0,8.0,5.0,0)
Robot transition covariance
0.0125⋅I2
Opponent transition covariance
0.00625⋅I2
Opponent pursuit speed
0.6
Appendix
Table 4. Configuration for the Continuous Laser Tag POMDP.
Parameter
Value
CVaR level α
0.1
Belief particles Np
10
Return samples Nr
600
Δ -estimation states
100
Samples per state (for Δ )
2000
k -NN neighbours
10
Appendix
Table 5. Planner parameters shared across all three environments.
Parameter
Value
Number of Episodes
300
Steps per Episode
30 (max)
Planning Depth
2
Return Samples ( Nb )
600
CVaR α
0.5
Number of Particles ( Np )
10
Appendix
Table 6. Planning configuration for the Continuous Laser Tag experiments.
Parameter
Value
Number of Episodes
300
Steps per Episode
30 (max)
Planning Depth
2
Return Samples ( Nb )
600
CVaR α
0.5
Number of Particles ( Np )
10
Appendix
Table 7. Planning configuration for the Light-Dark experiments.
Parameter
Value
Number of Episodes
300
Steps per Episode
30 (max)
Planning Depth
2
Return Samples ( Nb )
600
CVaR α
0.5
Number of Particles ( Np )
10
Appendix
Table 8. Planning configuration for the Continuous Push experiments.
Parameter
Value
State dimension
2
Belief feature dimension
4 (= 2× state dim)
Particles per belief Np
500
Appendix
Table 9. Belief representation parameters for BetaZero on Light-Dark.
Feature
Mean
Std
μˉx
3.56
1.91
μˉy
4.12
1.91
σˉx
0.302
0.407
σˉy
0.303
0.408
Value output
−24.34
36.64
Appendix
Table 10. Input normalisation statistics for the BetaZero network.
Layer
Size
Input
4
Hidden 1
128
Hidden 2
128
Value head output
1 (scalar)
Policy head output
4 (one logit per action)
Appendix
Table 11. BetaZero network architecture.
Parameter
Value
Search depth
10
Simulations per step (training)
100
Simulations per step (inference)
1000
Exploration constant c
219.13 (tuned)
Temperature τ
0.319 (tuned)
Value weight zq
1.0
Appendix
Table 12. MCTS parameters for BetaZero on Light-Dark.
Parameter
Value
Outer iterations
50
Episodes per iteration
50
Total training episodes
2500
Episode length
20 steps
Data collection
Full MCTS rollouts
Replay buffer
5 iterations
Appendix
Table 13. BetaZero training loop parameters.
Parameter
Value
Optimizer
Adam
Learning rate
10−4
Batch size
1024
Training epochs per iteration
50
Weight decay
10−5
Loss
Cross-entropy (policy) + MSE (value)
Appendix
Table 14. Optimizer and loss configuration for BetaZero training.
Hyperparameter
Search range
Best value
c (exploration constant)
[1.0,10×Δr]≈[1,220]
219.13
τ (temperature)
[0.05,0.5]
0.319
zq , zn
fixed
1.0, 1.0
Appendix
Table 15. Hyperparameter search results for BetaZero on Light-Dark.
Faculty of Mathematics Technion – Israel Institute of Technology Haifa, Israel · Stephen B. Klein Faculty of Aerospace Engineering; Faculty of Data and Decision Sciences Technion – Israel Institute of Technology Haifa, Israel
Universit´e de Montr´eal, Quebec, Canada · Mila - Quebec AI Institute, Quebec, Canada · GERAD & Department of Decision Sciences, HEC Montr´eal, Quebec, Canada