LLM2D
Surface Vision Mamba:利用双向状态空间模型进行高效的球面流形表示
Surface Vision Mamba: Leveraging Bidirectional State Space Model for Efficient Spherical Manifold Representation
作者: Rongzhao He, Weihao Zheng, Leilei Zhao, Ying Wang, Dalin Zhu, Dan Wu, Bin Hu
发布日期: 2/13/2025
arXiv ID: oai:arXiv.org:2501.14679v3

摘要

arXiv:2501.14679v3 通知类型: 替换-交叉 摘要:基于注意力的方法在球面皮层表面建模长程依赖关系方面表现出色,超越了传统的几何深度学习(GDL)模型。然而,它们广泛的推理时间和高内存需求为有限计算资源的大数据集应用带来了挑战。受到计算机视觉中状态空间模型的启发,我们引入了无注意力的Vision Mamba(Vim)到球面上,提出了一种适用于分析球面流形上数据的领域无关架构。我们的方法通过将球面数据表示为从细分icosphere导出的三角形碎片序列来实现表面拼接。所提出的Surface Vision Mamba(SiM)在使用新生儿大脑皮层表面度量值进行多个神经发育表型回归任务中进行了评估。实验结果表明,SiM 在 Ico-4 网格划分下相比 Surface Vision Transformer(SiT)实现了 4.8 倍的更快推理时间和 91.7%更低的内存消耗。进一步的敏感性分析进一步突显了 SiM 在识别细微的神经发育模式方面的潜力。代码可在 https://github.com/Rongzhao-He/surface-vision-mamba 获取。