English

Leveraging Transformers to Improve Breast Cancer Classification and Risk Assessment with Multi-modal and Longitudinal Data

Image and Video Processing 2023-11-16 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Breast cancer screening, primarily conducted through mammography, is often supplemented with ultrasound for women with dense breast tissue. However, existing deep learning models analyze each modality independently, missing opportunities to integrate information across imaging modalities and time. In this study, we present Multi-modal Transformer (MMT), a neural network that utilizes mammography and ultrasound synergistically, to identify patients who currently have cancer and estimate the risk of future cancer for patients who are currently cancer-free. MMT aggregates multi-modal data through self-attention and tracks temporal tissue changes by comparing current exams to prior imaging. Trained on 1.3 million exams, MMT achieves an AUROC of 0.943 in detecting existing cancers, surpassing strong uni-modal baselines. For 5-year risk prediction, MMT attains an AUROC of 0.826, outperforming prior mammography-based risk models. Our research highlights the value of multi-modal and longitudinal imaging in cancer diagnosis and risk stratification.

Keywords

Cite

@article{arxiv.2311.03217,
  title  = {Leveraging Transformers to Improve Breast Cancer Classification and Risk Assessment with Multi-modal and Longitudinal Data},
  author = {Yiqiu Shen and Jungkyu Park and Frank Yeung and Eliana Goldberg and Laura Heacock and Farah Shamout and Krzysztof J. Geras},
  journal= {arXiv preprint arXiv:2311.03217},
  year   = {2023}
}

Comments

ML4H 2023 Findings Track