Zhuo Z, Huang J, Lu K, Pan D, Feng S. A size-invariant convolutional network with dense connectivity applied to retinal vessel segmentation measured by a unique index.
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE 2020;
196:105508. [PMID:
32563893 DOI:
10.1016/j.cmpb.2020.105508]
[Citation(s) in RCA: 6] [Impact Index Per Article: 1.5] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Subscribe] [Scholar Register] [Received: 04/15/2019] [Revised: 03/15/2020] [Accepted: 04/12/2020] [Indexed: 06/11/2023]
Abstract
BACKGROUND AND OBJECTIVES
Retinal vessel segmentation (RVS) helps in diagnosing diseases such as hypertension, cardiovascular diseases, and others. Convolutional neural networks are widely used in RVS tasks. However, how to comprehensively evaluate the segmentation results and how to improve the networks' learning ability are two great challenges.
METHODS
In this paper, we proposed an ingenious index: fusion score (FS), which provides an overall measure for those binary images. The FS converts multiple metrics into a single target, and therefore facilitates the optimal threshold's selection and models' comparison. In addition, We simultaneously combined size-invariant feature maps and dense connectivity together to improve the traditional CNN's learning ability. Therefore, a size-invariant convolutional network with dense connectivity is designed for RVS. The size-invariant skill helps the deep layers create feature maps with high resolution. The dense connectivity technique is utilized to integrate those hierarchical features and reuse characteristic maps to enhance the network's learning ability. Finally, an optimized threshold is used on the output image to obtain a binary image.
RESULTS
The results of experiments conducted on two shared retinal image databases, DRIVE and STARE, demonstrate that our approach outperforms other techniques when evaluated in terms of F1-score, Matthews correlation coefficient (MCC), G-mean and FS. In addition, the cross training reveals that our method has stronger robustness with respect to training sets. Segmenting a 565 × 584 image only takes 39 ms with a single GPU (graphics processing unit).
CONCLUSIONS
Compared with those traditional metrics, the FS is a better indicator to measure the results of RVS tasks. The experimental results revealed that the proposed method is more suitable for real-world applications.
Collapse