Breast Cancer Tumour Masks Segmentation Using Modified U-Net with Channel Attention

Abstract

Abstract: A modified U-Net for biomedical image segmentation with channel attention is developed for breast cancer image segmentation. A unique step in biomedical image analysis is segmentation because it determines the accuracy of image analysis algorithms. Algorithms for accurate and early breast cancer detection in women using mammograms depend highly on the accuracy of the segmentation stage. In this thesis, in order to improve on the segmentation ability of the traditional UNet architecture, it was modified by varying the fixed kernel sizes that it was originally designed for. In this model, progressively increasing kernel sizes of 3x3, 5x5, 7x7, 9x9 and 11x11 were used, which is different from the architecture of UNet that depended on fixed 3x3 kernel sizes for all convolutional layers. The first Convolutional block of the proposed model contains two Conv2D layers with 3x3 kernel sizes, ReLU (Rectified Linear Unit) activation function, and 16 channel dimensions. A maxpooling2D layer with pool size of (2x2) was placed to downsample the image size before feeding to the next convolutional block which used the same activation function, with 32 channel dimension and 5x5 kernel size. The next convolutional block used 64 channel dimensions with 7x7 kernel size, then a 9x9 kernel size Conv2D block with 128 channel dimensions was placed before the Conv2D with upsampling. The squeeze and excitation block was placed after this convolutional layer and in-between the two convolutional blocks with 11x11 kernel sizes, to reallocate the weight of the essential features to ensure that more discriminate features are learnt from the breast cancer images to improve the overall representational strength of the network through the performance of dynamic feature recalibration of the image channels. This significantly improved the performance of the segmentation by explicitly modeling the interdependencies between channels in the convolutional layers. The increasing kernel sizes enables the model to learn more discriminate features from the images. Jaccard similarity index, F1 score, Recall, Precision, F2 score, and accuracy results were used to compare the result of the proposed model with other models such as U-Net model and Baseline model. The results shows a better performance of 0.88 for Jaccard similarity index, 0.94 for F1 score, 0.93 for recall, 0.92 for precision, 0.92 for F2 score, and an accuracy score of 0.98%. The model is good for biomedical image segmentation.