MDPs with a State Sensing Cost
Vansh Kapoor, Jayakrishnan Nair
TL;DR
This work addresses MDPs with an explicit cost for state sensing, formulating an augmented, countably infinite MDP \\mathcal{M}_k\\ that allows the agent to sense or remain blind at each step. It introduces SPI, a selective policy-improvement algorithm that greedily searches for improving blind-action sequences and leverages a myopic sensing value to guide updates, achieving near-optimal performance in practice. The authors establish provable guarantees through a threshold on the sensing cost for always-sense optimality and a truncation framework that yields computable lower bounds and suboptimality measures, enabling tractable planning for large or continuous state spaces. Empirically, SPI outperforms state-of-the-art POMDP solvers and the ATM heuristic across diverse domains (e.g., ICU-Sepsis, Frozen Lake, Taxi) with reasonable compute times, and extensions to non-uniform costs and continuous spaces are discussed, highlighting practical impact for healthcare, robotics, and sensor networks.
Abstract
In many practical sequential decision-making problems, tracking the state of the environment incurs a sensing/communication/computation cost. In these settings, the agent's interaction with its environment includes the additional component of deciding when to sense the state, in a manner that balances the value associated with optimal (state-specific) actions and the cost of sensing. We formulate this as an expected discounted cost Markov Decision Process (MDP), wherein the agent incurs an additional cost for sensing its next state, but has the option to take actions while remaining `blind' to the system state. We pose this problem as a classical discounted cost MDP with an expanded (countably infinite) state space. While computing the optimal policy for this MDP is intractable in general, we derive lower bounds on the optimal value function, which allow us to bound the suboptimality gap of any policy. We also propose a computationally efficient algorithm SPI, based on policy improvement, which in practice performs close to the optimal policy. Finally, we benchmark against the state-of-the-art via a numerical case study.
