BTS-ST: Swin transformer network for segmentation and classification of multimodality breast cancer images

Ahmed Iqbal,Muhammad Sharif
DOI: https://doi.org/10.1016/j.knosys.2023.110393
IF: 8.139
2023-02-16
Knowledge-Based Systems
Abstract:Breast cancer is considered the most commonly diagnosed cancer globally and falls second to lung cancer. For the early detection of breast tumors in women, breast cancer analysis using Ultrasound, Mammography, and MRI modalities is used as an initial screening process. Due to the random variation, irregular shapes, and blurred boundaries of tumor regions, the accurate segmentation of breast tumors is still a tricky task. The existing convolutional neural networks (CNNs) inherit their limitation by extracting global context information and, in most cases, proved less efficient in obtaining satisfactory results. As a solution, we proposed the BTS-ST network, a novel framework for breast tumor segmentation and classification that Swin Transformer inspires. The BTS-ST network incorporates Swin Transformer into traditional CNN-based UNet to improve global modeling capabilities. To improve the feature representation capability of irregularly shaped tumors, we first introduced a Spatial Interaction block (SIB), encoding spatial knowledge in the SwinsTransformer block by developing pixel-level correlation. The segmentation accuracy of small-scale tumor region is increased by building a Feature Compression block (FCB) to prevent information loss and compress smaller-scale features in patch tokensdown sampling of ST. Finally, a Relationship Aggregation block (RAB) is developed as a bridge among dual encoders to combine global dependencies from ST into the features from CNNshierarchically. The extensive experiments are performed on breast tumor segmentation and classification tasks utilizing multimodality Ultrasound, Mammogram, and MRI-based dataset. The results demonstrate that our proposed solution is comparatively better than other state-of-the-art methods.
computer science, artificial intelligence
What problem does this paper attempt to address?