KMeansGraphMIL: A Weakly Supervised Multiple Instance Learning Model for Predicting Colorectal Cancer Tumor Mutational Burden.

Abstract

Colorectal cancer (CRC) is one of the top three most lethal malignancies worldwide, posing a significant threat to human health. Recently proposed immunotherapy checkpoint blockade treatments have proven effective for CRC, but their use depends on measuring specific biomarkers in patients. Among these biomarkers, Tumor Mutational Burden (TMB) has emerged as a novel indicator, traditionally requiring Next-Generation Sequencing (NGS) for measurement, which is time-consuming, labor-intensive, and costly. To provide an economical and rapid way to predict patients' TMB, we propose the KMeansGraphMIL model based on weakly supervised multiple-instance learning (WSMIL). Compared to previous WSMIL models, KMeansGraphMIL leverages both the similarity of image patch feature vectors and the spatial relationships between patches. This approach improves the model's AUC to 0.8334 and significantly increases the recall to 0.7556. Thus, we present an economical and rapid framework for predicting CRC TMB, offering the potential for doctors to quickly develop treatment plans and saving patients substantial time and money.