Robust sample weighting to facilitate individualized treatment rule learning for a target population

Rui Chen, Jared D. Huling, Guanhua Chen, Menggang Yu

Research output: Contribution to journalArticlepeer-review

Abstract

Learning individualized treatment rules is an important topic in precision medicine. Current literature mainly focuses on deriving individualized treatment rules from a single source population. We consider the observational data setting when the source population differs from a target population of interest. Compared with causal generalization for the average treatment effect that is a scalar quantity, individualized treatment rule generalization poses new challenges due to the need to model and generalize the rules based on a prespecified class of functions that may not contain the unrestricted true optimal individualized treatment rule. The aim of this paper is to develop a weighting framework to mitigate the impact of such misspecification, and thus facilitate the generalizability of optimal individualized treatment rules from a source population to a target population. Our method seeks covariate balance over a nonparametric function class characterized by a reproducing kernel Hilbert space and can improve many individualized treatment rule learning methods that rely on weights. We show that the proposed method encompasses importance weights and overlap weights as two extreme cases, allowing for a better bias-variance trade-off in between. Numerical examples demonstrate that the use of our weighting method can greatly improve individualized treatment rule estimation for the target population compared with other weighting methods.

Original languageEnglish (US)
Pages (from-to)309-329
Number of pages21
JournalBiometrika
Volume111
Issue number1
DOIs
StatePublished - Mar 1 2024

Bibliographical note

Publisher Copyright:
© 2024 Oxford University Press. All rights reserved.

Keywords

  • Covariate balancing
  • Covariate shift
  • Generalizability
  • Precision medicine
  • Weighting adjustment

Fingerprint

Dive into the research topics of 'Robust sample weighting to facilitate individualized treatment rule learning for a target population'. Together they form a unique fingerprint.

Cite this