Importance of Data Augmentation and Transfer Learning on Retinal Vessel Segmentation

Deari S., Öksüz İ., ULUKAYA S.

29th Telecommunications Forum, TELFOR 2021, Virtual, Belgrade, Serbia, 23 - 24 November 2021, (Full Text)

  • Publication Type: Conference Paper / Full Text
  • Doi Number: 10.1109/telfor52709.2021.9653400
  • City: Virtual, Belgrade
  • Country: Serbia
  • Keywords: Data augmentation, Retinal blood vessel, Segmentation, Transfer learning, U-NET
  • Trakya University Affiliated: Yes

Abstract

Automatic segmentation of retinal fundus images for extracting blood vessels is an essential task in the diagnostic classification of hypertension, glaucoma, and diabetic retinopathy, which are the leading causes of blindness. In this paper, we employed a transfer learning strategy for improved retinal vessel extraction. Firstly, we trained the U-NET model on CHASE DB1 and DRIVE databases. By using data augmentations on datasets we enable the U-NET model to learn retinal vessel features better. We examined the data augmentation types, namely, pixel-level transformations and affine transformations. Secondly, we utilized the transfer learning approach on two datasets and achieved comparable results with the state-of-the-art studies on retinal vessel segmentation task. Also, we employed combination of affine and pixel-level transformations to further boost segmentation performance.