Abstract
We provide a framework for solving inverse problems with diffusion modelslearned from linearly corrupted data. Our method, Ambient Diffusion PosteriorSampling (A-DPS), leverages a generative model pre-trained on one type ofcorruption (e.g. image inpainting) to perform posterior sampling conditioned onmeasurements from a potentially different forward process (e.g. imageblurring). We test the efficacy of our approach on standard natural imagedatasets (CelebA, FFHQ, and AFHQ) and we show that A-DPS can sometimesoutperform models trained on clean data for several image restoration tasks inboth speed and performance. We further extend the Ambient Diffusion frameworkto train MRI models with access only to Fourier subsampled multi-coil MRImeasurements at various acceleration factors (R=2, 4, 6, 8). We again observethat models trained on highly subsampled data are better priors for solvinginverse problems in the high acceleration regime than models trained on fullysampled data. We open-source our code and the trained Ambient Diffusion MRImodels: https://github.com/utcsilab/ambient-diffusion-mri .