arXiv:2504.18983cs.CV2025-04被引 14

提出医学影像统一增强框架,验证多种混合增强策略效果

MediAug: Exploring Visual Augmentation in Medical Imaging

  • 构建六种混合增强方法的统一评估框架,适配卷积与视觉变压器
  • 在脑肿瘤和眼病数据集上,不同模型最优增强方法各不相同
  • 提供可复现基准,适合医疗图像增强研究者参考

数据增强对有限数据下的医学影像分类、病灶检测和器官分割至关重要。然而,自然图像与医学影像间存在显著领域差异,可能扭曲关键疾病特征;同时,现有增强研究多局限于单一任务或架构,先进混合策略的效果尚不明确。为此,我们提出MediAug,一个集成六种混合增强方法的统一评估框架,结合卷积网络(ResNet-50)与视觉变压器(ViT-B),在脑肿瘤MRI和眼病眼底图像数据集上进行系统评估。实验表明:在脑肿瘤分类中,ResNet-50使用MixUp达79.19%准确率,ViT-B使用SnapMix达99.44%;在眼病分类中,ResNet-50使用YOCO达91.60%,ViT-B使用CutMix达97.94%。代码将开源。

原文摘要 · Abstract (English)

Data augmentation is essential in medical imaging for improving classification accuracy, lesion detection, and organ segmentation under limited data conditions. However, two significant challenges remain. First, a pronounced domain gap between natural photographs and medical images can distort critical disease features. Second, augmentation studies in medical imaging are fragmented and limited to single tasks or architectures, leaving the benefits of advanced mix-based strategies unclear. To address these challenges, we propose a unified evaluation framework with six mix-based augmentation methods integrated with both convolutional and transformer backbones on brain tumour MRI and eye disease fundus datasets. Our contributions are threefold. (1) We introduce MediAug, a comprehensive and reproducible benchmark for advanced data augmentation in medical imaging. (2) We systematically evaluate MixUp, YOCO, CropMix, CutMix, AugMix, and SnapMix with ResNet-50 and ViT-B backbones. (3) We demonstrate through extensive experiments that MixUp yields the greatest improvement on the brain tumor classification task for ResNet-50 with 79.19% accuracy and SnapMix yields the greatest improvement for ViT-B with 99.44% accuracy, and that YOCO yields the greatest improvement on the eye disease classification task for ResNet-50 with 91.60% accuracy and CutMix yields the greatest improvement for ViT-B with 97.94% accuracy. Code will be available at https://github.com/AIGeeksGroup/MediAug.

医学影像数据增强视觉变压器混合增强

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