arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.15526cs.CV

探索眼底图像基础模型的迁移性:在高血压视网膜病变中的应用

Exploring the Transferability of a Foundation Model for Fundus Images: Application to Hypertensive Retinopathy

ETS Montreal(蒙特利尔高等工程技术学院);DIAGNOS Inc.(DIAGNOS公司)

Julio Silva-Rodriguez, Jihed Chelbi, Waziha Kabir, Hadi Chakor, Jose Dolz, Ismail Ben Ayed, Riadh Kobbi

AI总结:

本研究针对医学图像分类数据稀缺问题,探索视网膜视觉语言基础模型FLAIR在高血压视网膜病变眼底图像分类中的迁移性,对比其与ImageNet预训练在两种迁移学习方法下的性能,证实FLAIR迁移效果更优,为眼底图像分析发展提供新方向。

摘要:

使用在ImageNet上预训练的深度学习模型是应对医学图像分类中数据稀缺问题的传统方案。然而,相关研究表明,由于领域间差异极大,该策略带来的增益可能十分有限。目前,适配领域专用基础模型的范式正被证明是一种颇具前景的替代方案,但如何开展此类知识迁移,以及它具备的优势与局限性,仍在研究之中。针对眼底图像高血压视网膜病变诊断的CGI-HRDC挑战赛,为评估近期发布的视网膜视觉语言基础模型FLAIR的迁移性提供了绝佳契机。在本研究中,我们探索了将FLAIR特征作为眼底图像分类起点的潜力,并在两种主流迁移学习方法——线性探测(Linear Probing, LP)和微调(Fine-Tuning, FP)上,将其性能与ImageNet初始化的效果进行了对比。我们的实验观察显示,无论采用哪种方法,传统策略都能带来性能提升;相比之下,FLAIR模型的直接迁移可带来2.5%的增益。当对整个网络进行微调时,性能差距可扩大至4%。在该场景下我们证实,通过对分类器进行LP初始化来避免特征退化,能够最佳地复用丰富的预训练特征。尽管使用LP的直接迁移性能仍较为有限,但我们认为,诸如FLAIR这类基础模型将推动基于深度学习的眼底图像分析发展。

英文摘要:

Using deep learning models pre-trained on Imagenet is the traditional solution for medical image classification to deal with data scarcity. Nevertheless, relevant literature supports that this strategy may offer limited gains due to the high dissimilarity between domains. Currently, the paradigm of adapting domain-specialized foundation models is proving to be a promising alternative. However, how to perform such knowledge transfer, and the benefits and limitations it presents, are under study. The CGI-HRDC challenge for Hypertensive Retinopathy diagnosis on fundus images introduces an appealing opportunity to evaluate the transferability of a recently released vision-language foundation model of the retina, FLAIR. In this work, we explore the potential of using FLAIR features as starting point for fundus image classification, and we compare its performance with regard to Imagenet initialization on two popular transfer learning methods: Linear Probing (LP) and Fine-Tuning (FP). Our empirical observations suggest that, in any case, the use of the traditional strategy provides performance gains. In contrast, direct transferability from FLAIR model allows gains of 2.5%. When fine-tuning the whole network, the performance gap increases up to 4%. In this case, we show that avoiding feature deterioration via LP initialization of the classifier allows the best re-use of the rich pre-trained features. Although direct transferability using LP still offers limited performance, we believe that foundation models such as FLAIR will drive the evolution of deep-learning-based fundus image analysis.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑