paper-with-me

Papers

ParallelFlow: Parallelizing Linear Transformers via Flow Discretization

2025-04-01 · Nicola Muca Cirone, Cristopher Salvi

We present a theoretical framework for analyzing linear attention models through matrix-valued state space models (SSMs). Our approach, Parallel Flows, provides a perspective that systematically decouples temporal dynamics from implementation constraints, enabling independent analysis of critical algorithmic components: chunking, parallelization, and information aggregation. Central to this framework is the reinterpretation of chunking procedures as computations of the flows governing system dynamics. This connection establishes a bridge to mathematical tools from rough path theory, opening the door to new insights into sequence modeling architectures. As a concrete application, we analyze DeltaNet in a generalized low-rank setting motivated by recent theoretical advances. Our methods allow us to design simple, streamlined generalizations of hardware-efficient algorithms present in the literature, and to provide completely different ones, inspired by rough paths techniques, with provably lower complexity. This dual contribution demonstrates how principled theoretical analysis can both explain existing practical methods and inspire fundamentally new computational approaches.

📄 PDF Abstract BibTeX arXiv:2504.00492

Code (0)

등록된 구현이 없습니다.

Tasks

ChunkingState Space Models

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Attention 설명 없음

Similar Papers 제목 키워드 기반

Parallelizing Linear Transformers with the Delta Rule over Sequence Length

2024-06-10 · Songlin Yang, Bailin Wang, Yu Zhang, Yikang Shen 외

Transformers with linear attention (i.e., linear transformers) and state-space models have recently been suggested as a viable linear-time alternative to transformers with softmax attention. However, these models still u…

Language ModelingLanguage ModellingMambaState Space Models

Function graph transformers universally approximate operators between function spaces

2026-05-18 · Takashi Furuya, David Mis, Ivan Dokmanić, Maarten V. de Hoop 외 arxiv

We study the approximation of nonlinear operators between function spaces by transformers. Our approach is to lift functions to measures supported on their graphs and leverage a recently introduced measure-theoretic view…

Why Are Linear RNNs More Parallelizable?

2026-03-04 · William Merrill, Hongjian Jiang, Yanhong Li, Anthony Lin 외 arxiv

The community is increasingly exploring linear RNNs (LRNNs) as language models, motivated by their expressive power and parallelizability. While prior work establishes the expressivity benefits of LRNNs over transformers…

RWKV: Reinventing RNNs for the Transformer Era

2023-05-22 · Bo Peng, Eric Alcaide, Quentin Anthony, Alon Albalak 외

Transformers have revolutionized almost all natural language processing (NLP) tasks but suffer from memory and computational complexity that scales quadratically with sequence length. In contrast, recurrent neural networ…

Computational EfficiencyNatural Language Inference

Exact Flow Linear Attention: Exact Solution from Continuous-Time Dynamics

2025-12-14 · Jingdi Lei, Di Zhang, Soujanya Poria arxiv

In this paper, we introduce Exact Flow Linear Attention~(EFLA), an exact-flow formulation of delta-rule linear attention. We show that the delta-rule update can be interpreted as an explicit Euler discretization of an un…