MFA-UNet: a vessel segmentation method based on multi-scale feature fusion and attention module

Autor: Juan Cao, Jiaran Chen, Yuanyuan Gu, Jinjia Liu
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Frontiers in Neuroscience, Vol 17 (2023)
Druh dokumentu: article
ISSN: 1662-453X
DOI: 10.3389/fnins.2023.1249331
Popis: IntroductionThe accurate segmentation of retinal vessels is of utmost importance in the diagnosis of retinal diseases. However, the complex vessel structure often leads to poor segmentation performance, particularly in the case of microvessels.MethodsTo address this issue, we propose a vessel segmentation method composed of preprocessing and a multi-scale feature attention network (MFA-UNet). The preprocessing stage involves the application of gamma correction and contrast-limited adaptive histogram equalization to enhance image intensity and vessel contrast. The MFA-UNet incorporates the Multi-scale Fusion Self-Attention Module(MSAM) that adjusts multi-scale features and establishes global dependencies, enabling the network to better preserve microvascular structures. Furthermore, the multi-branch decoding module based on deep supervision (MBDM) replaces the original output layer to achieve targeted segmentation of macrovessels and microvessels. Additionally, a parallel attention mechanism is embedded into the decoder to better exploit multi-scale features in skip paths.ResultsThe proposed MFA-UNet yields competitive performance, with dice scores of 82.79/83.51/84.17/78.60/81.75/84.04 and accuracies of 95.71/96.4/96.71/96.81/96.32/97.10 on the DRIVE, STARE, CHASEDB1, HRF, IOSTAR and FIVES datasets, respectively.DiscussionIt is expected to provide reliable segmentation results in clinical diagnosis.
Databáze: Directory of Open Access Journals