Automated analysis of magnetic resonance images, e.g., by measuring volumetric changes, is essential for monitoring diseases impacting the brain's structural integrity. However, many existing evaluation tools are designed primarily for healthy tissue analysis. To facilitate the assessment of scans with pathological tissue, it is necessary to restore healthy tissue in affected areas. In this study, we explore and extend denoising diffusion probabilistic models (DDPMs) for the inpainting of 3D consistent healthy brain tissue.
We enhance state-of-the-art 2D, pseudo-3D, and 3D denoising diffusion probabilistic models (DDPMs) that operate in image space, as well as 3D latent and 3D wavelet DDPMs, allowing them to synthesize healthy brain tissue. Our evaluation reveals that the pseudo-3D model achieves the best performance in terms of structural similarity index, peak signal-to-noise ratio, and mean squared error. To highlight its clinical significance, we fine-tune this model on synthetic multiple sclerosis lesions and assess its performance on a downstream brain tissue segmentation task, where it surpasses the established FMRIB Software Library (FSL) lesion-filling method. More details can be found in [1].