Multi-GlaucNet: A multi-task model for optic disc segmentation, blood vessel segmentation and glaucoma detection

被引:0
作者
Xiong, Haoren [1 ]
Long, Fei [1 ]
Alam, Mohammad S. [2 ]
Sang, Jun [1 ]
机构
[1] Chongqing Univ, Sch Big Data & Software Engn, Chongqing 401331, Peoples R China
[2] Minnesota State Univ, Coll Sci Engn & Technol, Mankato, MN 56001 USA
关键词
Deep learning; Multi-task; Glaucoma; Detection; Segmentation; NETWORKS; IMAGES;
D O I
10.1016/j.bspc.2024.106850
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Glaucoma is a common and severe ocular disease that often leads to vision loss. The information on the optic disc (OD) and blood vessels in fundus images can significantly aid in glaucoma detection. In addition, the use of deep learning models for glaucoma detection is a highly effective approach. We propose a multi-task deep learning model called Multi-GlaucNet that can simultaneously segment the OD and blood vessels, thereby assisting doctors in diagnosing glaucoma. Multi-GlaucNet consists of three modules: the OD segmentation module, blood vessel segmentation module and glaucoma detection module. The OD segmentation module and blood vessel segmentation module both adopt an encoder-decoder structure to segment OD and blood vessels, respectively. The segmentation module is constructed using bottleneck layers in the encoding process and uses Pixel Shuffle and channel attention mechanisms in the decoding process. The detection module uses the ResNet50 network to perform glaucoma detection based on the features extracted from the segmentation modules. Multi-GlaucNet demonstrates outstanding performance in three areas. It achieves a Dice coefficient of 96.7% for OD segmentation on the ORIGA dataset, accuracy of 0.9798 and F1 score of 0.8562 for blood vessel segmentation on a mixed dataset. For glaucoma detection on the REFUGE dataset, it attains the highest accuracy of 0.967 and an area under the curve (AUC) of 0.950. These results validate the effectiveness of Multi-GlaucNet for glaucoma detection. The model's ability to perform multiple tasks with high accuracy and efficiency demonstrates that it is a valuable aiding tool for glaucoma diagnosis.
引用
收藏
页数:11
相关论文
共 44 条
[1]   Optic Disc and Optic Cup Segmentation Methodologies for Glaucoma Image Detection: A Survey [J].
Almazroa, Ahmed ;
Burman, Ritambhar ;
Raahemifar, Kaamran ;
Lakshminarayanan, Vasudevan .
JOURNAL OF OPHTHALMOLOGY, 2015, 2015
[2]   Robust Vessel Segmentation in Fundus Images [J].
Budai, A. ;
Bock, R. ;
Maier, A. ;
Hornegger, J. ;
Michelson, G. .
INTERNATIONAL JOURNAL OF BIOMEDICAL IMAGING, 2013, 2013 (2013)
[3]   CNNs for automatic glaucoma assessment using fundus images: an extensive validation [J].
Diaz-Pinto, Andres ;
Morales, Sandra ;
Naranjo, Valery ;
Koehler, Thomas ;
Mossi, Jose M. ;
Navea, Amparo .
BIOMEDICAL ENGINEERING ONLINE, 2019, 18 (1)
[4]   An Ensemble Classification-Based Approach Applied to Retinal Blood Vessel Segmentation [J].
Fraz, Muhammad Moazam ;
Remagnino, Paolo ;
Hoppe, Andreas ;
Uyyanonvara, Bunyarit ;
Rudnicka, Alicja R. ;
Owen, Christopher G. ;
Barman, Sarah A. .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2012, 59 (09) :2538-2548
[5]   Joint Optic Disc and Cup Segmentation Based on Multi-Label Deep Network and Polar Transformation [J].
Fu, Huazhu ;
Cheng, Jun ;
Xu, Yanwu ;
Wong, Damon Wing Kee ;
Liu, Jiang ;
Cao, Xiaochun .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2018, 37 (07) :1597-1605
[6]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[7]   End-to-end multi-task learning for simultaneous optic disc and cup segmentation and glaucoma classification in eye fundus images [J].
Hervella, Alvaro S. ;
Rouco, Jose ;
Novo, Jorge ;
Ortega, Marcos .
APPLIED SOFT COMPUTING, 2022, 116
[8]   Locating blood vessels in retinal images by piecewise threshold probing of a matched filter response [J].
Hoover, A ;
Kouznetsova, V ;
Goldbaum, M .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2000, 19 (03) :203-210
[9]   A Simplified Deep Network Architecture on Optic Cup and Disc Segmentation [J].
Huang, Guan-Ru ;
Hsiang, Tien-Ruey .
2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
[10]  
Islam Md Tariqul, 2019, 2019 IEEE International Conference on Signal Processing, Information, Communication & Systems (SPICSCON), P59, DOI 10.1109/SPICSCON48833.2019.9065162