arXiv:2606.02602cs.LGcs.CV2026-06

提出拓扑感知排序的图Mamba模型,提升病理切片生存分析精度与效率。

Graph Mamba Survival Analysis Based on Topology-Aware ordering

论文配图:Graph Mamba Survival Analysis Based on Topology-Aware ordering
图 1 · 摘自论文原文
  • 基于拓扑连通性设计节点排序,增强Mamba对图结构的敏感性
  • 在5个TCGA数据集上实现更高生存预测准确率,计算复杂度线性降低
  • 引入双向Mamba与GCN融合,兼顾空间结构建模与长程依赖捕捉

在计算病理学中,全切片图像(WSIs)的生存分析对患者预后评估至关重要,但面临多重技术挑战。尽管Transformer通过自注意力机制捕捉长距离依赖,其$O(N^2)$的时间复杂度在大规模WSI图结构中造成严重计算瓶颈。Mamba模型以线性复杂度突破该瓶颈,但其对输入顺序高度敏感,传统基于节点度或子图大小的排序方法未能充分考虑图数据的拓扑连通性,限制了其序列建模性能。此外,其单向架构无法利用图像的双向空间结构。为此,本文提出一种基于拓扑感知排序(TopoMamSurv)的图Mamba生存分析框架,以适配Mamba的顺序敏感特性。可视化实验进一步证实,通过拓扑感知排序(TAO)提取的节点具有更高相似性。我们还设计了双向Mamba模块,并集成图卷积网络(GCN),实现图像的双向空间上下文建模,构建‘局部聚合-全局捕获’的层次化特征学习架构。该框架通过系统性设计拓扑感知排序、双向语义建模与层次特征融合,有效调和了长程依赖建模、计算效率与空间结构利用之间的矛盾。该框架已在五个TCGA数据集上验证其综合性能优势。

原文摘要 · Abstract (English)

In computational pathology, Whole Slide Images (WSIs) survival analysis is crucial for patient prognosis assessment, but it faces multiple technical challenges. Although the Transformer captures long-range dependencies through its self-attention mechanism, its $O(N^2)$ time complexity causes a severe computational bottleneck in large-scale WSIs graph structures. The Mamba model breaks through the Transformer's computational bottleneck with linear complexity. But, owing to Mamba's high sensitivity to the order of input data, traditional node sorting methods in Graph Mamba, such as those based on node degree or subgraph size, fail to adequately account for the topological connectivity of graph data. This inadequacy consequently restricts the performance of Mamba's sequential modeling. Moreover, its unidirectional architecture cannot leverage the bidirectional spatial structure of images. To address these challenges, this paper proposes a novel Graph Mamba survival analysis framework based on topology-aware ordering (TopoMamSurv) to adapt to the sequential sensitivity of Mamba. Our visualization experiments further confirmed that the nodes extracted through the topology-aware ordering (TAO) strategy indeed exhibit higher similarity. Furthermore, we designed a bidirectional Mamba module and integrated a Graph Convolutional Network (GCN) to achieve bidirectional spatial context modeling of images, forming a hierarchical feature learning architecture for "local aggregation - global capture." This framework effectively reconciles the contradiction between long-range dependency modeling, computational efficiency, and spatial structure utilization in WSIs analysis through its systematic design of TAO, bidirectional semantic modeling, and hierarchical feature fusion. This framework has been validated for its comprehensive performance advantage on five TCGA datasets.

生存分析图神经网络Mamba病理图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。