用注意力引导的生成方法增强医学图像,不破坏关键诊断信息。
MedDiffuseMix: Preserving Diagnostic Evidence with Saliency-Aware Diffusion Medical Image Data Augmentation

- 基于诊断重要性区域分离,仅对非关键区域进行生成混合
- 在4个公开数据集上提升准确率与受试者工作特征曲线下面积
- 适合数据少、依赖精准诊断的医疗图像分类任务
有限的数据量、类别不平衡和域差异仍是可靠医学图像分类的主要障碍。传统增强方法虽能提升训练多样性,但可能扭曲具有诊断意义的结构;而无约束的生成增强可能引入标签不一致内容。本文提出MedDiffuseMix,一种基于显著性引导的扩散混合框架,用于可控医学图像增强。该方法利用分类器生成的显著性图,将高显著性诊断区域与低显著性背景区域分离,并主要对诊断重要性较低的区域实施扩散引导混合。自适应混合、高斯边界平滑及显著性保持约束,有效减少语义失真,并拒绝或削弱使模型关注偏离临床相关证据的样本。在四个公开基准数据集上评估:北美放射学会肺炎胸片数据集、骨科放射图像、PatchCamelyon以及乳腺癌组织病理图像分类数据集。使用卷积与Transformer类分类器的实验表明,MedDiffuseMix相比标准增强、Mixup、GenMix、SaliencyMix及基于扩散的基线方法,在准确率、F1分数与受试者工作特征曲线下面积方面均有提升。消融实验证实显著性引导、自适应区域混合与平滑边界融合的重要性。可视化归因分析进一步显示,MedDiffuseMix更有效地保留了具有诊断显著性的区域。结果表明,基于显著性的扩散混合是小样本医学图像分类的有效增强策略。
原文摘要 · Abstract (English)
Limited data availability, class imbalance, and domain variability remain major barriers to reliable medical image classification. Conventional augmentation can improve training diversity but may distort diagnostically informative structures, whereas unconstrained generative augmentation may introduce label-inconsistent content. This paper proposes MedDiffuseMix, a saliency-guided diffusion mixing framework for controlled medical image augmentation. The method uses classifier-derived saliency maps to separate high-saliency diagnostic regions from low-saliency background areas and applies diffusion-guided mixing mainly to regions with lower diagnostic importance. Adaptive mixing, Gaussian boundary blending, and a saliency-preservation constraint reduce semantic distortion and reject or attenuate samples that shift model attention away from clinically relevant evidence. The framework is evaluated on four public benchmarks: the Radiological Society of North America pneumonia chest radiography dataset, Musculoskeletal Radiographs, PatchCamelyon, and the Breast Cancer Histopathological Image Classification dataset. Experiments with convolutional and transformer-based classifiers show that MedDiffuseMix improves accuracy, F1-score, and area under the receiver operating characteristic curve compared with standard augmentation, Mixup, GenMix, SaliencyMix, and diffusion-based augmentation baselines. Ablation studies confirm the importance of saliency guidance, adaptive region mixing, and smooth boundary blending. Visual attribution analysis further indicates that MedDiffuseMix better preserves diagnostically salient regions. These results suggest that saliency-guided diffusion mixing is an effective augmentation strategy for limited-data medical image classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。