Search papers, labs, and topics across Lattice.
This paper introduces a two-stage framework for missing data imputation using latent diffusion models. A VAE-based imputer first learns compact features from incomplete data, and then a diffusion model is trained in the latent space. Results show that latent diffusion maintains higher sample quality and stability compared to pixel-space diffusion, especially with high missingness rates (up to 50\%).
Latent diffusion models can handle missing data far more robustly than pixel-space diffusion, maintaining high sample quality even with 50% data missing.
Diffusion models have emerged as powerful generative approaches for missing-data imputation, yet most existing methods operate directly in data space and degrade when training data are heavily incomplete. We investigate whether shifting diffusion to a learned latent representation improves robustness under missing-completely-at-random (MCAR) corruption. To this end, we propose a two-stage framework: a robust VAE-based imputer first learns compact semantic features from incomplete observations, and a diffusion model is then trained in the resulting latent space. Across training missing rates, we perform a controlled comparison against pixel-space diffusion models under the same incomplete-data setting. The latent diffusion model maintains high sample quality and remains stable up to 50\% missingness, while pixel-space diffusion degrades progressively as missingness increases. For downstream imputation, latent diffusion also achieves consistently better performance than pixel-space diffusion. These findings indicate that latent-space modeling mitigates artifact amplification from zero-imputed inputs and provides a more robust generative prior for incomplete-data learning. Overall, our results support latent diffusion as a strong and practically useful alternative to pixel-space diffusion for missing-data problems.
