Pragmatic Policy Development via Interpretable Behavior Cloning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Matsson, Anton, Rao, Yaochen, Litman, Heather J., Johansson, Fredrik D.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916858412662784
author Matsson, Anton
Rao, Yaochen
Litman, Heather J.
Johansson, Fredrik D.
author_facet Matsson, Anton
Rao, Yaochen
Litman, Heather J.
Johansson, Fredrik D.
contents Offline reinforcement learning (RL) holds great promise for deriving optimal policies from observational data, but challenges related to interpretability and evaluation limit its practical use in safety-critical domains. Interpretability is hindered by the black-box nature of unconstrained RL policies, while evaluation -- typically performed off-policy -- is sensitive to large deviations from the data-collecting behavior policy, especially when using methods based on importance sampling. To address these challenges, we propose a simple yet practical alternative: deriving treatment policies from the most frequently chosen actions in each patient state, as estimated by an interpretable model of the behavior policy. By using a tree-based model, which is specifically designed to exploit patterns in the data, we obtain a natural grouping of states with respect to treatment. The tree structure ensures interpretability by design, while varying the number of actions considered controls the degree of overlap with the behavior policy, enabling reliable off-policy evaluation. This pragmatic approach to policy development standardizes frequent treatment patterns, capturing the collective clinical judgment embedded in the data. Using real-world examples in rheumatoid arthritis and sepsis care, we demonstrate that policies derived under this framework can outperform current practice, offering interpretable alternatives to those obtained via offline RL.
format Preprint
id arxiv_https___arxiv_org_abs_2507_17056
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Pragmatic Policy Development via Interpretable Behavior Cloning
Matsson, Anton
Rao, Yaochen
Litman, Heather J.
Johansson, Fredrik D.
Machine Learning
Artificial Intelligence
Offline reinforcement learning (RL) holds great promise for deriving optimal policies from observational data, but challenges related to interpretability and evaluation limit its practical use in safety-critical domains. Interpretability is hindered by the black-box nature of unconstrained RL policies, while evaluation -- typically performed off-policy -- is sensitive to large deviations from the data-collecting behavior policy, especially when using methods based on importance sampling. To address these challenges, we propose a simple yet practical alternative: deriving treatment policies from the most frequently chosen actions in each patient state, as estimated by an interpretable model of the behavior policy. By using a tree-based model, which is specifically designed to exploit patterns in the data, we obtain a natural grouping of states with respect to treatment. The tree structure ensures interpretability by design, while varying the number of actions considered controls the degree of overlap with the behavior policy, enabling reliable off-policy evaluation. This pragmatic approach to policy development standardizes frequent treatment patterns, capturing the collective clinical judgment embedded in the data. Using real-world examples in rheumatoid arthritis and sepsis care, we demonstrate that policies derived under this framework can outperform current practice, offering interpretable alternatives to those obtained via offline RL.
title Pragmatic Policy Development via Interpretable Behavior Cloning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2507.17056