Self-Distilled Disentangled Learning for Counterfactual Prediction

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Xinshu, Gong, Mingming, Yao, Lina
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914834197512192
author Li, Xinshu
Gong, Mingming
Yao, Lina
author_facet Li, Xinshu
Gong, Mingming
Yao, Lina
contents The advancements in disentangled representation learning significantly enhance the accuracy of counterfactual predictions by granting precise control over instrumental variables, confounders, and adjustable variables. An appealing method for achieving the independent separation of these factors is mutual information minimization, a task that presents challenges in numerous machine learning scenarios, especially within high-dimensional spaces. To circumvent this challenge, we propose the Self-Distilled Disentanglement framework, referred to as $SD^2$. Grounded in information theory, it ensures theoretically sound independent disentangled representations without intricate mutual information estimator designs for high-dimensional representations. Our comprehensive experiments, conducted on both synthetic and real-world datasets, confirms the effectiveness of our approach in facilitating counterfactual inference in the presence of both observed and unobserved confounders.
format Preprint
id arxiv_https___arxiv_org_abs_2406_05855
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Distilled Disentangled Learning for Counterfactual Prediction
Li, Xinshu
Gong, Mingming
Yao, Lina
Machine Learning
Artificial Intelligence
The advancements in disentangled representation learning significantly enhance the accuracy of counterfactual predictions by granting precise control over instrumental variables, confounders, and adjustable variables. An appealing method for achieving the independent separation of these factors is mutual information minimization, a task that presents challenges in numerous machine learning scenarios, especially within high-dimensional spaces. To circumvent this challenge, we propose the Self-Distilled Disentanglement framework, referred to as $SD^2$. Grounded in information theory, it ensures theoretically sound independent disentangled representations without intricate mutual information estimator designs for high-dimensional representations. Our comprehensive experiments, conducted on both synthetic and real-world datasets, confirms the effectiveness of our approach in facilitating counterfactual inference in the presence of both observed and unobserved confounders.
title Self-Distilled Disentangled Learning for Counterfactual Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2406.05855