stat.MLJun 20, 2026

Variance-Tilted Diffusion Models for Diverse Sampling

Authors: Iskander AzangulovLeo ZhangKianoosh Ashouritaklimi

Abstract

Diffusion models are typically sampled independently, even when the downstream objective is to obtain a diverse set of candidates. We introduce a variance-weighted batch distribution that favours collections of samples with large empirical spread after a prescribed linear feature map. The target is specified explicitly, and the sampler is derived as the corresponding Doob hh-transform of independent diffusion dynamics. The resulting correction has a compact form: an interaction term that repels posterior denoised means, together with a curvature term that moves particles to the region of higher feature variance. This yields an interacting-particle sampler with a transparent probabilistic target rather than a heuristic repulsive drift.

Explore similar work

CardsList
  1. Control Variate Score Matching for Diffusion Models

    Dec 23, 2025Khaled Kahouli, Romuald Elie, Klaus-Robert Müller +3Score-Based Diffusion ModelEstimators