BVU-Net: A U-Net Modification by VGG-Batch Normalization for Retinal Blood Vessel Segmentation

Open

Anita Desiani, Erwin, Bambang Suprihatin, Yogi Wahyudi, Endro Setyo Cahyono, Muhammad Arhami

2022 International Journal of Intelligent Engineering and Systems Vol. 15 Issue 6 Article Cited by 10 SDG 16SDG 17 Quartile

Abstract

The study proposes a BVU-Net architecture that combines the advantages of VGG and U-Net. The VGG architecture has a smaller kernel size which speeds up the training process. VGG bears some resemblance to the Encoder portion of U-Net. In this study, the BVU-Net encoder uses the VGG architecture with the addition of batch normalization. The addition of batch normalization aims to help simplify weight initialization so that the training process is faster and reduces the risk of overfitting, while the decoder section still uses the U-Net architecture. BVUNet is expected to be able to overcome the weakness of U-Net architecture in retinal image blood vessel segmentation. The performance results produced by BVU-Net on the DRIVE dataset are 96.36% accuracy, 78.71 sensitivity, 98.1% specificity, F1-score 78.9, and IOU 0.65. The results of BVU-Net performance on the STARE dataset are accuracy of 96.39%, sensitivity of 79.35%, specificity of 98.52%, F1-score of 77.16%, and IoU of 0.63. Based on these results indicate that BVU-Net has a better performance on the DRIVE dataset in detecting retinal blood vessels than STARE. This can be seen from the sensitivity value of DRIVE which is higher than STARE. The BVU-Net IoU results on DRIVE and STARE are greater than 0.5 and the F1-score above 70%, indicating that BVU-Net is capable and balanced in detecting the intersection area between blood vessels and the background on retinal images © 2022, International Journal of Intelligent Engineering and Systems.All Rights Reserve

Affiliations

Mathematics Department, Mathematics and Natural Science Faculty, Universitas Sriwijaya, Indralaya, Indonesia; Computer Engineering Department, Computer Science Faculty, Universitas Sriwijaya, Indralaya, Indonesia; Informatics Technique Department, Politeknik Negeri Lhokseumawe, Lhokseumawa, Indonesia

Research at a Glance

Premium content — register to unlock

Research at a Glance

Register to unlock

Topics & SDG Alignment

Premium content — register to unlock

Topics & SDG Alignment

Register to unlock

Collaboration

Premium content — register to unlock

Collaboration

Register to unlock

Author Profile (Selected)

Premium content — register to unlock

Author Profile (Selected)

Register to unlock

References Overview

Premium content — register to unlock

References Overview

Register to unlock

Journal & Source

Premium content — register to unlock

Journal & Source

Register to unlock

Metadata & Integrity

Premium content — register to unlock

Metadata & Integrity

Register to unlock