机构:[1]School of Biological Science and Medical Engineering, Beihang University, Beijing, China[2]Hefei Innovation Research Institute, Beihang University, Hefei, China[3]Beijing Advanced Innovation Centre for Biomedical Engineering, Beihang University, Beijing, China[4]Beijing Advanced Innovation Centre for Big Data-Based Precision Medicine, Beihang University, Beijing, China[5]School of Biomedical Engineering, Anhui Medical University, Hefei, China[6]Beijing Tongren Eye Center, Beijing Tongren Hospital, Capital Medical University, Beijing, China首都医科大学附属北京同仁医院首都医科大学附属同仁医院
Deep learning methods have been successfully applied in medical image classification, segmentation and detection tasks. The U-Net architecture has been widely applied for these tasks. In this paper, we propose a U-Net variant for improved vessel segmentation in retinal fundus images. Firstly, we design a minimal U-Net (Mi-UNet) architecture, which drastically reduces the parameter count to 0.07M compared to 31.03M for the conventional U-Net. Moreover, based on Mi-UNet, we propose Salient U-Net (S-UNet), a bridge-style U-Net architecture with a saliency mechanism and with only 0.21M parameters. S-UNet uses a cascading technique that employs the foreground features of one net block as the foreground attention information of the next net block. This cascading leads to enhanced input images, inheritance of the learning experience of previous net blocks, and hence effective solution of the data imbalance problem. S-UNet was tested on two benchmark datasets, DRIVE and CHASE_DB1, with image sizes of 584 x 565 and 960 x 999, respectively. S-UNet was tested on the TONGREN clinical dataset with image sizes of 1880 x 2816. The experimental results show superior performance in comparison to other state-of-theart methods. Especially, for whole-image input from the DRIVE dataset, S-UNet achieved a Matthews correlation coefficient (MCC), an area under curve (AUC), and an Fl score of 0.8055, 0.9821, and 0.8303, respectively. The corresponding scores for the CHASE_DB1 dataset were 0.8065, 0.9867, and 0.8242, respectively. Moreover, our model shows an excellent performance on the TONGREN clinical dataset. In addition, S-UNet segments images of low, medium, and high resolutions in just 33ms, 91ms and 0.49s, respectively. This shows the real-time applicability of the proposed model.
基金:
National Key Research and Development Program of China under Grant 2016YFF0201002, the National Natural
Science Foundation of China under Grant 61301005 and 61572055, the Hefei Innovation Research Institute, Beihang University, and the Thousand Young
Talent Plan Station between Jicong Zhang and Jiangsu Yuwell Medical Equipment and Supply Co. Ltd.
语种:
外文
被引次数:
WOS:
中科院(CAS)分区:
出版当年[2018]版:
大类|2 区工程技术
小类|2 区计算机:信息系统3 区工程:电子与电气3 区电信学
最新[2023]版:
大类|3 区计算机科学
小类|3 区工程:电子与电气4 区计算机:信息系统4 区电信学
JCR分区:
出版当年[2017]版:
Q1ENGINEERING, ELECTRICAL & ELECTRONICQ1TELECOMMUNICATIONSQ1COMPUTER SCIENCE, INFORMATION SYSTEMS
最新[2023]版:
Q2COMPUTER SCIENCE, INFORMATION SYSTEMSQ2ENGINEERING, ELECTRICAL & ELECTRONICQ2TELECOMMUNICATIONS
第一作者机构:[1]School of Biological Science and Medical Engineering, Beihang University, Beijing, China[2]Hefei Innovation Research Institute, Beihang University, Hefei, China[3]Beijing Advanced Innovation Centre for Biomedical Engineering, Beihang University, Beijing, China[5]School of Biomedical Engineering, Anhui Medical University, Hefei, China
通讯作者:
通讯机构:[1]School of Biological Science and Medical Engineering, Beihang University, Beijing, China[2]Hefei Innovation Research Institute, Beihang University, Hefei, China[3]Beijing Advanced Innovation Centre for Biomedical Engineering, Beihang University, Beijing, China[4]Beijing Advanced Innovation Centre for Big Data-Based Precision Medicine, Beihang University, Beijing, China[5]School of Biomedical Engineering, Anhui Medical University, Hefei, China
推荐引用方式(GB/T 7714):
Jingfei Hu,Hua Wang,Shengbo Gao,et al.S-UNet: A Bridge-Style U-Net Framework With a Saliency Mechanism for Retinal Vessel Segmentation[J].IEEE ACCESS.2019,7:174167-174177.doi:10.1109/ACCESS.2019.2940476.
APA:
Jingfei Hu,Hua Wang,Shengbo Gao,Mingkun Bao,Tao Liu...&Jicong Zhang.(2019).S-UNet: A Bridge-Style U-Net Framework With a Saliency Mechanism for Retinal Vessel Segmentation.IEEE ACCESS,7,
MLA:
Jingfei Hu,et al."S-UNet: A Bridge-Style U-Net Framework With a Saliency Mechanism for Retinal Vessel Segmentation".IEEE ACCESS 7.(2019):174167-174177