arXiv:2505.23040cs.CV2025-05中稿 · ISBI2025被引 5

用CLIP改进医学图像分类,兼顾隐私与泛化能力。

Deep Modeling and Optimization of Medical Image Classification

  • 设计多模型融合的CLIP变体,结合CNN与ViT提升医学图像分类
  • 在HAM10000数据集上,MaxViT平均准确率达87.03%
  • 引入SVM可使Swin系列模型整体性能提升约2%

深度模型如卷积神经网络(CNN)和视觉变换器(ViT)在图像分类中表现优异,但需大量数据微调,在医疗领域因数据隐私问题难以实现。尽管对比语言图像预训练(CLIP)在自然图像领域表现良好,其在医学领域的潜力尚未充分探索。为此,本文提出三种策略:1)采用四组CNN与八组ViT作为图像编码器,构建新型CLIP变体,用于脑癌与皮肤癌分类;2)将12个深度模型与两种联邦学习技术结合,保护数据隐私;3)引入传统机器学习方法(如支持向量机,SVM),提升模型在未见域数据上的泛化能力。实验表明,在HAM10000数据集上,MaxViT通过多模态学习达到最高平均测试指标(AVG = 87.03%);在联邦学习设置下,ConvNeXt_L相较Swin_B(F1=81.33%)取得83.98%的高分。此外,使用SVM可使Swin系列模型在ISIC2018数据集上整体性能提升约2%。代码已公开于https://github.com/AIPMLab/SkinCancerSimulation。

原文摘要 · Abstract (English)

Deep models, such as convolutional neural networks (CNNs) and vision transformer (ViT), demonstrate remarkable performance in image classification. However, those deep models require large data to fine-tune, which is impractical in the medical domain due to the data privacy issue. Furthermore, despite the feasible performance of contrastive language image pre-training (CLIP) in the natural domain, the potential of CLIP has not been fully investigated in the medical field. To face these challenges, we considered three scenarios: 1) we introduce a novel CLIP variant using four CNNs and eight ViTs as image encoders for the classification of brain cancer and skin cancer, 2) we combine 12 deep models with two federated learning techniques to protect data privacy, and 3) we involve traditional machine learning (ML) methods to improve the generalization ability of those deep models in unseen domain data. The experimental results indicate that maxvit shows the highest averaged (AVG) test metrics (AVG = 87.03\%) in HAM10000 dataset with multimodal learning, while convnext\_l demonstrates remarkable test with an F1-score of 83.98\% compared to swin\_b with 81.33\% in FL model. Furthermore, the use of support vector machine (SVM) can improve the overall test metrics with AVG of $\sim 2\%$ for swin transformer series in ISIC2018. Our codes are available at https://github.com/AIPMLab/SkinCancerSimulation.

医学图像CLIP联邦学习SVM

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