Threshold Structure of Optimal Policies in Restart POMDPs

arXiv:2608.10936 · math.OC, cs.LG, math.PR · Submitted 2026-08-11 · Read on arXiv

Konstantin Avrachenkov, Alexey Piunovskiy, Yi Zhang

inria · University of Liverpool · University of Birmingham

math.OC, cs.LG, math.PR

Submitted: 2026-08-11

Updated: 2026-08-12

Comments: 8 pages, the paper has been accepted to IEEE CDC 2026

License: http://creativecommons.org/licenses/by/4.0/

Importance score: 75/100

The gist: We study a Restart POMDP (Partially Observable Markov Decision Process) on a general Borel state space, where the controller either lets the hidden state evolve unobserved or restarts the system and

Terminology

Summary

We study a Restart POMDP (Partially Observable Markov Decision Process) on a general Borel state space, where the controller either lets the hidden state evolve unobserved or restarts the system and observes the new state. Exploiting a sufficient-statistic representation consisting of the last observed state and the elapsed time since restart, we reduce the problem to a fully observed MDP. Under a natural one-step cost deterioration condition, we prove that optimal policies have a threshold structure in the elapsed time for both the discounted and total undiscounted cost criteria. When the state space is partially ordered and the kernel is stochastically monotone, we further show that the optimal threshold is nonincreasing in the state. For the average cost criterion, under additional assumptions of geometric ergodicity and domination of the transient gain, we establish analogous threshold results via the vanishing discount approach, after showing the uniform boundedness of the optimal thresholds and relative value functions.

Improvements for AI systems

Improvements to AI Systems:

  1. Adaptive Restart Decision-Making for Reinforcement Learning (RL) Agents:
  • Improvement: Integrate the threshold-based restart policy into RL agents operating in partially observable environments (e.g., robotics, network management). Instead of relying on heuristics or learned value functions alone, the agent can use the proven threshold structure (optimal restart time as a function of state) to decide when to reset its internal belief or environment, reducing exploration costs and improving sample efficiency.

  • Capability: The AI can autonomously determine the optimal moment to restart (e.g., reinitialize a simulation, reset a robot's pose, or refresh a recommendation context) based on elapsed time and last observed state, even in continuous, high-dimensional spaces.

  1. Provably Optimal Maintenance and Repair Scheduling in Predictive Maintenance:
  • Improvement: Apply the Restart POMDP framework to industrial systems where hidden degradation states evolve unobserved. The threshold policy (nonincreasing in state) provides a formal rule for when to perform a restart (e.g., replace a component) given the last inspection state and time since last maintenance.

  • Capability: An AI maintenance scheduler can guarantee cost-optimal intervention times under both discounted and long-run average cost criteria, even when degradation is only partially observable, leading to lower downtime and reduced operational costs.

  1. Efficient Exploration in Partially Observable Environments with Reset Mechanisms:
  • Improvement: Use the sufficient-statistic reduction (last observed state + elapsed time) to design a simplified MDP for AI agents that can reset their belief (e.g., in active perception, sensor re-calibration, or episodic memory clearing). The threshold structure allows the agent to know exactly when to force a reset to maximize long-term reward.

  • Capability: AI systems (e.g., autonomous vehicles, search-and-rescue drones) can decide when to discard stale information and re-observe the environment, improving decision speed and robustness without exhaustive belief tracking.

  1. Optimal Control of Queuing and Inventory Systems with Hidden States:
  • Improvement: Model inventory or queue lengths as hidden states with stochastic monotonicity. The threshold policy (nonincreasing in state) gives a simple, implementable rule for when to restart (e.g., clear a queue, reorder stock) based on the last known state and time since last action.

  • Capability: An AI controller for supply chains or call centers can make real-time decisions on when to reset or replenish, with proven optimality under average-cost criteria, even when exact current state is unknown.

  1. Robust Decision-Making in Healthcare Monitoring:
  • Improvement: For chronic disease management (e.g., diabetes, cardiac arrhythmias), where patient state is partially observable, the threshold policy dictates when to restart treatment (e.g., re-administer a drug, reset a wearable sensor) based on last measured vitals and time elapsed.

  • Capability: An AI health monitor can issue alerts or interventions at provably optimal times, balancing treatment cost against health deterioration, and adapt thresholds nonincreasingly with severity of the last observed state.

  1. Improved Convergence and Stability in Multi-Agent Systems with Reset Protocols:
  • Improvement: Use the vanishing discount approach and uniform boundedness results to design AI agents that periodically reset their internal models or communication protocols, ensuring stable long-term performance without divergence.

  • Capability: Distributed AI systems (e.g., swarm robotics, federated learning) can automatically decide when to re-synchronize or restart local models, guaranteeing convergence to optimal average-cost behavior even under partial observability and stochastic disturbances.

Abstract

We study a Restart POMDP (Partially Observable Markov Decision Process) on a general Borel state space, where the controller either lets the hidden state evolve unobserved or restarts the system and observes the new state. Exploiting a sufficient-statistic representation consisting of the last observed state and the elapsed time since restart, we reduce the problem to a fully observed MDP. Under a natural one-step cost deterioration condition, we prove that optimal policies have a threshold structure in the elapsed time for both the discounted and total undiscounted cost criteria. When the state space is partially ordered and the kernel is stochastically monotone, we further show that the optimal threshold is nonincreasing in the state. For the average cost criterion, under additional assumptions of geometric ergodicity and domination of the transient gain, we establish analogous threshold results via the vanishing discount approach, after showing the uniform boundedness of the optimal thresholds and relative value functions.

Related papers