Enhancing Breast Cancer Detection with Vision Transformers and Graph Neural Networks
Cai, Yeming, Li, Zhenglin, Wang, Yang
–arXiv.org Artificial Intelligence
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.
arXiv.org Artificial Intelligence
Jul-21-2025
- Country:
- Asia
- China > Hubei Province
- Wuhan (0.05)
- Japan (0.04)
- China > Hubei Province
- North America > United States
- Texas > Brazos County > College Station (0.04)
- Asia
- Genre:
- Research Report (0.64)
- Industry:
- Health & Medicine > Therapeutic Area > Oncology > Breast Cancer (0.96)
- Technology: