针对多发性硬化症脑部影像生成,强化病灶细节并保持整体结构。
Lesion-DDPM: Lesion-Enhanced 3D Diffusion for MS MRI Synthesis

- 通过多层级解剖掩码注入与病灶加权损失,增强病灶区域生成精度。
- 在真实病灶重建误差上优于所有对比模型,病灶分割Dice达0.616。
- 适合需要高质量合成数据的医学图像生成与病灶分割研究者。
3D FLAIR MRI是多发性硬化症(MS)脑部成像的标准序列之一,但公开的MS数据集规模较小且受扫描仪、采集协议和病灶模式差异影响显著。这种稀缺性和变异性制约了神经影像机器学习模型的发展,尤其对需保留微小稀疏病灶的生成模型构成挑战。本文提出Lesion-DDPM,一种3D条件扩散框架,通过多层级解剖掩码注入与病灶加权重建损失,强调病灶体素同时保持全局脑结构。基于经筛选的MSLesSeg数据子集,我们与代表性先进GAN及扩散模型进行对比,评估图像生成指标与下游3D U-Net分割性能。实验表明,Lesion-DDPM在所有方法中实现了最低的病灶区域重建误差;在下游3D U-Net病灶分割任务中,仅使用其生成数据训练的模型在真实MRI上达到Dice分数0.616,优于最佳对比合成数据集的0.569;当将其生成图像加入真实训练集时,Dice分数进一步提升至0.685。
原文摘要 · Abstract (English)
3D FLAIR MRI is widely recommended as one of the standard MRI sequences for brain imaging in multiple sclerosis (MS), but publicly available MS datasets remain relatively small and vary across scanners, acquisition protocols, and lesion patterns. This scarcity and variability hinder the development of robust neuroimaging machine learning models and are particularly challenging for generative models that aim to synthesize images while preserving small, sparse lesions. We propose Lesion-DDPM, a 3D conditional diffusion framework for lesion-aware FLAIR synthesis that incorporates multi-level anatomical mask injection together with a lesion-weighted reconstruction loss to emphasize lesion voxels while maintaining global brain structure. Using a curated subset of the MSLesSeg dataset, we compare Lesion-DDPM with representative state-of-the-art GAN- and diffusion-based models, assessing both image-generation metrics and downstream 3D U-Net segmentation. In our experiments, Lesion-DDPM achieved the lowest lesion-region reconstruction error among all methods. In a downstream 3D U-Net lesion segmentation task, a model trained only on Lesion-DDPM-generated scans and evaluated on real MRIs reached a Dice score of 0.616 compared with 0.569 for the best competing synthetic dataset. When Lesion-DDPM images were added to the real training set, the Dice score further increased to 0.685.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。