arXiv:2601.09044eess.IVcs.CV2026-01被引 1

用小波扩散模型生成3D脑部MRI,保留真实病灶同时补全周围组织。

POWDR: Pathology-preserving Outpainting with Wavelet Diffusion for 3D MRI

  • 以真实病灶为条件,生成解剖合理的新组织区域。
  • 合成数据使肿瘤分割Dice提升至0.7137,多样性显著提高。
  • 适合医疗影像数据不足时的增强,尤其对脑/膝关节有效。

医学影像数据常面临类别不平衡和病理样本稀缺问题,制约了分割、分类及视觉-语言任务的模型性能。为此,我们提出基于条件小波扩散模型的病理保留外扩框架POWDR,用于3D MRI。不同于传统增强或无条件生成,POWDR保留真实病理区域的同时生成解剖合理的周围组织,实现多样性提升而不虚构病灶。方法通过小波域条件控制增强高频细节,缓解潜在扩散模型的模糊问题。引入随机连通掩码训练策略,克服条件坍缩,提升病灶外区域多样性。在BraTS脑部MRI数据集上评估,并扩展至膝关节MRI,验证其组织无关性。定量指标(FID、SSIM、LPIPS)显示图像真实性,多样性分析表明随机掩码训练使余弦相似度从0.9947降至0.9580,KL散度从0.00026升至0.01494。临床评估显示,添加50例合成数据后,nnU-Net的肿瘤分割Dice从0.6992提升至0.7137。组织体积分析表明,合成图像与真实图像在脑脊液和灰质体积上无显著差异。结果证明POWDR是应对医疗影像数据稀缺与类别不平衡的有效方案,且可拓展至多种解剖结构,提供可控的病理保留合成数据生成框架。

原文摘要 · Abstract (English)

Medical imaging datasets often suffer from class imbalance and limited availability of pathology-rich cases, which constrains the performance of machine learning models for segmentation, classification, and vision-language tasks. To address this challenge, we propose POWDR, a pathology-preserving outpainting framework for 3D MRI based on a conditioned wavelet diffusion model. Unlike conventional augmentation or unconditional synthesis, POWDR retains real pathological regions while generating anatomically plausible surrounding tissue, enabling diversity without fabricating lesions. Our approach leverages wavelet-domain conditioning to enhance high-frequency detail and mitigate blurring common in latent diffusion models. We introduce a random connected mask training strategy to overcome conditioning-induced collapse and improve diversity outside the lesion. POWDR is evaluated on brain MRI using BraTS datasets and extended to knee MRI to demonstrate tissue-agnostic applicability. Quantitative metrics (FID, SSIM, LPIPS) confirm image realism, while diversity analysis shows significant improvement with random-mask training (cosine similarity reduced from 0.9947 to 0.9580; KL divergence increased from 0.00026 to 0.01494). Clinically relevant assessments reveal gains in tumor segmentation performance using nnU-Net, with Dice scores improving from 0.6992 to 0.7137 when adding 50 synthetic cases. Tissue volume analysis indicates no significant differences for CSF and GM compared to real images. These findings highlight POWDR as a practical solution for addressing data scarcity and class imbalance in medical imaging. The method is extensible to multiple anatomies and offers a controllable framework for generating diverse, pathology-preserving synthetic data to support robust model development.

3D MRI数据增强小波扩散病理保留

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。