Retinal vessel segmentation network based on rough attention fusion mechanism and Group Transformer
The morphological changes in retinal vessels play a crucial role in the diagnosis of early ophthalmic dis-eases.Beyond eye diseases,conditions such as diabetes and cardiovascular diseases can also be identified through the morphology of retinal vessels.However,retinal vessels possess a complex tissue structure and are easily influenced by factors such as lighting,making their accurate segmentation challenging.To address these issues,a retinal vessel seg-mentation network that initially incorporates a rough attention fusion module(RAFM)is proposed.This module is based on the theory of rough set upper and lower approximations,employing global max pooling and global average pooling to describe the upper and lower bounds of attention coefficients,and sequentially integrates channel attention mechanisms with spatial attention mechanisms.Subsequently,the RAFM is integrated into the Group Transformer U network(GT U-Net),constructing a retinal vessel segmentation network based on the rough attention fusion mecha-nism and Group Transformer.Finally,comparative experiments conducted on the publicly available DRIVE color fun-dus image dataset demonstrate that the network achieves an accuracy,F1 score,and AUC of 0.963 1,0.848 8,and 0.981 2,respectively,on the test set.Compared to the GT U-Net model,theF1score and AUC were improved by 0.35%and 0.21%,respectively;and when compared to other contemporary mainstream retinal vessel segmentation networks,it exhibits certain advantages.