A unified multi modal transformer framework for breast cancer recurrence prediction and survival analysis
Abstract
Breast cancer recurrence prediction is an important feature of post-treatment therapy, requiring accurate identification of both recurrence risk and time-to-event outcomes. In this paper, we offer a unified deep learning system that jointly performs survival analysis and multi-class recurrence classification, enabling full risk stratification for breast cancer patients. The proposed model includes a Transformer-based survival module to estimate time-until-recurrence, and an attention-guided classification module to differentiate between second primary cancer, low-risk, and high-risk recurrence instances. A multi-modal dataset comprising clinical, molecular, demographic, and lifestyle data is created from established sources like METABRIC, GSE2034, GSE2990, BCSC, and the Breast Cancer Coimbra dataset. The model uses cross-modal feature fusion, autoencoder-based dimensionality reduction, and attention-based feature attribution for applicability and accessibility. Experimental results show better accuracy, precision, recall, and F1-score of 99.12%, 98.75%, 99.08%, and 98.91%, outperforming standard machine learning and survival models. This unified paradigm gives doctors a powerful, interpretable tool for early intervention and personalized breast cancer treatment.
Fields of Study & Subjects
AI in cancer detection
Breast Cancer Treatment Studies
Infrared Thermography in Medicine
Publication details
- Journal / source
- Scientific Reports
- Publisher
- Springer Science and Business Media LLC
- Volume
- 16
- Issue
- 1
- Publication date
- 11 February 2026
- Language
- EN