35

Enhancing Breast Cancer Detection with Vision Transformers and Graph Neural Networks

Yeming Cai
Zhenglin Li
Yang Wang
Main:9 Pages
5 Figures
2 Tables
Abstract

Breast cancer is a leading cause of death among women globally, and early detection is critical for improving survival rates. This paper introduces an innovative framework that integrates Vision Transformers (ViT) and Graph Neural Networks (GNN) to enhance breast cancer detection using the CBIS-DDSM dataset. Our framework leverages ViT's ability to capture global image features and GNN's strength in modeling structural relationships, achieving an accuracy of 84.2%, outperforming traditional methods. Additionally, interpretable attention heatmaps provide insights into the model's decision-making process, aiding radiologists in clinical settings.

View on arXiv
Comments on this paper