Automated delineation of corneal layers on OCT images using a boundary-guided CNN

Lei Wang,Meixiao Shen,Qian Chang,Ce Shi,Yang Chen,Yuheng Zhou,Yanchun Zhang,Jiantao Pu,Hao Chen

DOI: https://doi.org/10.1016/j.patcog.2021.108158

IF: 8

2021-12-01

Pattern Recognition

Abstract:<p>Accurate segmentation of corneal layers depicted on optical coherence tomography (OCT) images is very helpful for quantitatively assessing and diagnosing corneal diseases (<em>e.g.</em>, keratoconus and dry eye). In this study, we presented a novel boundary-guided convolutional neural network (CNN) architecture (BG-CNN) to simultaneously extract different corneal layers and delineate their boundaries. The developed BG-CNN architecture used three convolutional blocks to construct two network modules on the basis of the classical U-Net network. We trained and validated the network on a dataset consisting of 1,712 OCT images acquired on 121 subjects using a 10-fold cross-validation method. Our experiments showed an average dice similarity coefficient (DSC) of 0.9691, an intersection over union (IOU) of 0.9411, and a Hausdorff distance (HD) of 7.4423 pixels. Compared with several other classical networks, namely U-Net, Attention U-Net, Asymmetric U-Net, BiO-Net, CE-Net, CPFnte, M-Net, and Deeplabv3, on the same dataset, the developed network demonstrated a promising performance, suggesting its unique strength in segmenting corneal layers depicted on OCT images.</p>

computer science, artificial intelligence,engineering, electrical & electronic