Combining Graph Neural Network and Mamba to Capture Local and Global Tissue Spatial Relationships in Whole Slide Images

Ruiwen Ding,Kha-Dinh Luong,Erika Rodriguez,Ana Cristina Araujo Lemos da Silva,William Hsu
2024-06-06
Abstract:In computational pathology, extracting spatial features from gigapixel whole slide images (WSIs) is a fundamental task, but due to their large size, WSIs are typically segmented into smaller tiles. A critical aspect of this analysis is aggregating information from these tiles to make predictions at the WSI level. We introduce a model that combines a message-passing graph neural network (GNN) with a state space model (Mamba) to capture both local and global spatial relationships among the tiles in WSIs. The model's effectiveness was demonstrated in predicting progression-free survival among patients with early-stage lung adenocarcinomas (LUAD). We compared the model with other state-of-the-art methods for tile-level information aggregation in WSIs, including tile-level information summary statistics-based aggregation, multiple instance learning (MIL)-based aggregation, GNN-based aggregation, and GNN-transformer-based aggregation. Additional experiments showed the impact of different types of node features and different tile sampling strategies on the model performance. This work can be easily extended to any WSI-based analysis. Code: <a class="link-external link-https" href="https://github.com/rina-ding/gat-mamba" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Machine Learning
What problem does this paper attempt to address?
This paper attempts to address the problem of how to extract spatial features from Whole Slide Images (WSIs) in computational pathology and effectively integrate these features when predicting the Progression-Free Survival (PFS) of early-stage non-small cell lung cancer (NSCLC) patients. ### Specific Problems: 1. **Extracting Spatial Features**: Due to the large size of WSIs, they are usually divided into smaller tiles. How to extract and integrate spatial features from these tiles to make predictions at the WSI level. 2. **Integrating Local and Global Relationships**: Existing methods often capture only local spatial relationships when integrating tile information, failing to capture global long-distance dependencies. How to combine local and global spatial relationships to improve prediction accuracy. 3. **Improving Prediction Performance**: How to surpass existing methods in predicting the Progression-Free Survival of early-stage lung adenocarcinoma (LUAD) patients by combining Graph Neural Network (GNN) and State Space Model (Mamba). ### Background: - **Non-Small Cell Lung Cancer (NSCLC)**: One of the most common types of lung cancer. Early-stage patients usually undergo radical surgery, but about 30-55% of patients experience disease recurrence within 5 years post-surgery. - **Current Clinical Standard**: The TNM staging system has limitations in assessing tumor size, local invasion, lymph node metastasis, and distant metastasis, making it ineffective for risk stratification. - **Quantitative Tissue Morphological Features**: Recent studies have shown that quantitative tissue morphological features extracted from H&E-stained WSIs can be used to predict recurrence or survival in early-stage NSCLC patients. ### Methodological Innovations: - **Combining GNN and Mamba**: The paper proposes a method that combines message-passing GNN and Mamba state space model to capture both local and global spatial relationships between tiles in WSIs. - **Systematic Exploration of Node Features and Sampling Strategies**: Investigated the impact of different node features and tile sampling strategies on model performance. - **Experimental Validation**: Extensive experiments were conducted using two public datasets to compare the performance of the proposed method with existing baseline models. ### Objectives: - **Improve Prediction Accuracy**: Enhance the accuracy of predicting the Progression-Free Survival of early-stage LUAD patients by combining local and global spatial relationships. - **Expand Application Scope**: The method can be easily extended to other WSI-based analysis tasks.