MAFI-Net: A Multi-level Attention Feature Interaction Network for Retinal Vessel Segmentation
摘要
Retinal vessel segmentation holds important clinical value in early screening and auxiliary diagnosis of fundus diseases. However, fundus images often suffer from a severe class imbalance problem, where tiny vessels occupy only a small proportion of pixels and are highly sensitive to noise interference and edge blurring. It hinders existing methods from accurately segmenting large vessels and microvessels simultaneously. To this end, a novel Multi-level Attention Feature Interaction Network (MAFI-Net) is proposed. To alleviate the imbalance at the training sample level, this paper proposes an automated coarse and fine vessel patch classification algorithm. In addition, a Multi-level Attention Guided Fusion (MAGF) module is designed to unify deep semantic cues with shallow spatial details, effectively mitigating the degradation of structure details during feature fusion. Meanwhile, a Multi-Constraint Guided Loss (MCGL) is proposed, which introduces constraints related to vessel skeleton and boundary information on top of traditional loss factors, thereby achieving more robust structural optimization. Experiments on DRIVE, STARE, and CHASE_DB1 show that MAFI-Net achieves competitive segmentation performance.