Enhancing Breast Cancer Detection with Vision Transformers and Graph Neural Networks
Yeming Cai
Zhenglin Li
Yang Wang
- ViT
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 arXivComments on this paper
