English

Action-Conditioned Risk Gating for Safety-Critical Control under Partial Observability

Machine Learning 2026-05-15 v1 Artificial Intelligence Systems and Control Systems and Control

Abstract

Many safety-critical control problems are modeled as risk-sensitive partially observable Markov decision processes, where the controller must make decisions from incomplete observations while balancing task performance against safety risk. Although belief-space planning provides a principled solution, maintaining and planning over beliefs can be computationally costly and sensitive to model specification in practical domains. We propose a lightweight risk-gated reinforcement learning approximation for risk-sensitive control under partial observability. The method constructs a compact finite-history proxy state and learns an action-conditioned predictor of near-term safety violation. This predicted candidate-action risk is used in two complementary ways: as a risk penalty during value learning, and as a decision-time gate that interpolates between optimistic and conservative ensemble value estimates. As a result, low-risk actions are evaluated closer to reward-seeking estimates, while high-risk actions are evaluated more conservatively. We evaluate the approach in two safety-critical partially observable domains: automated glucose regulation and safety-constrained navigation. Across adult and adolescent glucose-control cohorts, the method improves overall glycemic tradeoffs and substantially reduces runtime relative to a belief-space planning baseline. On Safety-Gym navigation benchmarks, it achieves a more favorable reward-cost balance than unconstrained RL and several standard safe-RL baselines. These results suggest that action-conditioned near-term risk can provide an effective local signal for approximate risk-sensitive POMDP control when full belief-space planning is impractical.

Keywords

Cite

@article{arxiv.2605.14246,
  title  = {Action-Conditioned Risk Gating for Safety-Critical Control under Partial Observability},
  author = {Yushen Liu and Yin-Jen Chen and Ziyi Chen and Tao Wang and Heng Huang and Xugui Zhou and Yanfu Zhang},
  journal= {arXiv preprint arXiv:2605.14246},
  year   = {2026}
}
R2 v1 2026-07-22T07:11:23.761Z