Partial Transportability for Domain Generalization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Jalaldoust, Kasra, Bellot, Alexis, Bareinboim, Elias
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913766826835968
author Jalaldoust, Kasra
Bellot, Alexis
Bareinboim, Elias
author_facet Jalaldoust, Kasra
Bellot, Alexis
Bareinboim, Elias
contents A fundamental task in AI is providing performance guarantees for predictions made in unseen domains. In practice, there can be substantial uncertainty about the distribution of new data, and corresponding variability in the performance of existing predictors. Building on the theory of partial identification and transportability, this paper introduces new results for bounding the value of a functional of the target distribution, such as the generalization error of a classifier, given data from source domains and assumptions about the data generating mechanisms, encoded in causal diagrams. Our contribution is to provide the first general estimation technique for transportability problems, adapting existing parameterization schemes such Neural Causal Models to encode the structural constraints necessary for cross-population inference. We demonstrate the expressiveness and consistency of this procedure and further propose a gradient-based optimization scheme for making scalable inferences in practice. Our results are corroborated with experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2503_23605
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Partial Transportability for Domain Generalization
Jalaldoust, Kasra
Bellot, Alexis
Bareinboim, Elias
Machine Learning
Artificial Intelligence
A fundamental task in AI is providing performance guarantees for predictions made in unseen domains. In practice, there can be substantial uncertainty about the distribution of new data, and corresponding variability in the performance of existing predictors. Building on the theory of partial identification and transportability, this paper introduces new results for bounding the value of a functional of the target distribution, such as the generalization error of a classifier, given data from source domains and assumptions about the data generating mechanisms, encoded in causal diagrams. Our contribution is to provide the first general estimation technique for transportability problems, adapting existing parameterization schemes such Neural Causal Models to encode the structural constraints necessary for cross-population inference. We demonstrate the expressiveness and consistency of this procedure and further propose a gradient-based optimization scheme for making scalable inferences in practice. Our results are corroborated with experiments.
title Partial Transportability for Domain Generalization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2503.23605