GRU-Net: Gaussian Attention Aided Dense Skip Connection Based MultiResUNet for Breast Histopathology Image Segmentation

Ayush Roy,Payel Pramanik,Sohom Ghosal,Daria Valenkova,Dmitrii Kaplun,Ram Sarkar
2024-08-01
Abstract:Breast cancer is a major global health concern. Pathologists face challenges in analyzing complex features from pathological images, which is a time-consuming and labor-intensive task. Therefore, efficient computer-based diagnostic tools are needed for early detection and treatment planning. This paper presents a modified version of MultiResU-Net for histopathology image segmentation, which is selected as the backbone for its ability to analyze and segment complex features at multiple scales and ensure effective feature flow via skip connections. The modified version also utilizes the Gaussian distribution-based Attention Module (GdAM) to incorporate histopathology-relevant text information in a Gaussian distribution. The sampled features from the Gaussian text feature-guided distribution highlight specific spatial regions based on prior knowledge. Finally, using the Controlled Dense Residual Block (CDRB) on skip connections of MultiResU-Net, the information is transferred from the encoder layers to the decoder layers in a controlled manner using a scaling parameter derived from the extracted spatial features. We validate our approach on two diverse breast cancer histopathology image datasets: TNBC and MonuSeg, demonstrating superior segmentation performance compared to state-of-the-art methods. The code for our proposed model is available on <a class="link-external link-https" href="https://github.com/AyushRoy2001/GRU-Net" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The main problems that this paper attempts to solve are the complex feature analysis and efficient information transfer in breast cancer pathological image segmentation. Specifically: 1. **Complex Feature Analysis**: Pathologists face challenges when analyzing pathological images, especially when dealing with complex cell structures and background noise. These tasks are not only time - consuming and labor - intensive but also require a high level of professional knowledge. Therefore, the development of effective computer - aided diagnosis tools is crucial for early detection and treatment planning. 2. **Efficient Information Transfer**: Traditional deep - learning models have deficiencies in handling multi - scale features and information flow control. For example, the standard U - Net, although performing well in image segmentation, still has certain limitations when dealing with complex backgrounds and multi - scale features. In addition, how to effectively incorporate text information (such as the descriptions in pathological reports) into the model is also an important research direction. For this reason, the paper proposes an improved MultiResU - Net model, named GRU - Net. The main innovations of this model include: 1. **Multi - Resolution Feature Extraction**: Using MultiResU - Net as the backbone network, it can effectively analyze and segment multi - scale complex features and ensure the effective flow of features through skip connections. 2. **Gaussian Distribution Attention Module (GdAM)**: Introducing an attention mechanism based on the Gaussian distribution, combined with pathology - related text information, highlighting specific spatial regions through statistical information and spatial features, thereby improving the segmentation performance of the model. 3. **Controlled Dense Residual Block (CDRB)**: Using CDRB in skip connections, reducing the transmission of irrelevant features through controlled information transfer from the encoder layer to the decoder layer, improving the robustness and accuracy of the model. The paper demonstrates the superiority of GRU - Net in segmentation performance by validating it on two different breast cancer pathological image datasets (TNBC and MonuSeg), especially significantly outperforming existing methods in metrics such as the Dice coefficient and the Intersection over Union (IoU).