Enabling clinical use of foundation models for computational pathology
Organizations: Institute for Cancer Genetics and Informatics, Oslo University Hospital, Oslo, Norway · Department of Pathology, University Medical Center Utrecht, Utrecht, The Netherlands · Department of Oncology, University of Oxford, Oxford, UK · CRUK Beatson Institute of Cancer Research, Garscube Estate, Glasgow, UK · Glasgow Tissue Research Facility, University of Glasgow, Queen Elizabeth University Hospital, Glasgow, UK · Area for Improvement and Digital Transformation, Norwegian Offshore Directorate, Stavanger, Norway · Pathology Department, Hospital Clínic, Barcelona, Spain · Institut d’Investigacions Biomèdiques August Pi I Sunyer (IDIBAPS), Barcelona, Spain · Department of Clinical Foundations, Universitat de Barcelona, Barcelona, Spain · School of Cancer Sciences, Wolfson Wohl Cancer Research Centre, University of Glasgow, Glasgow, UK · Institute of Clinical Medicine, University of Oslo, Oslo, Norway · Department of Gastrointestinal Surgery, Oslo University Hospital, Oslo, Norway · Gloucestershire Cellular Pathology Laboratory, Cheltenham General Hospital, Cheltenham, UK · Institute of Pathology, University Medical Center, Mainz, Germany · Department of Gastrointestinal Oncology, National Cancer Center Hospital East, Kashiwa, Japan · Translational Research Support Office, National Cancer Center Hospital East, Kashiwa, Japan · Department of Cancer Biology and Pathology, University College London, London, UK · Department of Gastroenterological Surgery, Osaka International Cancer Institute, Osaka, Japan · Centre for Human Genetics, University of Oxford, Oxford, UK · Oxford NIHR Comprehensive Biomedical Research Centre, Oxford University Hospitals NHS Foundation Trust, John Radcliffe Hospital, Oxford, UK · Nuffield Division of Clinical Laboratory Sciences, University of Oxford, Oxford, UK · Department of Informatics, University of Oslo, Oslo, Norway · Centre for Research-based Innovation Visual Intelligence, UiT The Arctic University of Norway, Tromsø, Norway
Abstract
Foundation models for computational pathology are expected to facilitate the development of high-performing, generalisable deep learning systems. However, in addition to biologically relevant features, current foundation models also capture pre-analytic and scanner-specific variation that bias the predictions made by downstream task-specific models trained on these features. Here we show that introducing novel robustness losses during downstream model training reduces sensitivity to technical variability. A purpose-designed comprehensive experimentation setup with 27,042 whole-slide images from 6,155 patients is used to train thousands of models from the features of eight well-known foundation models for computational pathology. In addition to a substantial improvement in robustness, our approach improves classification accuracy by focusing on biologically relevant features. It mitigates robustness limitations of foundation models for computational pathology without retraining the foundation models themselves, enabling development of models that are more suitable in real-world clinical use.
Figures & tables
| Ahus | Aker | Gloucester | VICTOR | Training | QUASAR 2 | TransSCOT | ||||||||
| Patient count | 58 | 388 | 265 | 427 | 1138 | 1112 | 2856 | |||||||
| Age | ||||||||||||||
| Years | 71 | (64–77) | 70 | (60–77) | 68 | (62–75) | 64 | (57–71) | 67 | (59–75) | 65 | (59–71) | 65 | (58–70) |
| Sex | ||||||||||||||
| Female | 34 | (59%) | 195 | (50%) | 144 | (54%) | 149 | (35%) | 522 | (46%) | 474 | (43%) | 1126 | (39%) |
| Male | 24 | (41%) | 193 | (50%) | 121 | (46%) | 278 | (65%) | 616 | (54%) | 638 | (57%) | 1730 | (61%) |
| Aker | Ahus | Gloucester | VICTOR | Mainz | DENEB | Training | Dutch T1 | |||||||||
| Patient count | 69 | 243 | 444 | 282 | 135 | 61 | 1234 | 290 | ||||||||
| Age | ||||||||||||||||
| Years | 71 | (59–77) | 73 | (63–79) | 70 | (64–77) | 64 | (58–71) | 73 | (66–79) | 61 | (52–72) | 69 | (61–77) | 68 | (63–74) |
| Sex | ||||||||||||||||
| Female | 36 | (52%) | 128 | (53%) | 182 | (41%) | 102 | (36%) | 60 | (44%) | 31 | (51%) | 539 | (44%) | 126 | (43%) |
| Male | 33 | (48%) | 115 | (47%) | 262 | (59%) | 180 | (64%) | 75 | (56%) | 30 | (49%) | 695 | (56%) | 164 | (57%) |
| Model | No robustness loss | Embedding loss | Score loss | Both losses | ||||
| c-index | H-optimus-0 | 0.661 0.014 | 0.682 0.013 | (6%) | 0.689 0.005 | (7%) | 0.694 0.006 | (10%) |
| H-optimus-1 | 0.692 0.011 | 0.689 0.009 | (-2%) | 0.695 0.004 | (1%) | 0.699 0.005 | (2%) | |
| Hibou-L | 0.654 0.012 | 0.678 0.007 | (7%) | 0.672 0.005 | (6%) | 0.680 0.004 | (7%) | |
| Phikon-v2 | 0.654 0.017 | 0.668 0.014 | (4%) | 0.666 0.004 | (4%) | 0.678 0.006 | (7%) | |
| Prov-GigaPath | 0.662 0.013 | 0.676 0.008 | (1%) | 0.673 0.004 | (0%) | 0.680 0.007 | (5%) | |
| UNI | 0.681 0.010 | 0.680 0.012 | (0%) | 0.670 0.005 | (0%) | 0.683 0.005 | (0%) | |
| Model | No robustness loss | Embedding loss | Score loss | Both losses | ||||
| Balanced accuracy | H-optimus-0 | 0.611 0.014 | 0.608 0.014 | (0%) | 0.647 0.014 | (10%) | 0.620 0.019 | (2%) |
| H-optimus-1 | 0.631 0.009 | 0.638 0.011 | (5%) | 0.628 0.018 | (2%) | 0.612 0.026 | (-5%) | |
| Hibou-L | 0.661 0.013 | 0.655 0.013 | (2%) | 0.662 0.013 | (6%) | 0.688 0.016 | (8%) | |
| Phikon-v2 | 0.618 0.011 | 0.617 0.014 | (6%) | 0.625 0.015 | (8%) | 0.620 0.008 | (0%) | |
| Prov-GigaPath | 0.635 0.011 | 0.620 0.009 | (0%) | 0.654 0.011 | (10%) | 0.651 0.015 | (4%) | |
| UNI | 0.636 0.011 | 0.640 0.014 | (8%) | 0.653 0.014 | (14%) | 0.605 0.013 | (-8%) | |