Lesion-DDPM: Lesion-Enhanced 3D Diffusion for MS MRI Synthesis
Lesion-DDPM is a 3D conditional diffusion model designed to synthesize high-fidelity FLAIR MRI images for multiple sclerosis (MS) by emphasizing small lesions. It uses multi-level anatomical mask injection and a lesion-weighted loss function to preserve subtle MS lesions while maintaining overall brain structure, outperforming existing generative models in both image quality and downstream segmentation tasks.