REVIEW 11 cited by
Graph-Mamba: Towards Long-Range Graph Sequence Modeling with Selective State Spaces
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
read the original abstract
Attention mechanisms have been widely used to capture long-range dependencies among nodes in Graph Transformers. Bottlenecked by the quadratic computational cost, attention mechanisms fail to scale in large graphs. Recent improvements in computational efficiency are mainly achieved by attention sparsification with random or heuristic-based graph subsampling, which falls short in data-dependent context reasoning. State space models (SSMs), such as Mamba, have gained prominence for their effectiveness and efficiency in modeling long-range dependencies in sequential data. However, adapting SSMs to non-sequential graph data presents a notable challenge. In this work, we introduce Graph-Mamba, the first attempt to enhance long-range context modeling in graph networks by integrating a Mamba block with the input-dependent node selection mechanism. Specifically, we formulate graph-centric node prioritization and permutation strategies to enhance context-aware reasoning, leading to a substantial improvement in predictive performance. Extensive experiments on ten benchmark datasets demonstrate that Graph-Mamba outperforms state-of-the-art methods in long-range graph prediction tasks, with a fraction of the computational cost in both FLOPs and GPU memory consumption. The code and models are publicly available at https://github.com/bowang-lab/Graph-Mamba.
Forward citations
Cited by 11 Pith papers
-
Benchmarking Sheaf Neural Networks for Inductive Tasks
On 14 inductive graph benchmarks, sheaf neural networks underperform strong GNN baselines, and their performance is driven more by the surrounding architecture than by the sheaf diffusion mechanism.
-
Navigating Hierarchy: Hyperbolic Learning on Brain Graphs for Disorder Diagnosis
A hyperbolic-space framework with geometric entailment constraints and a graph-aware Mamba module improves brain-network classification of ASD and MDD by explicitly modeling ROI-to-community-to-whole-brain hierarchy.
-
eMamba: Efficient Acceleration Framework for Mamba Models in Edge Computing
An end-to-end Mamba edge accelerator using hardware-friendly approximations, INT8 quantization, and NAS achieves 4.95x-5.62x lower latency and 1.63x-19.9x smaller models than ViT/CNN baselines.
-
TANGO: Graph Neural Dynamics via Learned Energy and Tangential Flows
TANGO adds a learnable energy gradient and an orthogonal tangential flow to GNN layers, improving long-range and heterophilic graph benchmarks.
-
Mamba-X: An End-to-End Vision Mamba Accelerator for Edge Computing Devices
A dedicated accelerator for Vision Mamba using a Kogge-Stone systolic scan array and hybrid 8-bit quantization achieves 2.3x end-to-end speedup and 11.5x energy-efficiency gain over an edge GPU with less than 1% top-1...
-
On Measuring Long-Range Interactions in Graph Neural Networks
The paper axiomatizes a distance-weighted influence measure of range and uses it to show that LRGB tasks differ sharply in how long-range they really are.
-
A Physics-Augmented GraphGPS Framework for the Reconstruction of 3D Riemann Problems from Sparse Data
A graph transformer with shock-aware attention and guided message-passing reconstructs 3D Riemann problem flows from 10% observed points more accurately than the benchmark ML models tested.
-
Message-Passing State-Space Models: Improving Graph Learning with Modern Sequence Modeling
A message-passing GNN based on a linear recurrence plus MLP readout achieves strong results on long-range, heterophilic, and spatio-temporal graph benchmarks.
-
From Layers to States: A State Space Model Perspective to Deep Neural Network Layer Dynamics
S6LA adds a selective state space recurrence across the layers of CNNs and vision transformers, giving consistent accuracy gains on ImageNet classification and COCO detection and segmentation.
-
HydraMamba: Multi-Head State Space Model for Global Point Cloud Learning
A state space model based point cloud network with shuffled Hilbert serialization, a convolutional bidirectional S6 branch, and multi-head S6 achieves new top scores on ModelNet40, ShapeNet, S3DIS, and ScanObjectNN.
-
MambaHash: Visual State Space Deep Hashing Model for Large-Scale Image Retrieval
MambaHash uses a grouped, multi-directional Mamba backbone for deep supervised hashing and reports the highest mean average precision on CIFAR-10 and IMAGENET, with marginal gains on NUS-WIDE.
Discussion (0). Continue with ORCID to comment.