摘要
arXiv:2501.14679v5 Announce Type: replace-cross
摘要:基于注意力的方法在球形皮层表面建模长程依赖关系方面表现出色,超越了传统的几何深度学习(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 获取。