Dual encoding feature filtering generalized attention UNET for retinal vessel segmentation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Islam, Md Tauhidul, Da-Wen, Wu, Qing-Qing, Tang, Kai-Yang, Zhao, Teng, Yin, Yan-Fei, Li, Wen-Yi, Shang, Jing-Yu, Liu, Hai-Xian, Zhang
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909634777841664
author Islam, Md Tauhidul
Da-Wen, Wu
Qing-Qing, Tang
Kai-Yang, Zhao
Teng, Yin
Yan-Fei, Li
Wen-Yi, Shang
Jing-Yu, Liu
Hai-Xian, Zhang
author_facet Islam, Md Tauhidul
Da-Wen, Wu
Qing-Qing, Tang
Kai-Yang, Zhao
Teng, Yin
Yan-Fei, Li
Wen-Yi, Shang
Jing-Yu, Liu
Hai-Xian, Zhang
contents Retinal blood vessel segmentation is crucial for diagnosing ocular and cardiovascular diseases. Although the introduction of U-Net in 2015 by Olaf Ronneberger significantly advanced this field, yet issues like limited training data, imbalance data distribution, and inadequate feature extraction persist, hindering both the segmentation performance and optimal model generalization. Addressing these critical issues, the DEFFA-Unet is proposed featuring an additional encoder to process domain-invariant pre-processed inputs, thereby improving both richer feature encoding and enhanced model generalization. A feature filtering fusion module is developed to ensure the precise feature filtering and robust hybrid feature fusion. In response to the task-specific need for higher precision where false positives are very costly, traditional skip connections are replaced with the attention-guided feature reconstructing fusion module. Additionally, innovative data augmentation and balancing methods are proposed to counter data scarcity and distribution imbalance, further boosting the robustness and generalization of the model. With a comprehensive suite of evaluation metrics, extensive validations on four benchmark datasets (DRIVE, CHASEDB1, STARE, and HRF) and an SLO dataset (IOSTAR), demonstrate the proposed method's superiority over both baseline and state-of-the-art models. Particularly the proposed method significantly outperforms the compared methods in cross-validation model generalization.
format Preprint
id arxiv_https___arxiv_org_abs_2506_02312
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Dual encoding feature filtering generalized attention UNET for retinal vessel segmentation
Islam, Md Tauhidul
Da-Wen, Wu
Qing-Qing, Tang
Kai-Yang, Zhao
Teng, Yin
Yan-Fei, Li
Wen-Yi, Shang
Jing-Yu, Liu
Hai-Xian, Zhang
Image and Video Processing
Computer Vision and Pattern Recognition
I.4; I.5
Retinal blood vessel segmentation is crucial for diagnosing ocular and cardiovascular diseases. Although the introduction of U-Net in 2015 by Olaf Ronneberger significantly advanced this field, yet issues like limited training data, imbalance data distribution, and inadequate feature extraction persist, hindering both the segmentation performance and optimal model generalization. Addressing these critical issues, the DEFFA-Unet is proposed featuring an additional encoder to process domain-invariant pre-processed inputs, thereby improving both richer feature encoding and enhanced model generalization. A feature filtering fusion module is developed to ensure the precise feature filtering and robust hybrid feature fusion. In response to the task-specific need for higher precision where false positives are very costly, traditional skip connections are replaced with the attention-guided feature reconstructing fusion module. Additionally, innovative data augmentation and balancing methods are proposed to counter data scarcity and distribution imbalance, further boosting the robustness and generalization of the model. With a comprehensive suite of evaluation metrics, extensive validations on four benchmark datasets (DRIVE, CHASEDB1, STARE, and HRF) and an SLO dataset (IOSTAR), demonstrate the proposed method's superiority over both baseline and state-of-the-art models. Particularly the proposed method significantly outperforms the compared methods in cross-validation model generalization.
title Dual encoding feature filtering generalized attention UNET for retinal vessel segmentation
topic Image and Video Processing
Computer Vision and Pattern Recognition
I.4; I.5
url https://arxiv.org/abs/2506.02312