State Space Model for New-Generation Network Alternative to Transformers: A Survey

ML TAVR Mamba/SSM
2024年04月15日
在深度学习后期,Transformer架构已经展示了其在预训练大模型和各种下游任务中的强大性能。然而,这种架构的巨大计算需求阻碍了许多研究人员。为了进一步降低注意力模型的复杂度,已经做出了许多努力来设计更有效的方法。其中,状态空间模型(SSM)作为可能替代基于自注意力的Transformer模型的方法,在近年来越来越受到关注。在本文中,我们首次全面回顾了这些工作,并提供实验比较和分析,以更好地展示SSM的特点和优势。具体来说,我们首先详细描述了原理,以帮助读者快速掌握SSM的关键思想。之后,我们深入审查了现有SSM及其各种应用,包括自然语言处理、计算机视觉、图形、多模态和多媒体、点云/事件流、时间序列数据和其他领域。此外,我们对这些模型进行了统计比较和分析,希望读者能够了解不同结构在各种任务上的有效性。然后,我们提出了可能的研究方向,以更好地促进SSM的理论模型和应用的发展。更多相关工作将在以下GitHub上持续更新:https://github.com/Event-AHU/Mamba_State_Space_Model_Paper_List。
In the post-deep learning era, the Transformer architecture has demonstrated its powerful performance across pre-trained big models and various downstream tasks. However, the enormous computational demands of this architecture have deterred many researchers. To further reduce the complexity of attention models, numerous efforts have been made to design more efficient methods. Among them, the State Space Model (SSM), as a possible replacement for the self-attention based Transformer model, has drawn more and more attention in recent years. In this paper, we give the first comprehensive review of these works and also provide experimental comparisons and analysis to better demonstrate the features and advantages of SSM. Specifically, we first give a detailed description of principles to help the readers quickly capture the key ideas of SSM. After that, we dive into the reviews of existing SSMs and their various applications, including natural language processing, computer vision, graph, multi-modal and multi-media, point cloud/event stream, time series data, and other domains. In addition, we give statistical comparisons and analysis of these models and hope it helps the readers to understand the effectiveness of different structures on various tasks. Then, we propose possible research points in this direction to better promote the development of the theoretical model and application of SSM. More related works will be continuously updated on the following GitHub: https://github.com/Event-AHU/Mamba_State_Space_Model_Paper_List.
许愿