From Cells to Survival: Hierarchical Analysis of Cell Inter-Relations in Multiplex Microscopy for Lung Cancer Prognosis
Olle Edgren Schüllerqvist, Jens Baumann, Joakim Lindblad, Love Nordling, Artur Mezheyeuski, Patrick Micke, Nataša Sladoje
TL;DR
The paper addresses prognosis in NSCLC by leveraging the tumor microenvironment captured in multiplex imaging. It introduces HiGINE, a hierarchical graph isomorphism network with edge features that encodes local and global cell interactions, incorporating cancer stage as multimodal information. Across two public NSCLC datasets, HiGINE achieves superior risk stratification compared with baselines, with ablations showing benefits from edge weighting, hierarchical core modeling, and stage fusion. The approach emphasizes interpretability and robustness, offering a practical framework for prognosis based on spatial cell inter-relations in the TME.
Abstract
The tumor microenvironment (TME) has emerged as a promising source of prognostic biomarkers. To fully leverage its potential, analysis methods must capture complex interactions between different cell types. We propose HiGINE -- a hierarchical graph-based approach to predict patient survival (short vs. long) from TME characterization in multiplex immunofluorescence (mIF) images and enhance risk stratification in lung cancer. Our model encodes both local and global inter-relations in cell neighborhoods, incorporating information about cell types and morphology. Multimodal fusion, aggregating cancer stage with mIF-derived features, further boosts performance. We validate HiGINE on two public datasets, demonstrating improved risk stratification, robustness, and generalizability.
