4 个月前

用于退行性视网膜眼底图像中标志点检测的带有层次瓶颈注意机制的U-Net

用于退行性视网膜眼底图像中标志点检测的带有层次瓶颈注意机制的U-Net

摘要

眼底摄影在临床实践中常用于记录年龄相关性黄斑变性(AMD)、青光眼和糖尿病视网膜病变(DR)等视网膜退行性疾病的存在和严重程度,其中黄斑中心凹和视盘(OD)是重要的视网膜标志。然而,在视网膜退行过程中,病灶、玻璃膜疣和其他视网膜异常的出现极大地复杂化了自动标志检测和分割。本文提出了一种改进的U-Net架构——HBA-U-Net,该网络通过层次瓶颈注意力机制增强了U-Net主干。该网络包含一个新颖的瓶颈注意力模块,该模块结合并优化了自注意力、通道注意力和相对位置注意力,以突出可能对退行性视网膜中黄斑中心凹和视盘分割至关重要的视网膜异常。HBA-U-Net在不同数据集和眼部条件下的黄斑中心凹检测中取得了最先进的结果(ADAM:欧氏距离(ED)25.4像素,REFUGE:32.5像素,IDRiD:32.1像素),在AMD的视盘分割中也表现优异(ADAM:Dice系数(DC)0.947),并且在DR的视盘检测中表现出色(IDRiD:欧氏距离20.5像素)。我们的研究结果表明,HBA-U-Net可能非常适合在多种视网膜退行性疾病存在的情况下进行标志检测。

代码仓库

基准测试

基准方法指标
fovea-detection-on-adamHBA-U-Net
Euclidean Distance (ED): 25.4
fovea-detection-on-idridHBA-U-Net
Euclidean Distance (ED): 32.1
fovea-detection-on-refugeHBA-U-Net
Euclidean Distance (ED): 32.5
optic-disc-detection-on-idridHBA-U-Net
Euclidean Distance (ED): 20.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于退行性视网膜眼底图像中标志点检测的带有层次瓶颈注意机制的U-Net | 论文 | HyperAI超神经