English

MPG-Net: Multi-Prediction Guided Network for Segmentation of Retinal Layers in OCT Images

Image and Video Processing 2020-09-30 v1 Computer Vision and Pattern Recognition Machine Learning

Abstract

Optical coherence tomography (OCT) is a commonly-used method of extracting high resolution retinal information. Moreover there is an increasing demand for the automated retinal layer segmentation which facilitates the retinal disease diagnosis. In this paper, we propose a novel multiprediction guided attention network (MPG-Net) for automated retinal layer segmentation in OCT images. The proposed method consists of two major steps to strengthen the discriminative power of a U-shape Fully convolutional network (FCN) for reliable automated segmentation. Firstly, the feature refinement module which adaptively re-weights the feature channels is exploited in the encoder to capture more informative features and discard information in irrelevant regions. Furthermore, we propose a multi-prediction guided attention mechanism which provides pixel-wise semantic prediction guidance to better recover the segmentation mask at each scale. This mechanism which transforms the deep supervision to supervised attention is able to guide feature aggregation with more semantic information between intermediate layers. Experiments on the publicly available Duke OCT dataset confirm the effectiveness of the proposed method as well as an improved performance over other state-of-the-art approaches.

Keywords

Cite

@article{arxiv.2009.13634,
  title  = {MPG-Net: Multi-Prediction Guided Network for Segmentation of Retinal Layers in OCT Images},
  author = {Zeyu Fu and Yang Sun and Xiangyu Zhang and Scott Stainton and Shaun Barney and Jeffry Hogg and William Innes and Satnam Dlay},
  journal= {arXiv preprint arXiv:2009.13634},
  year   = {2020}
}

Comments

EUSIPCO2020