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

2021

10.19678/j.issn.1000-3428.0056831

计算机工程   影响因子和分区

1000-3428

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 multiscale 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.

attention mechanism corneal nerve Dice coefficient loss function multi-scale residual ResU-Net structure

英语

其他

2-s2.0-85118937494

Scopus

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

Hao，Huaying,Zhao，Kun,Su，Pan,et al. A Corneal Nerve Segmentation Algorithm Based on Improved ResU-Net[J]. 计算机工程,2021,47(1):217-223.

Hao，Huaying,Zhao，Kun,Su，Pan,Zhang，Hui,Zhao，Yitian,&Liu，Jiang.(2021).A Corneal Nerve Segmentation Algorithm Based on Improved ResU-Net.计算机工程,47(1),217-223.

Hao，Huaying,et al."A Corneal Nerve Segmentation Algorithm Based on Improved ResU-Net".计算机工程 47.1(2021):217-223.