arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.03430eess.IVcs.CV

计算高效时代:医学图像分析中状态空间模型的综合综述

Computation-Efficient Era: A Comprehensive Survey of State Space Models in Medical Image Analysis

School of Biomedical Engineering(生物医学工程学院);University of British Columbia(不列颠哥伦比亚大学);Tarbiat Modares University(塔比亚特·莫达雷斯大学);University of Regensburg(雷根斯堡大学);South Dakota State University(南达科他州立大学);RWTH Aachen University(亚琛工业大学);Lunit Inc.(Lunit公司);Northwestern University(西北大学);Fraunhofer Institute for Digital Medicine(弗劳恩霍夫数字医学研究院)

Moein Heidari, Sina Ghorbani Kolahi, Sanaz Karimijafarbigloo, Bobby Azad, Afshin Bozorgpour, Soheila Hatami, Reza Azad, Ali Diba, Ulas Bagci, Dorit Merhof, Ilker Hacihaliloglu

AI总结:

本文系统综述了以 Mamba 为代表的状态空间模型在医学图像分析中的应用,涵盖理论基础、分类体系、关键挑战与未来方向,旨在帮助研究者把握这一计算高效的新范式。

摘要:

序列建模在各个领域都发挥着至关重要的作用,历史上循环神经网络是执行这些任务的主要方法。然而,Transformer 的出现因其卓越的性能改变了这一范式。在这些进展的基础上,Transformer 已与 CNN 并列为学习视觉表示的两大基础模型。然而,Transformer 受到其注意力机制 $\mathcal{O}(N^2)$ 复杂度的制约,而 CNN 缺乏全局感受野和动态权重分配能力。状态空间模型(State Space Models, SSMs),特别是具有选择机制和硬件感知架构的 \textit{\textbf{Mamba}} 模型,近来在序列建模和视觉表示学习领域引起了极大的关注,它通过提供无限上下文长度并在输入序列上保持线性复杂度而具备显著效率,从而挑战了 Transformer 的主导地位。借助计算机视觉领域的进展,医学成像已迎来 Mamba 模型的新纪元。为了帮助研究人员应对这一浪潮,本综述旨在提供一份关于医学成像中 Mamba 模型的百科全书式回顾。具体而言,我们首先进行全面的理论回顾,奠定 SSMs 的基础,包括 Mamba 架构及其在此背景下用于序列建模范式的替代方案。接着,我们对医学领域的 Mamba 模型进行了结构化分类,并基于其应用、成像模态和目标器官引入了一种多样化的分类方案。最后,我们总结了关键挑战,讨论了 SSMs 在医学领域的不同未来研究方向,并提出了若干方向以满足该领域的需求。此外,我们已将本文所讨论的研究及其开源实现整理在我们的 GitHub 仓库中。

英文摘要:

Sequence modeling plays a vital role across various domains, with recurrent neural networks being historically the predominant method of performing these tasks. However, the emergence of transformers has altered this paradigm due to their superior performance. Built upon these advances, transformers have conjoined CNNs as two leading foundational models for learning visual representations. However, transformers are hindered by the $\mathcal{O}(N^2)$ complexity of their attention mechanisms, while CNNs lack global receptive fields and dynamic weight allocation. State Space Models (SSMs), specifically the \textit{\textbf{Mamba}} model with selection mechanisms and hardware-aware architecture, have garnered immense interest lately in sequential modeling and visual representation learning, challenging the dominance of transformers by providing infinite context lengths and offering substantial efficiency maintaining linear complexity in the input sequence. Capitalizing on the advances in computer vision, medical imaging has heralded a new epoch with Mamba models. Intending to help researchers navigate the surge, this survey seeks to offer an encyclopedic review of Mamba models in medical imaging. Specifically, we start with a comprehensive theoretical review forming the basis of SSMs, including Mamba architecture and its alternatives for sequence modeling paradigms in this context. Next, we offer a structured classification of Mamba models in the medical field and introduce a diverse categorization scheme based on their application, imaging modalities, and targeted organs. Finally, we summarize key challenges, discuss different future research directions of the SSMs in the medical domain, and propose several directions to fulfill the demands of this field. In addition, we have compiled the studies discussed in this paper along with their open-source implementations on our GitHub repository.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑