A Sampling-Based Domain Generalization Study with Diffusion Generative Models

NeurIPS 2025 Workshop FPI Submission9 Authors

Track: Main Track

Keywords: Diffusion model, domain generalization, latent sampling

TL;DR: We reformulate the domain generalization problem from learning-the-mapping to sampling-from-latent, leveraging the latent properties of out-of-domain (OOD) samples within diffusion models to synthesize new OOD data without fine-tuning.

Abstract: In this work, we investigate the domain generalization capabilities of diffusion models in the context of synthesizing images that are distinct from the training data. Instead of fine-tuning, we tackle this challenge from a sampling-based perspective using frozen, pre-trained diffusion models. Specifically, we demonstrate that arbitrary out-of-domain (OOD) images establish Gaussian priors in the latent spaces of a given model after inversion, and that these priors are separable from those of the original training domain. This OOD latent property allows us to synthesize new images of the target unseen domain by discovering qualified OOD latent encodings in the inverted noisy spaces, without altering the pre-trained models. Our cross-model and cross-domain experiments show that the proposed sampling-based method can expand the latent space and generate unseen images without impairing the generation quality of the original domain. We also showcase a practical application of our approach using astrophysical data, highlighting the potential of this generalization paradigm in data-sparse fields such as scientific exploration.

Submission Number: 9