4.6 Article

AFENet: Attention Fusion Enhancement Network for Optic Disc Segmentation of Premature Infants

Journal

FRONTIERS IN NEUROSCIENCE
Volume 16, Issue -, Pages -

Publisher

FRONTIERS MEDIA SA
DOI: 10.3389/fnins.2022.836327

Keywords

optic disc segmentation; multiscale features; attention mechanism; fundus images; premature infants

Categories

Funding

  1. National Key R&D Program of China [2018YFA0701700]
  2. National Nature Science Foundation of China [U20A20170, 61622114]

Ask authors/readers for more resources

This article proposes a novel neural network model (AFENet) for the accurate segmentation of optic disc (OD) in fundus images of premature infants. By fusing high-level semantic information and multiscale low-level detailed information from different levels, the model effectively addresses the challenges of complexity, non-uniform illumination, and low contrast between background and target area. Experimental results demonstrate that the proposed model achieves excellent performance in OD segmentation.
Retinopathy of prematurity and ischemic brain injury resulting in periventricular white matter damage are the main causes of visual impairment in premature infants. Accurate optic disc (OD) segmentation has important prognostic significance for the auxiliary diagnosis of the above two diseases of premature infants. Because of the complexity and non-uniform illumination and low contrast between background and the target area of the fundus images, the segmentation of OD for infants is challenging and rarely reported in the literature. In this article, to tackle these problems, we propose a novel attention fusion enhancement network (AFENet) for the accurate segmentation of OD in the fundus images of premature infants by fusing adjacent high-level semantic information and multiscale low-level detailed information from different levels based on encoder-decoder network. Specifically, we first design a dual-scale semantic enhancement (DsSE) module between the encoder and the decoder inspired by self-attention mechanism, which can enhance the semantic contextual information for the decoder by reconstructing skip connection. Then, to reduce the semantic gaps between the high-level and low-level features, a multiscale feature fusion (MsFF) module is developed to fuse multiple features of different levels at the top of encoder by using attention mechanism. Finally, the proposed AFENet was evaluated on the fundus images of preterm infants for OD segmentation, which shows that the proposed two modules are both promising. Based on the baseline (Res34UNet), using DsSE or MsFF module alone can increase Dice similarity coefficients by 1.51 and 1.70%, respectively, whereas the integration of the two modules together can increase 2.11%. Compared with other state-of-the-art segmentation methods, the proposed AFENet achieves a high segmentation performance.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.6
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available