Attention-enriched Deeper UNet (ADU-NET) for Disease Diagnosis in Breast Ultrasound and Retina Fundus Images

Chukwuebuka Joseph Ejiyi,Zhen Qin,Victor K. Agbesi,Makuachukwu Bennedith Ejiyi,Ijeoma A. Chikwendu,Oluwatoyosi F. Bamisile,Favour Ezinne Onyekwere,Olusola O. Bamisile
DOI: https://doi.org/10.1007/s13748-024-00340-1
2024-01-01
Progress in Artificial Intelligence
Abstract:In image segmentation, effective upsampling plays a pivotal role in recovering lost spatial information during the process of downsampling. Standard skip connections designed to mitigate this and prevalent in most models, often fall short of maintaining high segmentation performance, because of the loss of spatial information when it is transferred from the encoder to the decoder parts of the network. Responding to these limitations, we introduce the Attention-Enriched Deeper UNet (ADU-Net) a novel framework designed and tested on two medical image modalities – breast ultrasound images (BUSI) and retinal fundus images (RFI) for the effective transfer of information from the encoder to the decoder. The ADU-Net seamlessly combines global context modules (GCM) and progressive context refinement modules (PCRM) in the skip connection in a U-shaped network structure to capture the contextual information and enhance feature attention respectively. Another portion of the network, ConvGroup blocks set at its base facilitate the integration of convolutional operations while the deeper UNet was chosen for improved feature representation, handling of complex data, and enhanced hierarchical feature extraction. With these components, the ADU-Net effectively captures rich spatial details and crucial contextual information from input images resulting in precise and robust segmentation. ADU-Net achieved 76.49
What problem does this paper attempt to address?