Hypergraph-based Multi-instance Contrastive Reinforcement Learning for Annotation-free Pan-cancer Survival Prediction on Whole Slide Histology Images
Cheng Lu,Chao Tang,Yanfen Cui,Jun Liu,Zhenhui Li,Xiuming Zhang,Suo Yao,Huan Lin,Dacheng Yang,Zhishun Liu,Wei Zhao,Shiwei Luo,Ke Zhao,Yun Zhu,Guangjun Yang,Lixu Yan,Shuting Chen,Xiangtian Zhao,Yingqiu Huo,Zhiyang Chen,Hongbo Liu,Jiahui Ma,Wenfeng He,Anant Madabhushi,Jinglei Tang,Zaiyi Liu
DOI: https://doi.org/10.21203/rs.3.rs-4918308/v1
2024-01-01
Abstract:The computational challenges inherent to using digital pathology for prognosis derive from the fact that a typical gigapixel slide may consist of thousands of image tiles and previous models lack the capability on modeling the crucial slide-level contextual information, thereby resulting in suboptimal performance. Considering the aforementioned challenges, we propose a Hypergraph-based Multi-instance Contrastive Reinforcement learning model (HeMiCoRe). The HeMiCoRe employs a novel hypergraph construction method that can encode local morphology and spatial relationship provided by the dynamic selected image tiles simultaneously. The HeMiCoRe used the cluster-restricted local and across-cluster global information of WSIs, with hypergraph-level contrastive learning for better generalization. To assess the efficacy of the HeMiCoRe, we conducted extensive internal and external validation experiments for survival prediction on 5,195 whole slide images from 4,624 patients across 10 major cancer types. The HeMiCoRe attains state-of-the-art performance on 8 cancer types and outstanding generalization performance. Furthermore, we demonstrate that the representative patch selector trained by reinforcement learning has the potential to exploit highly relevant image tiles for survival prediction. In summary, the HeMiCoRe is a hypergraph neural network-based model, with enhanced capabilities in capturing both the intra- and inter-cluster interaction mechanism, which makes it outperform existing weakly supervised classification models in pan-cancer survival prediction tasks. This model could be adapted in clinical practice for prognosticating cancer patients and facilitating clinical decision makings.