Robust total retina thickness segmentation in optical coherence tomography images using convolutional neural networks

Abstract

We developed a fully automated system using a convolutional neural network (CNN) for total retina segmentation in optical coherence tomography (OCT) that is robust to the presence of severe retinal pathology. A generalized U-net network architecture was introduced to include the large context needed to account for large retinal changes. The proposed algorithm outperformed qualitative and quantitatively two available algorithms. The algorithm accurately estimated macular thickness with an error of 14.0 +/- 22.1 microm, substantially lower than the error obtained using the other algorithms (42.9 +/- 116.0 microm and 27.1 +/- 69.3 microm, respectively). These results highlighted the proposed algorithm's capability of modeling the wide variability in retinal appearance and obtained a robust and reliable retina segmentation even in severe pathological cases.