Survival Modeling from Whole Slide Images via Patch-Level Graph Clustering and Mixture Density Experts
Ardhendu Sekhar, Vasu Soni, Keshav Aske, Garima Jain, Pranav Jeevan, Amit Sethi
TL;DR
This work tackles cancer survival prediction from WSIs by introducing a four-module framework that jointly addresses prognostic tissue heterogeneity and spatial coherence. It combines Quantile-Based Patch Filtering, Graph-Regularized Patch Clustering, Hierarchical Feature Aggregation, and Expert-Guided Mixture Density Modeling to estimate complex, multi-modal survival distributions with interpretable components. Across TCGA LUAD, KIRC, and BRCA cohorts, the method achieves state-of-the-art time-dependent concordance indices ($\text{TDC} = 0.653$, $0.719$, $0.733$ respectively), demonstrating strong discrimination and calibration from histopathology alone. The approach emphasizes interpretability through cluster-aware representations and a multi-expert density model, offering practical potential for prognosis and risk stratification in personalized oncology.
Abstract
We propose a modular framework for predicting cancer specific survival directly from whole slide pathology images (WSIs). The framework consists of four key stages designed to capture prognostic and morphological heterogeneity. First, a Quantile Based Patch Filtering module selects prognostically informative tissue regions through quantile thresholding. Second, Graph Regularized Patch Clustering models phenotype level variations using a k nearest neighbor graph that enforces spatial and morphological coherence. Third, Hierarchical Feature Aggregation learns both intra and inter cluster dependencies to represent multiscale tumor organization. Finally, an Expert Guided Mixture Density Model estimates complex survival distributions via Gaussian mixtures, enabling fine grained risk prediction. Evaluated on TCGA LUAD, TCGA KIRC, and TCGA BRCA cohorts, our model achieves concordance indices of 0.653 ,0.719 ,and 0.733 respectively, surpassing existing state of the art approaches in survival prediction from WSIs.
