Causal K-Means Clustering

2024-05-05 · Kwangho Kim, Jisu Kim, Edward H. Kennedy

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: Causal k-Means Clustering, which harnesses the widely-used 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 in a study of treatment programs for adolescent substance abuse.

📄 PDF Abstract BibTeX arXiv:2405.03083

Code (0)

등록된 구현이 없습니다.

Tasks

Clusteringcounterfactual

Methods 이 논문이 사용한 방법론

k-Means Clustering k-Means Clustering is a clustering algorithm that divides a training set into $k$ different clusters of examples that are near each other. It works by initializing $k$…

Time-Series K-means in Causal Inference and Mechanism Clustering for Financial Data

2022-01-24 · Minheng Xiao

This paper investigates the application of Time Series K-means (TS-K-means) within the context of causal inference and mechanism clustering of financial time series data. Traditional clustering approaches like K-means of…

Causal Inference Clustering Dynamic Time Warping Time Series

Hierarchical and Density-based Causal Clustering

2024-11-02 · Kwangho Kim, Jisu Kim, Larry A. Wasserman, Edward H. Kennedy

Understanding treatment effect heterogeneity is vital for scientific and policy research. However, identifying and evaluating heterogeneous treatment effects pose significant challenges due to the typically unknown subgr…

Clustering counterfactual regression

Multilayer GNN for Predictive Maintenance and Clustering in Power Grids

2025-07-09 · Muhammad Kazim, Harun Pirim, Chau Le, Trung Le 외 arxiv

Unplanned power outages cost the US economy over $150 billion annually, partly due to predictive maintenance (PdM) models that overlook spatial, temporal, and causal dependencies in grid failures. This study introduces a…

Graph Neural Network

Normalization based K means Clustering Algorithm

2015-03-03 · Deepali Virmani, Shweta Taneja, Geetika Malhotra

K-means is an effective clustering technique used to separate similar data into groups based on initial centroids of clusters. In this paper, Normalization based K-means clustering algorithm(N-K means) is proposed. Propo…

Clustering

Randomized Dimensionality Reduction for k-means Clustering

2011-10-13 · Christos Boutsidis, Anastasios Zouzias, Michael W. Mahoney, Petros Drineas

We study the topic of dimensionality reduction for $k$-means clustering. Dimensionality reduction encompasses the union of two approaches: \emph{feature selection} and \emph{feature extraction}. A feature selection based…

Clustering Dimensionality Reduction feature selection