Stabilized direct learning for efficient estimation of individualized treatment rules

© 2022 The International Biometric Society..

In recent years, the field of precision medicine has seen many advancements. Significant focus has been placed on creating algorithms to estimate individualized treatment rules (ITRs), which map from patient covariates to the space of available treatments with the goal of maximizing patient outcome. Direct learning (D-Learning) is a recent one-step method which estimates the ITR by directly modeling the treatment-covariate interaction. However, when the variance of the outcome is heterogeneous with respect to treatment and covariates, D-Learning does not leverage this structure. Stabilized direct learning (SD-Learning), proposed in this paper, utilizes potential heteroscedasticity in the error term through a residual reweighting which models the residual variance via flexible machine learning algorithms such as XGBoost and random forests. We also develop an internal cross-validation scheme which determines the best residual model among competing models. SD-Learning improves the efficiency of D-Learning estimates in binary and multi-arm treatment scenarios. The method is simple to implement and an easy way to improve existing algorithms within the D-Learning family, including original D-Learning, Angle-based D-Learning (AD-Learning), and Robust D-learning (RD-Learning). We provide theoretical properties and justification of the optimality of SD-Learning. Head-to-head performance comparisons with D-Learning methods are provided through simulations, which demonstrate improvement in terms of average prediction error (APE), misclassification rate, and empirical value, along with a data analysis of an acquired immunodeficiency syndrome (AIDS) randomized clinical trial.

Medienart:

E-Artikel

Erscheinungsjahr:

2023

Erschienen:

2023

Enthalten in:

Zur Gesamtaufnahme - volume:79

Enthalten in:

Biometrics - 79(2023), 4 vom: 29. Dez., Seite 2843-2856

Sprache:

Englisch

Beteiligte Personen:

Shah, Kushal S [VerfasserIn]
Fu, Haoda [VerfasserIn]
Kosorok, Michael R [VerfasserIn]

Links:

Volltext

Themen:

D-Learning
Heteroscedasticity
Individualized treatment rule
Journal Article
Multi-arm treatments
Precision medicine
Randomized Controlled Trial
Statistical machine learning

Anmerkungen:

Date Completed 21.12.2023

Date Revised 31.01.2024

published: Print-Electronic

Citation Status MEDLINE

doi:

10.1111/biom.13818

funding:

Förderinstitution / Projekttitel:

PPN (Katalog-ID):

NLM350972427