2024/05/27 by Zhen Qin, Xuyang Shen, Qin, Zhen +11 · 1 citation
Computer Science · #Computation and Language (cs.CL) #Data Mining and Machine Learning Applications #FOS: Computer and information sciences
paper · pdf · doi:10.48550/arxiv.2405.17383
openalex publication_date 2024/05/27 · openalex created_date 2024/05/30 · openalex updated_date 2026/07/28
We present the Linear Complexity Sequence Model (LCSM), a comprehensive solution that unites various sequence modeling techniques with linear complexity, including linear attention, state space model, long convolution, and linear RNN, within a single framework. The goal is to enhance comprehension of these models by analyzing the impact of each component from a cohesive and streamlined viewpoint. Specifically, we segment the modeling processes of these models into three distinct stages: Expand, Oscillation, and Shrink (EOS), with each model having its own specific settings. The Expand stage involves projecting the input signal onto a high-dimensional memory state. This is followed by recursive operations performed on the memory state in the Oscillation stage. Finally, the memory state is projected back to a low-dimensional space in the Shrink stage. We perform comprehensive experiments to analyze the impact of different stage settings on language modeling and retrieval tasks. Our results show that data-driven methods are crucial for the effectiveness of the three stages in language modeling, whereas hand-crafted methods yield better performance in retrieval tasks.