Semi-Supervised Learning under General Causal Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Moore, Archer, Shim, Heejung, Zhu, Jingge, Gong, Mingming
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912670722031616
author Moore, Archer
Shim, Heejung
Zhu, Jingge
Gong, Mingming
author_facet Moore, Archer
Shim, Heejung
Zhu, Jingge
Gong, Mingming
contents Semi-supervised learning (SSL) aims to train a machine learning model using both labelled and unlabelled data. While the unlabelled data have been used in various ways to improve the prediction accuracy, the reason why unlabelled data could help is not fully understood. One interesting and promising direction is to understand SSL from a causal perspective. In light of the independent causal mechanisms principle, the unlabelled data can be helpful when the label causes the features but not vice versa. However, the causal relations between the features and labels can be complex in real world applications. In this paper, we propose a SSL framework that works with general causal models in which the variables have flexible causal relations. More specifically, we explore the causal graph structures and design corresponding causal generative models which can be learned with the help of unlabelled data. The learned causal generative model can generate synthetic labelled data for training a more accurate predictive model. We verify the effectiveness of our proposed method by empirical studies on both simulated and real data.
format Preprint
id arxiv_https___arxiv_org_abs_2510_22567
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Semi-Supervised Learning under General Causal Models
Moore, Archer
Shim, Heejung
Zhu, Jingge
Gong, Mingming
Machine Learning
Semi-supervised learning (SSL) aims to train a machine learning model using both labelled and unlabelled data. While the unlabelled data have been used in various ways to improve the prediction accuracy, the reason why unlabelled data could help is not fully understood. One interesting and promising direction is to understand SSL from a causal perspective. In light of the independent causal mechanisms principle, the unlabelled data can be helpful when the label causes the features but not vice versa. However, the causal relations between the features and labels can be complex in real world applications. In this paper, we propose a SSL framework that works with general causal models in which the variables have flexible causal relations. More specifically, we explore the causal graph structures and design corresponding causal generative models which can be learned with the help of unlabelled data. The learned causal generative model can generate synthetic labelled data for training a more accurate predictive model. We verify the effectiveness of our proposed method by empirical studies on both simulated and real data.
title Semi-Supervised Learning under General Causal Models
topic Machine Learning
url https://arxiv.org/abs/2510.22567