Compositional World Knowledge leads to High Utility Synthetic data

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gaudi, Sachit, Sreekumar, Gautam, Boddeti, Vishnu
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911006946492416
author Gaudi, Sachit
Sreekumar, Gautam
Boddeti, Vishnu
author_facet Gaudi, Sachit
Sreekumar, Gautam
Boddeti, Vishnu
contents Machine learning systems struggle with robustness, under subpopulation shifts. This problem becomes especially pronounced in scenarios where only a subset of attribute combinations is observed during training -a severe form of subpopulation shift, referred as compositional shift. To address this problem, we ask the following question: Can we improve the robustness by training on synthetic data, spanning all possible attribute combinations? We first show that training of conditional diffusion models on limited data lead to incorrect underlying distribution. Therefore, synthetic data sampled from such models will result in unfaithful samples and does not lead to improve performance of downstream machine learning systems. To address this problem, we propose CoInD to reflect the compositional nature of the world by enforcing conditional independence through minimizing Fisher's divergence between joint and marginal distributions. We demonstrate that synthetic data generated by CoInD is faithful and this translates to state-of-the-art worst-group accuracy on compositional shift tasks on CelebA.
format Preprint
id arxiv_https___arxiv_org_abs_2503_04687
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Compositional World Knowledge leads to High Utility Synthetic data
Gaudi, Sachit
Sreekumar, Gautam
Boddeti, Vishnu
Machine Learning
Machine learning systems struggle with robustness, under subpopulation shifts. This problem becomes especially pronounced in scenarios where only a subset of attribute combinations is observed during training -a severe form of subpopulation shift, referred as compositional shift. To address this problem, we ask the following question: Can we improve the robustness by training on synthetic data, spanning all possible attribute combinations? We first show that training of conditional diffusion models on limited data lead to incorrect underlying distribution. Therefore, synthetic data sampled from such models will result in unfaithful samples and does not lead to improve performance of downstream machine learning systems. To address this problem, we propose CoInD to reflect the compositional nature of the world by enforcing conditional independence through minimizing Fisher's divergence between joint and marginal distributions. We demonstrate that synthetic data generated by CoInD is faithful and this translates to state-of-the-art worst-group accuracy on compositional shift tasks on CelebA.
title Compositional World Knowledge leads to High Utility Synthetic data
topic Machine Learning
url https://arxiv.org/abs/2503.04687