Author Login Editor-in-Chief Peer Review Editor Work Office Work

Computer Engineering ›› 2021, Vol. 47 ›› Issue (1): 217-223. doi: 10.19678/j.issn.1000-3428.0056831

• Graphics and Image Processing • Previous Articles     Next Articles

A Corneal Nerve Segmentation Algorithm Based on Improved ResU-Net

HAO Huaying1,2, ZHAO Kun2,3, SU Pan2, ZHANG Hui3, ZHAO Yitian2, LIU Jiang2,4   

  1. 1. Faculty of Mechanical Engineering & Mechanics, Ningbo University, Ningbo, Zhejiang 315211, China;
    2. Cixi Institute of Biomedical Engineering, Ningbo Institute of Materials Technology & Engineering, Chinese Academy of Sciences, Ningbo, Zhejiang 315201, China;
    3. Information and Control Engineering School, Shenyang Jianzhu University, Shenyang 110168, China;
    4. Department of Computer Science and Engineering, Southern University of Science and Technology, Shenzhen, Guangdong 518055, China
  • Received:2019-12-06 Revised:2020-01-31 Published:2020-02-07

一种基于改进ResU-Net的角膜神经分割算法

郝华颖1,2, 赵昆2,3, 苏攀2, 张辉3, 赵一天2, 刘江2,4   

  1. 1. 宁波大学 机械工程与力学学院, 浙江 宁波 315211;
    2. 中国科学院宁波材料技术与工程研究所 慈溪生物医学工程研究所, 浙江 宁波 315201;
    3. 沈阳建筑大学 信息与控制工程学院, 沈阳 110168;
    4. 南方科技大学 计算机科学与工程系, 广东 深圳 518055
  • 作者简介:郝华颖(1996-),女,硕士研究生,主研方向为医学图像处理;赵昆,硕士研究生;苏攀(通信作者),博士;张辉,副教授、硕士;赵一天,副研究员、博士;刘江,教授、博士。
  • 基金资助:
    国家自然科学基金(61906181);中国博士后科学基金(2019M652156)。

Abstract: The automatic segmentation of corneal nerve images is crucial to the diagnosis and screening of several diseases such as diabetic neuropathy,but it suffers from the low segmentation efficiency caused by the low contrast of corneal nerve images and the existence of non-neural structures.To address the problem,this paper proposes a novel corneal nerve segmentation algorithm based on attention mechanism,which introduces multi-scale residual module, attention mechanism module,multi-scale image input module,and multi-layer loss function output module into the ResU-Net structure.The multi-scale residual module is used to add multi-scale representation information into the residual module to improve the multi-scale feature extraction ability of the convolutional layer.The attention mechanism module consisting of channel and spatial attentions uses the network to optimize the weight of the target features in the encoder and decoder,so as to enhance the features of the target area as well as suppress the features of background and noise area. Furthermore,the multi-scale image input and multi-layer function output modules are added to supervise the feature learning of each network layer.Experimental results show that the proposed method outperforms the existing mainstream segmentation algorithms with its Area Under Curve(AUC) reaching 0.990 and its Sencificity(Sen) reaching 0.880.

Key words: corneal nerve, multi-scale residual, attention mechanism, ResU-Net structure, Dice coefficient loss function

摘要: 角膜神经图像的自动分割对于糖尿病神经病变等疾病的诊断与筛查至关重要。针对由于角膜神经图像存在对比度低且包含非神经结构而造成分割效率较低的问题,在ResU-Net结构基础上引入多尺度残差、注意力机制、多尺度图像输入与多层损失函数输出模块,提出一种基于注意力机制的角膜神经分割算法。多尺度残差模块通过在残差模块中加入多尺度表征信息以提高卷积层提取多尺度特征的能力,而注意力机制模块在双重注意力作用下,利用网络对编码器与解码器中的目标特征进行权重优化,使得在增强图像目标区域特征的同时抑制背景及噪声区域,并采用多尺度图像输入与多层函数输出模块以监督网络中每一层的特征学习。实验结果表明,与主流分割算法相比,该算法的分割效果更优,且曲线下面积与敏感度分别可达到0.990和0.880。

关键词: 角膜神经, 多尺度残差, 注意力机制, ResU-Net结构, Dice系数损失函数

CLC Number: