English

Nonparametric Additive Value Functions: Interpretable Reinforcement Learning with an Application to Surgical Recovery

Machine Learning 2025-06-02 v2 Machine Learning

Abstract

We propose a nonparametric additive model for estimating interpretable value functions in reinforcement learning, with an application in optimizing postoperative recovery through personalized, adaptive recommendations. While reinforcement learning has achieved significant success in various domains, recent methods often rely on black-box approaches such as neural networks, which hinder the examination of individual feature contributions to a decision-making policy. Our novel method offers a flexible technique for estimating action-value functions without explicit parametric assumptions, overcoming the limitations of the linearity assumption of classical algorithms. By incorporating local kernel regression and basis expansion, we obtain a sparse, additive representation of the action-value function, enabling local approximation and retrieval of nonlinear, independent contributions of select state features and the interactions between joint feature pairs. We validate our approach through a simulation study and apply it to spine disease recovery, uncovering recommendations aligned with clinical knowledge. This method bridges the gap between flexible machine learning techniques and the interpretability required in healthcare applications, paving the way for more personalized interventions.

Keywords

Cite

@article{arxiv.2308.13135,
  title  = {Nonparametric Additive Value Functions: Interpretable Reinforcement Learning with an Application to Surgical Recovery},
  author = {Patrick Emedom-Nnamdi and Timothy R. Smith and Jukka-Pekka Onnela and Junwei Lu},
  journal= {arXiv preprint arXiv:2308.13135},
  year   = {2025}
}

Comments

28 pages, 13 figures

R2 v1 2026-06-28T12:03:57.815Z