Transcriptome-informed multi-modal AI for predicting neoadjuvant therapy response from breast cancer biopsies
Organizations: Ataraxis AI, New York, NY, USA. · Breast Oncology Program, Dana-Farber Cancer Institute, Boston, MA, USA. · Harvard Medical School, Boston, MA, USA. · Division of Medical Oncology, Rutgers Cancer Institute, New Brunswick, NJ, USA. ·
Abstract
Scarcity of labeled data limits development of deep learning biomarkers in oncology. We develop a two-stage AI model predicting pathological complete response (pCR) to neoadjuvant therapy in breast cancer. The first stage learns the transcriptome from histopathology using 8,742 patients across 32 cancer types, corroborated by pathologist review and spatial agreement with measured expression. This simplifies the second stage to predicting pCR from inferred expression and clinical variables. Developed using 1,080 patients (five cohorts) and evaluated in 1,412 patients (nine cohorts), the model achieves a pooled AUROC of 0.79 (95% CI, 0.73-0.85), discriminating responders within molecular subtypes. It outperforms histopathological biomarkers, remaining stable across intratumoral sampling and with minimal biopsy tissue. Ablations show transcriptome-wide inference improves discrimination over clinical variables alone or one-stage pathology models, and robustness by avoiding genomic assays' gene selection constraints. These results indicate that biologically informed compression may generalize to data-sparse applications in precision oncology.
Figures & tables
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Development cohorts (N=1,080) | Evaluation cohorts (N=1,412) |
| Age median, [IQR] | 54.00 [45.00–63.00] | 52.29 [44.00–62.00] |
| Race | ||
| Asian | 7 (0.65%) | 10 (0.71%) |
| Black or African American | 15 (1.39%) | 119 (8.43%) |
| White | 315 (29.17%) | 315 (22.31%) |
| Other/Unknown | 743 (68.80%) | 968 (68.56%) |
| Category | Protege | UChicago | Nexus | Dordrecht | IMPRESS | Jagiellonian | Shaare Zedek | Providence | Łódź |
| Patients | 391 | 289 | 153 | 146 | 126 | 116 | 73 | 67 | 51 |
| Age median [IQR] | 56 [46–65] | 51 [42–61] | 50 [44–56] | 52 [42–61] | 54 [46–59] | 50 [39–57] | 48 [43–59] | 53 [43–63] | 63 [53–69] |
| Race | |||||||||
| Asian | 5 (1%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 5 (7%) | 0 (0%) |
| Black or African American | 117 (30%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 2 (3%) | 0 (0%) |
| White | 263 (67%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 0 (0%) | 52 (78%) | 0 (0%) |
| Training | Validation | |||
| Project | Patients | WSIs | Patients | WSIs |
| ACC † | 46 | 183 | 9 | 40 |
| BLCA | 293 | 337 | 84 | 108 |
| BRCA | 835 | 886 | 212 | 227 |
| CESC | 213 | 220 | 54 | 57 |
| CHOL † | 28 | 28 | 7 | 7 |
| Cancer type | Patients | RNA-seq samples | WSIs |
| BRCA | 106 | 106 | 106 |
| CCRCC | 222 | 314 | 340 |
| COAD | 104 | 104 | 104 |
| GBM | 157 | 157 | 210 |
| HNSCC | 110 | 110 | 142 |
| LSCC | 108 | 109 | 132 |
| Hyperparameter | Range |
| Shared by all tuned neural networks | |
| Elastic-net regularization weight ( ) | |
| Elastic-net mixing weight ( ) | |
| Intermediate feature dimension ( ) | |
| Pathology branch | |
| Pathology-branch dropout | |
| Method | As published | Recomputed |
| TNBC ( ) | ||
| cTILs | 0.708 (0.574–0.841) | 0.691 (0.553–0.820) |
| LTR | 0.686 (0.550–0.821) | 0.583 (0.439–0.727) |
| ITR | 0.715 (0.584–0.845) | 0.568 (0.430–0.713) |
| HER2 ( ) | ||
| cTILs | 0.679 (0.543–0.814) | 0.623 (0.481–0.760) |
| Signature | Source | Total | Retained |
| Merck T-cell-inflamed GEP | Ayers et al. [3] | 18 | 18 |
| Oncotype DX | Paik et al. [79] | 21 | 20 |
| HER2DX | Prat et al. [86] | 32 | 28 |
| Davoli immune evasion | Davoli et al. [23] | 42 | 39 |
| Danaher immune cell markers | Danaher et al. [22] | 60 | 47 |
| MammaPrint | Van’t Veer et al. [106] | 70 | 50 |
Clinical Outcome Prediction
Clinical outcome prediction uses patient records, imaging, signals, and biomarkers to estimate risks, disease trajectories, treatment response, and survival.
See all 261 papers