Bi-level Graphs for Cellular Pattern Discovery

Published: 28 Oct 2023, Last Modified: 16 Dec 2023NeurIPS2023-AI4Science PosterEveryoneRevisionsBibTeX
Keywords: tumor microenvironment, single-cell, prognosis, spatial analysis, graph learning, graph kernel, breast cancer
TL;DR: Bi-level Graph Learning Unveil Prognosis-Relevant Tumor Microenvironment Patterns
Abstract: The tumor microenvironment is widely recognized for its central role in driving cancer progression and influencing prognostic outcomes. Despite extensive research efforts dedicated to characterizing this complex and heterogeneous environment, considerable challenges persist. In this study, we introduce a novel data-driven approach for identifying tumor microenvironment patterns that, we show, are closely tied to patient prognoses. Our methodology relies on the construction of a bi-level graph model: (i) a cellular graph, which models the intricate tumor microenvironments, and (ii) a population graph that captures inter-patient similarities, given their respective cellular graphs, by means of a soft Weisfeiler-Lehman kernel. This systematic integration of information across different scales enables us to identify patient subgroups exhibiting unique prognoses while unveiling certain tumor microenvironment patterns that characterize them. We demonstrate our approach in a cohort of breast cancer patients, and identify crucial tumor microenvironment patterns associated with patient prognosis. Our study provides valuable insights into the prognostic implications of the breast tumor microenvironment, and this methodology holds the potential to analyze other cancers.
Submission Track: Original Research
Submission Number: 101