Dual-Stage U-Shape Convolutional Network for Esophageal Tissue Segmentation in OCT Images

Abstract

Automatic segmentation is the crucial step for esophageal optical coherence tomography (OCT) image processing, which is able to highlight diagnosis-related tissue layers and provide characteristics such as shape and thickness for esophageal disease diagnosis. This study proposes a dual-stage framework using a specifically designed encoder-decoder network configuration for accurate and reliable esophageal layer segmentation, which is named as the dual-stage U-shape convolutional network (D-UCN). The proposed approach utilized one UCN to locate the target tissue region, which is followed by another UCN with similar architecture to achieve the final segmentation. In this way, the proposed strategy effectively solves the problems encountered in our previous studies, such as disturbance from neighboring diagnostically unrelated tissues, probe protection sheaths from the imaging equipment and the inevitable speckle noise. Experimental results on esophageal OCT B-scans from C57BL mice demonstrated that the proposed dual-stage framework achieved performance comparable to manual segmentation. The effectiveness and advantages of the dual-stage strategy are also confirmed in comparison with graph theory dynamic program (GTDP) and U-Net.