Ultrasound Breast Tumor Detection based on Vision Graph Neural Network

Breast cancer is the most commonly diagnosed cancer in women in the United States. Early detection of breast tumors enables prompt determination of cancer status, significantly boosting patient survival rate. Non-invasive and non-ionizing ultrasound imaging is a widely used diagnosing modality in clinic. To assist clinicians in breast cancer diagnosis, we implemented a vision graph neural networks (ViG)-based pipeline that can achieve accurate binary classification (normal vs. breast tumor) and multiclass classification (normal, benign, and malignant) from breast ultrasound images. Our results demonstrated that the average accuracy of ViG is 100.00% for binary and 87.18% for multiclass classification tasks. To the best of our knowledge, this is the first end-to-end, graph-feature-based deep learning pipeline to achieve accurate breast tumor detection from ultrasound images. The proposed ViG-based classifier is accessible for clinical implementation and has the potential to enhance lesion detection from ultrasound images.