CL-MIL: Multi-Instance Learning on Whole-Slide Images with Self-Supervised Contrastive Learning
Pengfei Xiu, Dongying Liu, Sheng Wang, Hongliang Wang, Wei Wang, Shun Li · 2024
Histopathological examination of tissue slides is considered the gold standard for clinical cancer diagnosis. However, it heavily relies on pathologists' subjective judgment and involves substantial workload. Existing convolutional neural network (CNN)-based whole slide image (WSI) diagnostic methods suffer from high annotation costs and complex data acquisition processes. Therefore, this study proposes a weakly supervised multiple instance learning method based on self-supervised contrastive learning (CL-MIL) for WSI classification. By integrating self-supervised contrastive learning strategies, leveraging data augmentation through H&E stain separation, and introducing causal intervention via backdoor adjustment to mitigate dataset bias, our approach significantly enhances feature representation and classification performance in pathological images. Experimental results on the Camelyon16 and TCGA lung cancer datasets demonstrate that our method outperforms other comparable WSI classification models across various evaluation metrics, achieving higher diagnostic efficiency and accuracy. Additionally, our method exhibits good interpretability, enabling precise localization of pathological regions and providing robust technical support for clinical applications.