dorsal/arxiv
View SchemaDimension-reduced outcome-weighted learning for estimating individualized treatment regimes in observational studies
| Authors | Sungtaek Son, Eardi Lila, Kwun Chuen Gary Chan |
|---|---|
| Categories | |
| ArXiv ID | 2601.06782vv1 |
| URL | https://arxiv.org/abs/2601.06782 |
| License | http://creativecommons.org/licenses/by/4.0/ |
Abstract
Individualized treatment regimes (ITRs) aim to improve clinical outcomes by assigning treatment based on patient-specific characteristics. However, existing methods often struggle with high-dimensional covariates, limiting accuracy, interpretability, and real-world applicability. We propose a novel sufficient dimension reduction approach that directly targets the contrast between potential outcomes and identifies a low-dimensional subspace of the covariates capturing treatment effect heterogeneity. This reduced representation enables more accurate estimation of optimal ITRs through outcome-weighted learning. To accommodate observational data, our method incorporates kernel-based covariate balancing, allowing treatment assignment to depend on the full covariate set and avoiding the restrictive assumption that the subspace sufficient for modeling heterogeneous treatment effects is also sufficient for confounding adjustment. We show that the proposed method achieves universal consistency, i.e., its risk converges to the Bayes risk, under mild regularity conditions. We demonstrate its finite sample performance through simulations and an analysis of intensive care unit sepsis patient data to determine who should receive transthoracic echocardiography.
{
"annotation_id": "ac5c451f-d56e-40eb-900f-adc72332bf9d",
"date_created": "2026-02-17T05:53:08.761000Z",
"date_modified": "2026-02-17T05:53:08.761000Z",
"file_hash": "6e249dc286071ba1c0a5cda4d424b83b9c1cb2c849b9e4274d6e08e81449af06",
"private": false,
"record": {
"abstract": "Individualized treatment regimes (ITRs) aim to improve clinical outcomes by assigning treatment based on patient-specific characteristics. However, existing methods often struggle with high-dimensional covariates, limiting accuracy, interpretability, and real-world applicability. We propose a novel sufficient dimension reduction approach that directly targets the contrast between potential outcomes and identifies a low-dimensional subspace of the covariates capturing treatment effect heterogeneity. This reduced representation enables more accurate estimation of optimal ITRs through outcome-weighted learning. To accommodate observational data, our method incorporates kernel-based covariate balancing, allowing treatment assignment to depend on the full covariate set and avoiding the restrictive assumption that the subspace sufficient for modeling heterogeneous treatment effects is also sufficient for confounding adjustment. We show that the proposed method achieves universal consistency, i.e., its risk converges to the Bayes risk, under mild regularity conditions. We demonstrate its finite sample performance through simulations and an analysis of intensive care unit sepsis patient data to determine who should receive transthoracic echocardiography.",
"arxiv_id": "2601.06782",
"authors": [
"Sungtaek Son",
"Eardi Lila",
"Kwun Chuen Gary Chan"
],
"categories": [
"stat.ML",
"cs.LG",
"stat.ME"
],
"license": "http://creativecommons.org/licenses/by/4.0/",
"title": "Dimension-reduced outcome-weighted learning for estimating individualized treatment regimes in observational studies",
"url": "https://arxiv.org/abs/2601.06782",
"version": "v1"
},
"schema_id": "dorsal/arxiv",
"source": {
"execution_id": "713ecf24-8d0b-4567-95e6-959c96f44139",
"id": "arXiv Dataset",
"type": "Model",
"variant": "snapshot-2026-01-17",
"version": "0.1.0"
},
"user_id": 1000002
}