Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Xiao Zhou, Luoyi Sun, Dexuan He, Wenbin Guan, Ge Wang, Ruifen Wang, Lifeng Wang, Xiaojun Yuan, Xin Sun, Ya Zhang, Kun Sun, Yanfeng Wang, Weidi Xie• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationPCam (test)
Accuracy65.68
97
Image ClassificationBACH (test)
Top-1 Acc79.17
59
Pathology Image ClassificationBreakHis (test)
Top-1 Accuracy39.19
46
Pathology Image ClassificationSICAP-MIL (test)
Top-1 Accuracy63.86
37
Pathology Image ClassificationSkinCancer (test)
Top-1 Accuracy81.38
37
Pathology Image ClassificationSKINTUMOR (test)
Top-1 Accuracy91.21
37
Pathology Image ClassificationLC-LUNG (test)
Top-1 Accuracy98.11
37
Pathology Image ClassificationLC-COLON (test)
Top-1 Accuracy100
37
Pathology Image ClassificationNCT-CRC (test)
Top-1 Accuracy96.09
37
Pathology Image ClassificationPanNuke (test)
Top-1 Accuracy89.62
37
Showing 10 of 13 rows

Other info

Follow for update