This paper delves into the challenges and advancements in the field of medical image segmentation, particularly focusing on breast cancer diagnosis. The authors propose a novel Transformer-based segmentation model that addresses the limitations of traditional convolutional neural networks (CNNs), such as U-Net, in accurately localizing and segmenting small lesions within breast cancer images. The model introduces an axial attention mechanism to enhance the computational efficiency and address the issue of global contextual information that is often overlooked by CNNs. Additionally, the paper discusses improvements tailored to the small dataset challenge, including the incorporation of relative position information and a gated axial attention mechanism to refine the model's focus on relevant features. The proposed model aims to significantly improve the segmentation accuracy of breast cancer images, offering a more efficient and effective tool for computer-aided diagnosis.
@article{arxiv.2409.12347,
title = {Axial Attention Transformer Networks: A New Frontier in Breast Cancer Detection},
author = {Weijie He and Runyuan Bao and Yiru Cang and Jianjun Wei and Yang Zhang and Jiacheng Hu},
journal= {arXiv preprint arXiv:2409.12347},
year = {2024}
}