Research · Clinical and scientific decision-making

Clinical and scientific decision-making

The applied settings that generate the lab's methodological problems — clinical care and automated experimentation.

Overview

The lab’s other four areas are methodological; this one supplies the problems they address. Clinical care exhibits every structural difficulty at once: the recorded state is a fraction of the patient’s state, the log was generated by clinicians conditioning on information that was never written down, outcomes resolve over days, some trajectories cannot be recovered from, and a single decision fixes several interacting quantities. Automated experimentation presents the same structure under a different cost model — select the next configuration under uncertainty while earlier attempts remain unresolved, against a budget that makes exploration materially expensive. The two settings share enough structure for methods to transfer and differ enough that the transfer is diagnostic.

Themes

  • Partial observability and state construction — a policy over clinical time series acts on a learned summary of an irregularly sampled history, and the summary is a modeling decision with measurable consequences for the resulting policy (An Empirical Study of Representation Learning for Reinforcement Learning in Healthcare)
  • Confounded observational records — treatment decisions in a medical record respond to unobserved covariates, which makes the identifying assumptions rather than the estimator the substantive part of any claim; see offline RL
  • Transfer across populations and sites — grounding transfer in counterfactual reasoning rather than distributional similarity, and encoding the invariances that make it possible as inductive bias (Counterfactual Transfer via Inductive Bias in Clinical Settings)
  • Unrecoverable clinical states — negative outcomes in a record identify states and treatments to avoid even where the data cannot support an optimal policy (Medical Dead-ends and Learning to Identify High-Risk States and Treatments), which is developed further under irreversibility and risk
  • Composed interventions — clinical and experimental actions set multiple interacting quantities per decision, so the faithful formulation is combinatorial; see structured action spaces
  • Person-centered policies — the target is not a policy that is optimal in expectation over a population but one that adapts to the individual and the context, holds up outside its training distribution, and respects the preferences and constraints of its intended use
  • Sequential experimental design — formulating automated experimentation as a sequential decision problem in which the readout is a delayed proxy and prior decisions are still pending; see delayed and unreliable feedback

Open questions

  • Which assumptions a decision-support claim rests on, and how to state them so that a clinician or experimentalist can judge whether they hold in a given case
  • How to distinguish variation in observed treatment attributable to physiology from variation attributable to clinician judgment or resource constraint, and how to keep a learned policy from reproducing the latter
  • What evidence is required before a policy learned offline is worth evaluating prospectively, and what an evaluation protocol short of deployment can establish
  • How much of the clinical formulation carries over to experimental pipelines, where the horizon is set by an experiment cycle rather than by physiology

Also relevant: Clinically Motivated Sequential Decision Making Under Uncertainty in Offline Settings, which sets out the modeling decisions that anchor method development in the intended use case rather than in a benchmark.

  • Healthcare
  • Clinical
  • Scientific Discovery

Publications

Related papers

2024

Clinically Motivated Sequential Decision Making Under Uncertainty in Offline Settings

Taylor W. Killian

PhD Thesis, University of Toronto, Department of Computer Science

The modeling decisions that let you draw actionable conclusions from sequentially observed healthcare data — anchoring method development in the intended real-world use case rather than the benchmark.

2023

Risk Sensitive Dead-end Identification in Safety-Critical Offline Reinforcement Learning

Taylor W. Killian, Sonali Parbhoo, Marzyeh Ghassemi

Transactions on Machine Learning Research (TMLR)

A risk-sensitive treatment of dead-end discovery that uses distributional RL for value estimation, flagging unrecoverable states earlier and tunably, according to the risk tolerance of the task.

Continuous Time Evidential Distributions for Irregular Time Series

Taylor W. Killian, Haoran Zhang, Thomas Hartvigsen, Ava Amini

Interpretable Machine Learning in Healthcare Workshop, ICML 2023

Extends evidential deep learning to continuous time so it can handle the irregularly sampled series that clinical data actually produces, giving stable predictions and calibrated uncertainty that tightens as evidence accumulates.

2022

Counterfactually Guided Policy Transfer in Clinical Settings

Taylor W. Killian, Marzyeh Ghassemi, Shalmali Joshi

Conference on Health, Inference and Learning (CHIL) 2022

2021

Medical Dead-ends and Learning to Identify High-Risk States and Treatments

Mehdi Fatemi, Taylor W. Killian, Jayakumar Subramanian, Marzyeh Ghassemi

NeurIPS 2021

In data-constrained offline settings an optimal policy may simply not be recoverable. Negative outcomes in the data can still be used to identify behaviors to avoid, guarding against the overoptimistic decisions that reduced data availability invites.

2020

An Empirical Study of Representation Learning for Reinforcement Learning in Healthcare

Taylor W. Killian, Haoran Zhang, Jayakumar Subramanian, Mehdi Fatemi, Marzyeh Ghassemi

ML4H: Machine Learning for Health Workshop at NeurIPS

How the choice of state representation changes what an offline RL agent learns from clinical time series, evaluated across a range of representation learning approaches.

Counterfactual Transfer via Inductive Bias in Clinical Settings

Taylor W. Killian, Marzyeh Ghassemi, Shalmali Joshi

Inductive Biases, Invariances and Generalization in RL (BIG) Workshop, ICML

All publications