4 个月前

利用联合注意力网络提高光学相干断层扫描图像中视网膜退行性病变检测的鲁棒性

利用联合注意力网络提高光学相干断层扫描图像中视网膜退行性病变检测的鲁棒性

摘要

噪声数据以及不同眼科疾病导致的眼部外观相似性给自动化专家系统在准确检测视网膜疾病方面带来了重大挑战。此外,知识迁移性的缺乏和对不合理大数据集的需求限制了当前机器学习系统的临床应用。为了提高系统的鲁棒性,需要更好地理解视网膜子空间变形如何导致不同程度的疾病严重性,并利用这些理解来优先考虑特定疾病的模型细节。本文提出了一种基于特定疾病特征表示的新架构,该架构由两个联合网络组成——一个用于监督编码疾病模型,另一个以无监督的方式生成注意力图以保留特定疾病的时空信息。我们在公开数据集上的实验结果表明,所提出的联合网络显著提高了最先进的视网膜疾病分类网络在未见数据集上的准确性和鲁棒性。

代码仓库

SharifAmit/Robust_Joint_Attention
官方
tf
GitHub 中提及

基准测试

基准方法指标
retinal-oct-disease-classification-onJoint-Attention-Network OpticNet-71
Acc: 99.68
retinal-oct-disease-classification-onJoint-Attention-Network ResNet50-v1
Acc: 100
retinal-oct-disease-classification-onJoint-Attention-Network MobileNet-v2
Acc: 99.36
retinal-oct-disease-classification-on-oct2017Joint-Attention-Network OpticNet-71
Acc: 77.4
retinal-oct-disease-classification-on-oct2017Joint-Attention-Network MobileNet-v2
Acc: 95.6
retinal-oct-disease-classification-on-oct2017Joint-Attention-Network ResNet50-v1
Acc: 92.4

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
利用联合注意力网络提高光学相干断层扫描图像中视网膜退行性病变检测的鲁棒性 | 论文 | HyperAI超神经