Optimized-Unet: Novel Algorithm for Parapapillary Atrophy Segmentation

Frontiers in Neuroscience

Han Li¹  Cheng Wan¹  Yanwu Xu²  Jiasheng Wu³  Chenghu Wang⁴  Guofan Cao⁴  Zhipeng Yan⁴  Weihua Yang⁴  Qin Jiang⁴ 

[1] College of Electronic and Information Engineering/College of Integrated Circuits, Nanjing University of Aeronautics and Astronautics, Nanjing, China;Intelligent Healthcare Unit, Baidu, Beijing, China;School of Electronic Information and Communications, Huazhong University of Science and Technology, Wuhan, China;The Affiliated Eye Hospital of Nanjing Medical University, Nanjing, China;

关键词: medical image segmentation;    high myopia;    parapapillary atrophy;    convolutional neural network;    fundus image;   

DOI  :  10.3389/fnins.2021.758887

来源: DOAJ

【 摘 要 】

In recent years, an increasing number of people have myopia in China, especially the younger generation. Common myopia may develop into high myopia. High myopia causes visual impairment and blindness. Parapapillary atrophy (PPA) is a typical retinal pathology related to high myopia, which is also a basic clue for diagnosing high myopia. Therefore, accurate segmentation of the PPA is essential for high myopia diagnosis and treatment. In this study, we propose an optimized Unet (OT-Unet) to solve this important task. OT-Unet uses one of the pre-trained models: Visual Geometry Group (VGG), ResNet, and Res2Net, as a backbone and is combined with edge attention, parallel partial decoder, and reverse attention modules to improve the segmentation accuracy. In general, using the pre-trained models can improve the accuracy with fewer samples. The edge attention module extracts contour information, the parallel partial decoder module combines the multi-scale features, and the reverse attention module integrates high- and low-level features. We also propose an augmented loss function to increase the weight of complex pixels to enable the network to segment more complex lesion areas. Based on a dataset containing 360 images (Including 26 pictures provided by PALM), the proposed OT-Unet achieves a high AUC (Area Under Curve) of 0.9235, indicating a significant improvement over the original Unet (0.7917).

【 授权许可】

Unknown