Plug-and-Play Logit Fusion for Heterogeneous Pathology Foundation Models
Authors: Gexin Huang, Anqi Li, Yusheng Tan, Beidi Zhao, Gang Wang, Zu-Hua Gao, Xiaoxiao Li
Organizations: University of British Columbia, Vancouver, BC, Canada · Vector Institute, Toronto, ON, Canada · Washington University in St. Louis, St. Louis, MO, USA
Pathology foundation models (FMs) have become central to computational histopathology, offering strong transfer performance across a wide range of diagnostic and prognostic tasks. The rapid proliferation of pathology foundation models creates a model-selection bottleneck: no single model is uniformly best, yet exhaustively adapting and validating many candidates for each downstream endpoint is prohibitively expensive. We address this challenge with a lightweight and novel model fusion strategy, LogitProd, which treats independently trained FM-based predictors as fixed experts and learns sample-adaptive fusion weights over their slide-level outputs. The fusion operates purely on logits, requiring no encoder retraining and no feature-space alignment across heterogeneous backbones. We further provide a theoretical analysis showing that the optimal weighted product fusion is guaranteed to perform at least as well as the best individual expert under the training objective. We systematically evaluate LogitProd on 22 benchmarks spanning WSI-level classification, tile-level classification, gene mutation prediction, and discrete-time survival modeling. LogitProd ranks first on 20/22 tasks and improves the average performance across all tasks by ~3% over the strongest single expert. LogitProd enables practitioners to upgrade heterogeneous FM-based pipelines in a plug-and-play manner, achieving multi-expert gains with ∼12× lower training cost than feature-fusion alternatives.
Figures & tables
Figure 1: Overview of LogitProd. Frozen FM experts output logits; LogitProd derives confidence/entropy/disagreement cues to predict sample-adaptive weights and fuses experts via weighted product fusion, enabling efficient multi-task prediction without re-encoding or feature alignment.
Figure 2: Evaluation across 22 pathology tasks. a–f, Gene mutation prediction (mAUC): a , mean across five genes; b–f , per-gene performance with prevalence. g–i, TIL classification (AUC) across six datasets. j, WSI-level tumour diagnosis (AUC). k, Breast carcinoma subtyping (AUC). l–q, C-index distributions across six TCGA cohorts for all FM-based experts and LogitProd. Box plots summarize cross-validation folds. r, Task-wide rank heatmap. The rightmost column reports mean rank, and dashed lines separate task groups.
School of Computer Science and Engineering, South China University of Technology, China · School of Computing and Mathematical Sciences, University of Leicester, UK · Leicester Cancer Research Centre, University of Leicester, UK +4
Institute for Predictive Deep Learning for Medicine and Healthcare, Giessen University, Germany · Department of Medical Informatics, University Medicine Gottingen, Germany · Department of Applied Mathematics and Computer Science, Technical University of Denmark, Denmark +8
The Alan Turing Institute, London, United Kingdom · University of Manchester, Manchester, United Kingdom · The Institute of Cancer Research, London, United Kingdom +1