轻量微调医学模型,实现可解释的皮肤病变精准分割
PEFT-MedSAM: Efficient Fine-Tuning of Medical Foundation Models for Explainable Skin Lesion Segmentation
- 仅训练轻量掩码解码器,冻结预训练编码器保持高效
- 在ISIC 2018上达0.9411 Dice,优于全量训练U-Net和零样本MedSAM
- 结合Grad-CAM与指认游戏验证可解释性,临床可信度高
基于深度学习的皮肤病变自动分割对早期发现黑色素瘤具有重要意义,但现有方法性能普遍不佳。本文提出参数高效微调方法PEFT-MedSAM,用于适配医学版Segment Anything模型(MedSAM)以实现皮肤病变分割。该方法仅训练轻量级掩码解码器,冻结预训练图像编码器与提示编码器。在ISIC 2018基准数据集上的实验显示,其Dice系数达0.9411,交并比为0.8918,显著优于全量训练的U-Net(Dice 0.8715)与零样本MedSAM(Dice 0.8997)。在外部验证数据集PH2上,获得0.9467的Dice系数(±0.0310标准差)。通过Wilcoxon符号秩检验(p < 0.0001)与95%置信区间[0.9364, 0.9447]验证结果稳健。为提升临床可信度,采用Grad-CAM可解释性分析及指认游戏评估方法,在519张图像的验证集上达到98.27%准确率,确认模型能正确识别病变区域。
原文摘要 · Abstract (English)
Automated segmentation of skin lesions using deep learning models for dermoscopic images can be very helpful in finding melanomas earlier than they would normally be detected. However, most deep learning methods available do not perform well. The aim of this paper is to present a parameter-efficient fine-tuning method called PEFT-MedSAM for adapting the Medical Segment Anything Model (MedSAM) to automatically segment dermoscopic skin lesions. The PEFT-MedSAM method uses only the lightweight mask decoder for training the model while keeping the pre-trained image encoder and prompt encoder frozen. The experiments performed on the ISIC 2018 benchmark dataset shows that PEFT-MedSAM obtains a dice coefficient of .9411 and an intersection over union value of .8918 when compared to both a fully trained U-Net baseline (.8715 dice coefficient) and zero-shot MedSAM inference (.8997 dice coefficient). The external validation of the model using PH2 dataset shows .9467 dice coefficient with +/- .0310 standard deviation. Supportive evidence for these claims include a p-value less than .0001 for Wilcoxon signed rank tests comparing the two datasets and bootstrap-estimated 95% confidence intervals of [.9364,.9447] that represent the estimated range of possible values for the average dice coefficient obtained by repeating the test. To increase clinical trustworthiness, we used Grad-CAM explainability along with a pointing game based evaluation methodology to evaluate the CNN baseline model on the validation set. The results showed that we had an accuracy rate of 98.27% on the validation set of 519 images and confirmed that the model classified regions containing skin lesions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。