Knowledge-enhanced Pretraining for Vision-language Pathology Foundation Model on Cancer Diagnosis
About
Vision-language foundation models have shown great promise in computational pathology but remain primarily data-driven, lacking explicit integration of medical knowledge. We introduce KEEP (KnowledgE-Enhanced Pathology), a foundation model that systematically incorporates disease knowledge into pretraining for cancer diagnosis. KEEP leverages a comprehensive disease knowledge graph encompassing 11,454 diseases and 139,143 attributes to reorganize millions of pathology image-text pairs into 143,000 semantically structured groups aligned with disease ontology hierarchies. This knowledge-enhanced pretraining aligns visual and textual representations within hierarchical semantic spaces, enabling deeper understanding of disease relationships and morphological patterns. Across 18 public benchmarks (over 14,000 whole-slide images) and 4 institutional rare cancer datasets (926 cases), KEEP consistently outperformed existing foundation models, showing substantial gains for rare subtypes. These results establish knowledge-enhanced vision-language modeling as a powerful paradigm for advancing computational pathology.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | PCam (test) | Accuracy65.68 | 97 | |
| Image Classification | BACH (test) | Top-1 Acc79.17 | 59 | |
| Pathology Image Classification | BreakHis (test) | Top-1 Accuracy39.19 | 46 | |
| Pathology Image Classification | SICAP-MIL (test) | Top-1 Accuracy63.86 | 37 | |
| Pathology Image Classification | SkinCancer (test) | Top-1 Accuracy81.38 | 37 | |
| Pathology Image Classification | SKINTUMOR (test) | Top-1 Accuracy91.21 | 37 | |
| Pathology Image Classification | LC-LUNG (test) | Top-1 Accuracy98.11 | 37 | |
| Pathology Image Classification | LC-COLON (test) | Top-1 Accuracy100 | 37 | |
| Pathology Image Classification | NCT-CRC (test) | Top-1 Accuracy96.09 | 37 | |
| Pathology Image Classification | PanNuke (test) | Top-1 Accuracy89.62 | 37 |