Data Augmentation is More Important Than Model Architectures for Retinal Vessel Segmentation

While various deep learning models have recently been proposed or applied to improve the segmentation of vessels in retinal images, the performance gap between different models are often quite small. Such small difference may come from their limited generalization capabilities due to small training data. By simply augmenting data with oriented image patches extracted from the limited training images, we are surprised to observe that even a very simple U-Net with these augmented training patches can outperform the state-of-the-art models with much more complicated architectures or training schemes, and initial gaps between models have become negligible or disappeared. This suggests that it might be more crucial to explore effective data augmentations to extract richer visual information from limited training data, rather than solely focusing on developing other novel deep learning techniques for retinal vessel segmentation.

[1]  Albert C. S. Chung,et al.  Deep supervision with additional labels for retinal vessel segmentation task , 2018, MICCAI.

[2]  Yanning Zhang,et al.  Multiscale Network Followed Network Model for Retinal Vessel Segmentation , 2018, MICCAI.

[3]  Tianfu Wang,et al.  A Cross-Modality Learning Approach for Vessel Segmentation in Retinal Images , 2016, IEEE Transactions on Medical Imaging.

[4]  George D. C. Cavalcanti,et al.  Unsupervised Retinal Vessel Segmentation Using Combined Filters , 2016, PloS one.

[5]  Roberto Marcondes Cesar Junior,et al.  Retinal Vessel Segmentation Using the 2-D Morlet Wavelet and Supervised Classification , 2005, ArXiv.

[6]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[7]  Guang Yang,et al.  Automatic Brain Tumor Detection and Segmentation Using U-Net Based Fully Convolutional Networks , 2017, MIUA.

[8]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[9]  Stephen Lin,et al.  DeepVessel: Retinal Vessel Segmentation via Deep Learning and Conditional Random Field , 2016, MICCAI.

[10]  Roberto Marcondes Cesar Junior,et al.  Retinal vessel segmentation using the 2-D Gabor wavelet and supervised classification , 2005, IEEE Transactions on Medical Imaging.

[11]  Zhuowen Tu,et al.  Deeply-Supervised Nets , 2014, AISTATS.

[12]  Lei Zhang,et al.  Multi-level deep supervised networks for retinal vessel segmentation , 2017, International Journal of Computer Assisted Radiology and Surgery.

[13]  C. Paterson,et al.  Measuring retinal vessel tortuosity in 10-year-old children: validation of the Computer-Assisted Image Analysis of the Retina (CAIAR) program. , 2009, Investigative ophthalmology & visual science.

[14]  Max A. Viergever,et al.  Ridge-based vessel segmentation in color images of the retina , 2004, IEEE Transactions on Medical Imaging.