用轻量CNN做自监督预训练,提升眼底图疾病识别效果
A BERT-Style Self-Supervised Learning CNN for Disease Identification from Retinal Images
- 基于MobileNet设计BERT式自监督学习框架
- 在英国生物银行数据上预训练后显著提升多种疾病识别性能
- 适合标签稀缺的医学图像分析场景
深度学习在医学影像分析中取得突破,但依赖大量标注数据。获取高质量标注成本高,难以实现。视觉变换器(ViT)虽能利用海量无标签数据缓解此问题,但计算量大且缺乏局部特征捕捉能力。本研究采用轻量级CNN框架nn-MobileNet,构建类BERT的自监督学习方法,在英国生物银行的无标签眼底图像上进行预训练,以提升下游任务表现。实验验证了该模型在阿尔茨海默病(AD)、帕金森病(PD)及多种视网膜疾病识别中的有效性。结果表明,该方法可显著改善下游任务性能。研究证明,在标注数据稀缺的情况下,结合CNN结构与先进自监督学习策略,能有效提升医学图像分析的泛化能力。
原文摘要 · Abstract (English)
In the field of medical imaging, the advent of deep learning, especially the application of convolutional neural networks (CNNs) has revolutionized the analysis and interpretation of medical images. Nevertheless, deep learning methods usually rely on large amounts of labeled data. In medical imaging research, the acquisition of high-quality labels is both expensive and difficult. The introduction of Vision Transformers (ViT) and self-supervised learning provides a pre-training strategy that utilizes abundant unlabeled data, effectively alleviating the label acquisition challenge while broadening the breadth of data utilization. However, ViT's high computational density and substantial demand for computing power, coupled with the lack of localization characteristics of its operations on image patches, limit its efficiency and applicability in many application scenarios. In this study, we employ nn-MobileNet, a lightweight CNN framework, to implement a BERT-style self-supervised learning approach. We pre-train the network on the unlabeled retinal fundus images from the UK Biobank to improve downstream application performance. We validate the results of the pre-trained model on Alzheimer's disease (AD), Parkinson's disease (PD), and various retinal diseases identification. The results show that our approach can significantly improve performance in the downstream tasks. In summary, this study combines the benefits of CNNs with the capabilities of advanced self-supervised learning in handling large-scale unlabeled data, demonstrating the potential of CNNs in the presence of label scarcity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。