Deep-learning-based breast cancer stage prediction from H&E-stained whole-slide images in resource-constrained settings

Read full text
Authors: Zsolt Bedőházi, András Biricz, Oz Kilim, Nick Foster, Barbara Gregus, Anna-Mária Tőkés, Péter Pollner, István Csabai, Beatrice S. Knudsen
Cited by: 1
References: 36

Abstract

Artificial intelligence shows promise for evaluating primary breast cancer, including nodal status and molecular subtype. Here, we present a resource-aware deep learning pipeline that combines a Vision Transformer feature extractor with an attention-based multiple instance learning (MIL) aggregator to predict pathological tumor-node-metastasis (pTNM) stage from hematoxylin and eosin whole-slide images (WSIs). Motivated by deployment in constrained settings, we operate at 2.5× magnification (≈4.0 μm/pixel), well below the 20–40× typically used in computational pathology. For feature extraction, we evaluated three backbones: (1) the UNI foundation model, (2) UNI fine-tuned on the BReAst Carcinoma Subtyping (BRACS) dataset, and (3) a ResNet-50 fine-tuned on BRACS. The embeddings from the best-performing UNI fine-tuned network were used as input to the MIL model, which was trained and validated on 247 WSIs from 214 patients in the internal cohort. Performance was assessed on three test sets: an internal hold-out from the same cohort (82 WSIs from 72 patients), curated subsets of the Nightingale High-Risk Breast Cancer Prediction (NG) dataset (9489 WSIs from 574 patients), and TCGA-BRCA (731 WSIs from 678 patients), all preprocessed identically at 2.5×. The pipeline achieved area under the receiver operating characteristic curve values of 0.663, 0.672, and 0.632 for the internal, NG, and TCGA-BRCA test sets, respectively. Whereas operating at 2.5× may limit access to fine cellular cues, our results indicate that stage-relevant information can still be captured at this resolution. This study provides a transparent, compute-efficient WSI-only baseline for pTNM stage prediction from WSIs, supporting feasibility in resource-constrained environments. • Demonstrated WSI-only pTNM stage prediction at 2.5× for stages I–III. • Delivered a transparent, compute-efficient baseline for low-resource settings. • Combined UNI foundation model with BRACS fine-tuning and attention MIL. • Externally validated on Nightingale (NG) and TCGA-BRCA with consistent 2.5× preprocessing. • Achieve ROC AUCs: 0.663 (internal), 0.672 (NG), 0.632 (TCGA-BRCA).

Fields of Study & Subjects

AI in cancer detection Breast Cancer Treatment Studies Cell Image Analysis Techniques

Publication details

Journal / source
Journal of Pathology Informatics
Publisher
Elsevier BV
Volume
21
Pages
100644–100644
Publication date
27 January 2026
Language
EN