Semi-Supervised Learning for Deep Causal Generative Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ibrahim, Yasin, Warr, Hermione, Kamnitsas, Konstantinos
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909252048650240
author Ibrahim, Yasin
Warr, Hermione
Kamnitsas, Konstantinos
author_facet Ibrahim, Yasin
Warr, Hermione
Kamnitsas, Konstantinos
contents Developing models that are capable of answering questions of the form "How would x change if y had been z?'" is fundamental to advancing medical image analysis. Training causal generative models that address such counterfactual questions, though, currently requires that all relevant variables have been observed and that the corresponding labels are available in the training data. However, clinical data may not have complete records for all patients and state of the art causal generative models are unable to take full advantage of this. We thus develop, for the first time, a semi-supervised deep causal generative model that exploits the causal relationships between variables to maximise the use of all available data. We explore this in the setting where each sample is either fully labelled or fully unlabelled, as well as the more clinically realistic case of having different labels missing for each sample. We leverage techniques from causal inference to infer missing values and subsequently generate realistic counterfactuals, even for samples with incomplete labels.
format Preprint
id arxiv_https___arxiv_org_abs_2403_18717
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Semi-Supervised Learning for Deep Causal Generative Models
Ibrahim, Yasin
Warr, Hermione
Kamnitsas, Konstantinos
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Developing models that are capable of answering questions of the form "How would x change if y had been z?'" is fundamental to advancing medical image analysis. Training causal generative models that address such counterfactual questions, though, currently requires that all relevant variables have been observed and that the corresponding labels are available in the training data. However, clinical data may not have complete records for all patients and state of the art causal generative models are unable to take full advantage of this. We thus develop, for the first time, a semi-supervised deep causal generative model that exploits the causal relationships between variables to maximise the use of all available data. We explore this in the setting where each sample is either fully labelled or fully unlabelled, as well as the more clinically realistic case of having different labels missing for each sample. We leverage techniques from causal inference to infer missing values and subsequently generate realistic counterfactuals, even for samples with incomplete labels.
title Semi-Supervised Learning for Deep Causal Generative Models
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.18717