arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2405.03083stat.MEcs.LGstat.ML
收藏

因果k均值聚类

Causal K-Means Clustering

Department of Statistics, Korea University(高丽大学统计系);Department of Statistics, Seoul National University(首尔大学统计系);Department of Statistics and Data Science, Carnegie Mellon University(卡内基梅隆大学统计与数据科学系)

Kwangho Kim, Jisu Kim, Edward H. Kennedy

AI总结:

本文提出因果k均值聚类方法,用于识别未知的子群体结构,通过非参数效率理论和双机器学习开发偏倚校正估计器,实现快速根n收敛和渐近正态性,适用于多治疗水平的现代结果研究。

摘要:

因果效应通常通过总体汇总来表征。当存在异质治疗效应时,这些汇总可能提供不完整的画面。由于子群体结构通常未知,识别和评估子群体效应比总体效应更具挑战性。我们提出了一种新的解决方案:因果k均值聚类,利用k均值聚类算法来揭示未知的子群体结构。我们的研究问题与传统聚类设置有显著不同,因为要聚类的变量是未知的反事实函数。我们提出了一个插件估计器,该估计器使用现成的算法简单且易于实现,并研究了其收敛速率。我们还开发了基于非参数效率理论和双机器学习的新偏倚校正估计器,并证明该估计器在大型非参数模型中实现了快速根n速率和渐近正态性。我们提出的方法特别适用于具有多个治疗水平的现代结果研究。此外,我们的框架可以扩展到使用通用伪结果的聚类,如部分观察到的结果或其他未知函数。最后,我们通过模拟探索有限样本性质,并使用移动支持的慢性低背痛自我管理研究来说明所提出的方法。

英文摘要:

Causal effects are often characterized with population summaries. These might provide an incomplete picture when there are heterogeneous treatment effects across subgroups. Since the subgroup structure is typically unknown, it is more challenging to identify and evaluate subgroup effects than population effects. We propose a new solution to this problem: \emph{Causal k-Means Clustering}, which leverages the k-means clustering algorithm to uncover the unknown subgroup structure. Our problem differs significantly from the conventional clustering setup since the variables to be clustered are unknown counterfactual functions. We present a plug-in estimator which is simple and readily implementable using off-the-shelf algorithms, and study its rate of convergence. We also develop a new bias-corrected estimator based on nonparametric efficiency theory and double machine learning, and show that this estimator achieves fast root-n rates and asymptotic normality in large nonparametric models. Our proposed methods are especially useful for modern outcome-wide studies with multiple treatment levels. Further, our framework is extensible to clustering with generic pseudo-outcomes, such as partially observed outcomes or otherwise unknown functions. Finally, we explore finite sample properties via simulation, and illustrate the proposed methods using a study of mobile-supported self-management for chronic low back pain.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑