REVIEW 3 major objections 5 minor 2 cited by
Feature-Based Belief Aggregation for Partially Observable Markov Decision Problems
T0 review · 3 major / 5 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read A two-stage feature-based aggregation scheme approximates the optimal cost of a POMDP with a uniform error bound $\epsilon/(1-\alpha)$, and with a lower-bound guarantee when representative beliefs span the belief simplex.
desk verdict A clean, practical two-stage aggregation scheme for POMDPs with a correct central error bound; the Prop. 6 sign error is a typo, not a proof killer, but the particle-filter experiments sit outside the theory. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the aggregate MDP whose state space is the finite grid of representative feature beliefs $\tilde Q$. Two mappings carry the compression: $D$ disaggregates a representative feature belief into a full belief over original states, and $\Phi$ projects any updated belief back onto the feature-belief simplex; composing them with the belief update $F$ gives the one-step transition operator $H$ of Eq. (16), which is monotone and an $\alpha$-contraction in the max norm. Solving $r^*=Hr^*$ by value iteration or asynchronous value iteration yields the cost vector used in the interpolation formula (18). The error analysis works entirely through the footprint sets $S_{\tilde q}$ (beliefs whose interpolation weights attach to $\tilde q$): the bound says the worst error is the largest fluctuation of $J^*$ inside any footprint, amplified by the factor $1/(1-\alpha)$. The lower-bound result leans on concavity of $J^*$ in the belief vector together with the convex-hull condition (42).
What would settle it
Take a small POMDP for which $J^*$ can be computed exactly, such as the paper's treasure-hunting problem with two sites, build a feature aggregation with overlapping footprint sets, and evaluate $|\tilde J(b)-J^*(b)|$ on a dense grid of beliefs. If any evaluation exceeds $\epsilon/(1-\alpha)$ with $\epsilon$ defined as in Eq. (39), Proposition 5 is false; if none does, the bound stands for that instance and its tightness can be measured directly.
Extended reading notes
Core claim
At the center of the paper is the observation that a POMDP's belief space can be compressed twice without losing a provable handle on optimal cost: first compress the hidden state space into feature states via aggregation and disaggregation probabilities, then compress the feature-belief simplex into a finite grid of representative feature beliefs. Solving the resulting finite MDP gives a cost vector $r^*$, and the interpolation formula $\tilde J(b)=\sum_{\tilde q}\psi_{\Phi(b)\tilde q}r^*_{\tilde q}$ maps those representative costs back to every original belief. The central result, Proposition 5, states that this approximation is uniformly within $\epsilon/(1-\alpha)$ of $J^*$, where $\epsilon$ is the maximum variation of $J^*$ over the footprint sets $S_{\tilde q}=\{b:\psi_{\Phi(b)\tilde q}>0\}$. The bound requires no assumption that footprint sets partition the belief space, which is exactly what lets the feature-based construction overlap and still work. Proposition 6 adds that when the images $\tilde B$ of the representative beliefs span the whole belief simplex through Eq. (42), the approximation is a lower bound for $J^*$, and it is linear when the representative set has $n$ elements. The biased variant replaces $J^*$ by the residual $J^*-V$ in the definition of $\epsilon$, which explains why a good guess $V$ buys accuracy with fewer representative beliefs.
Load-bearing premise
The theoretical error bound assumes the belief update $F$ is computed exactly, but in the two largest experiments it is replaced by a 100-particle rejection-sampling particle filter with no error estimate, so the reported costs are not covered by the bound.
Editorial extensions
If this is right
- With a fixed representative grid, the approximation error is controlled by how much $J^*$ varies inside footprint sets, so refining the grid where $J^*$ changes most should shrink the bound.
- Because Proposition 5 does not require footprint sets to be disjoint, features can map many original states to shared feature states, which is what lets the method scale to state spaces with tens of thousands of hidden states.
- When the convex-hull condition holds, the approximation is a conservative lower bound and can be used safely in rollout or policy-improvement procedures.
- Biased aggregation reuses any existing cost approximation $V$: the error term becomes the variation of $J^*-V$, so a good $V$ yields comparable accuracy with far fewer representative feature beliefs.
- The aggregate MDP inherits standard value-iteration convergence, so computational effort scales with the size of the feature grid rather than with the full hidden-state count.
Reading between the lines
- The bound suggests an adaptive design loop the paper does not develop: choose feature partitions and representative grids to minimize footprint variation, guided by local estimates of $J^*$ from a smaller exact problem.
- Because the proof does not rely on POMDP-specific structure, the same two-stage analysis transfers to any large-state MDP with a feature embedding, a point the authors assign to a companion analysis.
- In the two largest experiments the belief update is replaced by a 100-particle rejection-sampling particle filter, so the reported costs carry particle noise that the $\epsilon/(1-\alpha)$ guarantee does not cover; a direct test would compare exact-filter and particle-filter results on a moderately sized instance.
- Biased aggregation could be iterated: feed the resulting approximation back in as the next bias function and re-solve, which should drive the residual variation and hence $\epsilon$ toward zero even on a coarse representative grid.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. This paper proposes a two-stage aggregation scheme for infinite-horizon discounted finite-state POMDPs. In the first stage, unobservable states are aggregated into feature states via disaggregation and aggregation distributions; in the second, beliefs over feature states are projected onto a finite set of representative feature beliefs. The resulting aggregate MDP is solved by value iteration, and its fixed point is interpolated to form an approximate cost function J~ used in one-step lookahead. The paper's main theoretical results are Proposition 5, an epsilon/(1-alpha) error bound for |J~ - J*| where epsilon is the maximal variation of J* over footprint sets, and Proposition 6, a lower-bound result under a convex-hull condition (Eq. 42). A biased-aggregation extension (Props. 7-8) modifies stage costs using an approximate value function. Experiments on treasure hunting, Rocksample, and CAGE-2 show competitive costs with lower computational effort.
Significance. If the claims hold, this is a useful and fairly general framework. The error-bound proof is self-contained and does not rely on a partition of the belief space, which relaxes assumptions in the classical Tsitsiklis and Van Roy bound. The lower-bound result gives a clean sufficient condition based on convexity of J* and convex-hull coverage of the belief simplex. The paper ships code and clear algorithmic descriptions, and the experimental comparison is broad. The main caveats are that the bound is not computable a priori (it depends on the unknown J* and footprint geometry), and the large-scale experiments use a particle-filter approximation of the belief update that is outside the assumptions of the theorems; both should be acknowledged and addressed in revision.
major comments (3)
- [Section 3.3, Eq. (48)] Eq. (48) in the proof of Proposition 6 has the wrong inequality. The line immediately before it, Eq. (47), states J*(D(q)) >= min_u [ ... ] = (H rhat)(q), i.e., H rhat <= rhat, not H rhat >= rhat. As printed, H rhat >= rhat would imply r* >= rhat under the monotone-contraction argument, which is the opposite of the needed r* <= rhat. Please correct Eq. (48) and the following sentence; with H rhat <= rhat, the monotone-contraction argument indeed gives r* <= rhat, so the lower-bound conclusion is recoverable.
- [Appendix A.2 and Section 4] Appendix A.2 states that for RS(10,10) and CAGE-2 the belief estimator F is replaced by a 100-particle rejection-sampling particle filter, while Props. 5-8 assume F is computed exactly. No bias, variance, or consistency estimate for this replacement is provided, so the error bounds of Props. 5-8 do not cover the costs reported in Figs. 10-13. The paper should either restrict the theoretical claims to exact-F settings, add an analysis or sensitivity study for the particle filter, or explicitly designate the large-scale experimental claims as heuristic.
- [Section 4 and Appendix A.2] The lower-bound guarantee of Proposition 6 depends critically on the convex-hull condition Eq. (42). The experimental scheme described in Appendix A.2 uses the nearest-neighbor hard-aggregation rule of Eq. (10), which does not satisfy Eq. (42) in the compressed regimes (rho=1 or rho=2) used for RS(7,8), RS(10,10), and CAGE-2. The paper should state that the lower-bound theorem applies to the constructions in Examples 3.3 and 5.1 rather than to these experiments; otherwise readers may attribute a guarantee to the large-scale results that the theory does not provide.
minor comments (5)
- [Figure 8] The y-axis label reads '\||J* - J~||'; the stray vertical bar should be removed.
- [Section 5.2] Props. 7 and 8 are stated without proofs; since they are central to the biased-aggregation contribution, a brief proof sketch or an explicit reduction to Props. 5-6 via Eq. (50) would improve verifiability.
- [Section 3.2] The converse half of Proposition 5 is asserted without display; writing out the symmetric argument would prevent the sign confusion that later appears in Proposition 6.
- [Section 2.3] The phrase 'we can see that the unobservable states i involved in our algorithm form a set X-hat' is informal; X-hat is defined after the statement and should be introduced before use.
- [Figures 7 and 9] The caption text '10Q, with m=... representative feature beliefs' appears garbled; please restate, for example, as 'the one-dimensional feature-belief simplex Q'.
Circularity Check
No significant circularity: the aggregation error bound is proven by a self-contained DP contraction argument; self-citations provide framework only.
full rationale
Prop. 5's bound |J~(b) − J*(b)| ≤ ε/(1−α) is not circular: ε is defined as the true optimal cost's variation over the footprint sets (Eq. 39), and the proof establishes Hr ≤ r for the vector r_~q = inf_{b∈S_~q} J*(b) + ε/(1−α), then invokes monotonicity and contractivity of H (Prop. 1, a standard argument) to obtain r* ≤ r and the upper bound, with the symmetric sup-vector giving the lower bound. The bound's content is the contraction factor 1/(1−α) and the honest dependence on J*; Fig. 8 even shows the bound is loose, so it is not manufactured to match the method. Prop. 6's lower-bound claim depends on concavity of J* (cited to Sondik 1978 and Krishnamurthy 2016, both external) and on the explicit convex-hull condition Eq. (42); it does not assume the conclusion J~ ≤ J*. The derivation chain is therefore self-contained and does not reduce to its inputs. Two non-circularity concerns should be noted. First, the displayed inequality H r̂ ≥ r̂ in Eq. (48) has the wrong direction — Eq. (47) yields H r̂ ≤ r̂ — and the claimed implication r̂ ≥ r* follows only with the corrected sign; this is a correctness or typographical issue in the manuscript's own argument, not an equation-level equivalence of input and output. Second, several framework citations are to the authors' own prior work ([Ber12] aggregation, [Ber19a] biased aggregation, [LB25] the companion MDP bound), but the load-bearing identities are re-derived in-paper (e.g., J* = Ṽ + V in Eq. (50)), so the self-citations are not load-bearing reductions. The manuscript itself flags its limitations: Props. 7 and 8 are stated without proofs (Section 5.2), and for RS(10,10) and CAGE-2 the belief estimator F is approximated by a 100-particle rejection-sampling particle filter (Appendix A.2) while Props. 5–8 assume exact F; these are support gaps for the experimental claims, not circularity.
Assumptions & free parameters
free parameters (4)
- discretization resolution rho =
rho=10 for treasure hunting; rho=3 for RS(4,4); rho=2 for RS(5,5) and RS(5,7); rho=1 for RS(7,8), RS(10,10) and CAGE-2
- particle filter size =
100
- CAGE-2 feature mapping =
attacker-state, attacker-target, decoy-state; feature space size 427,500
- RS(10,10) feature mapping =
3x3 grid positions; feature space size 9,216
assumptions (6)
- standard math J* is bounded and uniquely satisfies the Bellman equation (4).
- domain assumption The optimal cost function J* is concave over the belief simplex.
- domain assumption The feature sets Ix are disjoint and each state is assigned to at most one feature set, with disaggregation and aggregation probabilities satisfying Eqs. (5) and (6).
- domain assumption The belief aggregation probabilities satisfy Eq. (8), and for the lower-bound theorem the convex hull condition (42) holds.
- domain assumption The bias function V in Section 5 is a bounded approximation to J*.
- domain assumption The belief update F follows Bayes' rule and can be evaluated exactly in the analytical examples.
Cite this review
Pith. "Pith review of Feature-Based Belief Aggregation for Partially Observable Markov Decision Problems." pith.science (2026). https://pith.science/paper/U5GTAHLM
@misc{pith2026250704646,
author = {Pith},
title = {Pith review of: Feature-Based Belief Aggregation for Partially Observable Markov Decision Problems},
year = {2026},
howpublished = {\url{https://pith.science/paper/U5GTAHLM}},
note = {Machine review of arXiv:2507.04646}
}
read the original abstract
We consider a finite-state partially observable Markov decision problem (POMDP) with an infinite horizon and a discounted cost, and we propose a new method for computing a cost function approximation that is based on features and aggregation. In particular, using the classical belief-space formulation, we construct a related Markov decision problem (MDP) by first aggregating the unobservable states into feature states, and then introducing representative beliefs over these feature states. This two-stage aggregation approach facilitates the use of dynamic programming methods for solving the aggregate problem and provides additional design flexibility. The optimal cost function of the aggregate problem can in turn be used within an on-line approximation in value space scheme for the original POMDP. We derive a new bound on the approximation error of our scheme. In addition, we establish conditions under which the cost function approximation provides a lower bound for the optimal cost. Finally, we present a biased aggregation approach, which leverages an optimal cost function estimate to improve the quality of the approximation error of the aggregate problem.
Figures
Figures from the paper (13 more)
Forward citations
Cited by 2 Pith papers
-
Recovery Control in Replicated Systems through Autonomous Multiagent Rollout
Autonomous multiagent rollout with signaling approximates near-optimal recovery policies for replicated systems, scaling to 70 replicas and lowering cost versus practical baselines.
-
Online Incident Response Planning under Model Misspecification through Bayesian Learning and Belief Quantization
MOBAL learns a model of an ongoing cyberattack with Bayesian updates and computes incident responses with a quantized version of that model, giving robustness to model misspecification on CAGE-2.
Reference graph
Works this paper leans on
- [1]
- [2]
-
[3]
Bertsekas
Dimitri P. Bertsekas. Dynamic Programming and Optimal Control: Vol. II . Athena Scientific Belmont, 4th edition, 2012
2012
- [4]
-
[5]
Bertsekas
Dimitri P. Bertsekas. Feature-based aggregation and deep reinforcement learning: A survey and some new implementations. IEEE/CAA Journal of Automatica Sinica , 6(1):1--31, 2018
2018
- [6]
-
[7]
Bertsekas
Dimitri P. Bertsekas. Reinforcement Learning and Optimal Control . Athena Scientific, 2019
2019
-
[8]
Dimitri P. Bertsekas and John N. Tsitsiklis. Neuro-Dynamic Programming . Athena Scientific Belmont, MA, 1996
work page 1996
Show all 35 references
-
[9]
TTCP CAGE Challenge 2
CAGE. TTCP CAGE Challenge 2 . In AAAI-22 Workshop on Artificial Intelligence for Cyber Security (AICS) , 2022. https://github.com/cage-challenge/cage-challenge-2
2022
-
[10]
A survey of convergence results on particle filtering methods for practitioners
Dan Crisan and Arnaud Doucet. A survey of convergence results on particle filtering methods for practitioners . IEEE Transactions on Signal Processing , 50(3):736--746, 2002
2002
-
[11]
Cobbe, Jacob Hilton, Oleg Klimov, and John Schulman
Karl W. Cobbe, Jacob Hilton, Oleg Klimov, and John Schulman. Phasic policy gradient. In International Conference on Machine Learning , pages 2020--2027. PMLR, 2021
2020
-
[12]
Sunberg, Edward Balaban, Tim A
Maxim Egorov, Zachary N. Sunberg, Edward Balaban, Tim A. Wheeler, Jayesh K. Gupta, and Mykel J. Kochenderfer. POMDP s.jl: A framework for sequential decision making under uncertainty. Journal of Machine Learning Research , 18(26):1--5, 2017
2017
-
[13]
Lupu, and Dimitri Bertsekas
Kim Hammar, Yuchao Li, Tansu Alpcan, Emil C. Lupu, and Dimitri Bertsekas. Adaptive network security policies via belief aggregation and rollout, 2025. Working paper, to appear
2025
-
[14]
Software for the paper ``Feature-Based Belief Aggregation for Partially Observable Markov Decision Problems" , 2025
Kim Hammar, Yuchao Li, and Dimitri Bertsekas. Software for the paper ``Feature-Based Belief Aggregation for Partially Observable Markov Decision Problems" , 2025. Example code is available at https://github.com/Limmen/feature_based_pomdp_aggregation. The CAGE-2 code is availab...
2025
-
[15]
Partially Observed Markov Decision Processes: From Filtering to Controlled Sensing
Vikram Krishnamurthy. Partially Observed Markov Decision Processes: From Filtering to Controlled Sensing . Cambridge University Press, 2016
2016
-
[16]
An error bound for aggregation in approximate dynamic programming
Yuchao Li and Dimitri Bertsekas. An error bound for aggregation in approximate dynamic programming. arXiv preprint arXiv:2507.01324 , 2025
2025 arXiv
-
[17]
Littman, Anthony R
Michael L. Littman, Anthony R. Cassandra, and Leslie Pack Kaelbling. Learning policies for partially observable environments: Scaling up. In Machine Learning Proceedings 1995 , pages 362--370. Elsevier, 1995
1995
-
[18]
William S. Lovejoy. Computationally feasible bounds for partially observed Markov decision processes. Operations Research , 39(1):162--175, 1991
1991
-
[19]
Recurrent model-free RL can be a strong baseline for many POMDP s
Tianwei Ni, Benjamin Eysenbach, and Ruslan Salakhutdinov. Recurrent model-free RL can be a strong baseline for many POMDP s. In Proceedings of the 39th International Conference on Machine Learning , volume 162 of Proceedings of Machine Learning Research , pages 16691--16723. P...
2022
-
[20]
Ong, Shao Wei Png, David Hsu, and Wee Sun Lee
Sylvie C.W. Ong, Shao Wei Png, David Hsu, and Wee Sun Lee. Planning under uncertainty for robotic tasks with mixed observability. The International Journal of Robotics Research , 29(8):1053--1068, 2010
2010
-
[21]
Value-directed compression of POMDPs
Pascal Poupart and Craig Boutilier. Value-directed compression of POMDPs . In Advances in Neural Information Processing Systems , volume 15. MIT Press, 2002
2002
-
[22]
Anytime point-based approximations for large POMDPs
Joelle Pineau, Geoffrey Gordon, and Sebastian Thrun. Anytime point-based approximations for large POMDPs . Journal of Artificial Intelligence Research , 27:335--380, 2006
2006
-
[23]
Finding approximate POMDP solutions through belief compression
Nicholas Roy, Geoffrey Gordon, and Sebastian Thrun. Finding approximate POMDP solutions through belief compression. Journal of artificial intelligence research , 23:1--40, 2005
2005
-
[24]
Sunberg and Mykel J
Zachary N. Sunberg and Mykel J. Kochenderfer. Online algorithms for POMDP s with continuous state, action, and observation spaces. In Proceedings of the 28th International Conference on Automated Planning and Scheduling (ICAPS) , 2018
2018
-
[25]
Edward J. Sondik. The optimal control of partially observable Markov processes over the infinite horizon: Discounted costs. Operations Research , 26(2):282--304, 1978
1978
-
[26]
Heuristic search value iteration for POMDPs
Trey Smith and Reid Simmons. Heuristic search value iteration for POMDPs . In Proceedings of the 20th Conference on Uncertainty in Artificial Intelligence , UAI '04, page 520–527. AUAI Press, 2004
2004
-
[27]
Monte-Carlo planning in large POMDPs
David Silver and Joel Veness. Monte-Carlo planning in large POMDPs . In Advances in Neural Information Processing Systems , volume 23, 2010
2010
-
[28]
Proximal policy optimization algorithms
John Schulman, Filip Wolski, Prafulla Dhariwal, Alec Radford, and Oleg Klimov. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347 , 2017
2017 arXiv
-
[29]
DESPOT : Online POMDP planning with regularization
Adhiraj Somani, Nan Ye, David Hsu, and Wee Sun Lee. DESPOT : Online POMDP planning with regularization. Advances in neural information processing systems , 26, 2013
2013
-
[30]
On the asymptotic optimality of finite approximations to Markov decision processes with Borel spaces
Naci Saldi, Serdar Y\" u ksel, and Tam\' a s Linder. On the asymptotic optimality of finite approximations to Markov decision processes with Borel spaces. Mathematics of Operations Research , 42(4):945--978, 2017
2017
-
[31]
Tsitsiklis and Benjamin van Roy
John N. Tsitsiklis and Benjamin van Roy. Feature-based methods for large scale dynamic programming . Machine Learning , 22(1):59--94, Mar 1996
1996
-
[32]
Adaptive online packing-guided search for POMDPs
Chenyang Wu, Guoyu Yang, Zongzhang Zhang, Yang Yu, Dong Li, Wulong Liu, and Jianye Hao. Adaptive online packing-guided search for POMDPs . In Advances in Neural Information Processing Systems , volume 34, pages 28419--28430. Curran Associates, Inc., 2021
2021
-
[33]
Bertsekas
Huizhen Yu and Dimitri P. Bertsekas. Discretized approximations for POMDP with average cost. In Proceedings of the 20th Conference on Uncertainty in Artificial Intelligence , UAI '04, page 619–627, Arlington, Virginia, USA, 2004. AUAI Press
2004
-
[34]
Rong Zhou and Eric A. Hansen. An improved grid-based approximation algorithm for POMDPs . In International Joint Conference on Artificial Intelligence , 2001
2001
-
[35]
Optimal control of Markov processes with incomplete state information
Karl Johan Åström. Optimal control of Markov processes with incomplete state information . Journal of Mathematical Analysis and Applications , 10(1):174--205, 1965
1965
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.