3 个月前

CASS:面向医学图像分析的跨架构自监督学习

CASS:面向医学图像分析的跨架构自监督学习

摘要

深度学习与计算机视觉的最新进展已显著降低了自动化医学图像分析的技术门槛,使算法能够处理无标签图像并提升性能。然而,现有技术普遍面临极高的计算资源需求,且在减小批量大小(batch size)或训练轮次(training epochs)时性能急剧下降。本文提出一种新型自监督学习方法——跨架构自监督学习(Cross Architectural Self-Supervision, CASS),该方法同时利用Transformer与卷积神经网络(CNN)的互补优势。实验结果表明,在四个不同数据集上,相较于当前最先进的自监督学习方法,CASS训练的CNN与Transformer模型在仅使用1%标注数据时平均提升3.8%,使用10%标注数据时提升5.9%,在使用100%标注数据时提升达10.13%,同时训练时间减少69%。此外,CASS对批量大小和训练轮次的变化表现出更强的鲁棒性。值得注意的是,其中一个测试数据集包含自身免疫性疾病患者的组织病理学切片,该疾病因数据稀缺而在医学影像研究中长期被低估。相关代码已开源,可于GitHub获取。

代码仓库

pranavsinghps1/CASS
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
classification-on-autoimmune-datasetCASS
F1 score: 0.8894
classification-on-autoimmune-datasetDINO
F1 score: 0.8639
classification-on-brain-tumor-mri-datasetDINO
F1 score: 0.9909
classification-on-isic-2019CASS
Balanced Multi-Class Accuracy: 0.6519
partial-label-learning-on-autoimmune-datasetDINO
F1 score: 0.8445
partial-label-learning-on-autoimmune-datasetCASS
F1 score: 0.8717
partial-label-learning-on-isic-2019CASS
Balanced Multi-Class Accuracy: 0.7258

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CASS:面向医学图像分析的跨架构自监督学习 | 论文 | HyperAI超神经