Search papers, labs, and topics across Lattice.
The paper introduces HGP-Mamba, a Mamba-based multimodal framework for cancer survival risk prediction that integrates histological data with protein features generated from Whole Slide Images (WSIs) using a protein feature extractor (PFE) based on pretrained foundation models. HGP-Mamba incorporates Local Interaction-aware Mamba (LiAM) for fine-grained feature interaction and Global Interaction-enhanced Mamba (GiEM) to enhance modality fusion. Experiments on four cancer datasets demonstrate HGP-Mamba achieves state-of-the-art performance and computational efficiency.
Mamba, typically used for sequence modeling, now powers state-of-the-art cancer survival risk prediction by fusing histological data with AI-generated protein features, outperforming existing methods.
Recent advances in multimodal learning have significantly improved cancer survival risk prediction. However, the joint prognostic potential of protein markers and histopathology images remains underexplored, largely due to the high cost and limited availability of protein expression profiling. To address this challenge, we propose HGP-Mamba, a Mamba-based multimodal framework that efficiently integrates histological with generated protein features for survival risk prediction. Specifically, we introduce a protein feature extractor (PFE) that leverages pretrained foundation models to derive high-throughput protein embeddings directly from Whole Slide Images (WSIs), enabling data-efficient incorporation of molecular information. Together with histology embeddings that capture morphological patterns, we further introduce the Local Interaction-aware Mamba (LiAM) for fine-grained feature interaction and the Global Interaction-enhanced Mamba (GiEM) to promote holistic modality fusion at the slide level, thus capture complex cross-modal dependencies. Experiments on four public cancer datasets demonstrate that HGP-Mamba achieves state-of-the-art performance while maintaining superior computational efficiency compared with existing methods. Our source code is publicly available at <a href="https://github.com/Daijing-ai/HGP-Mamba.git">this https URL</a>.
