用线性复杂度扫描视频,高效保留空间结构并加速训练。
$\text{PKS}^4$:Parallel Kinematic Selective State Space Scanners for Efficient Video Understanding

- 引入并行时序扫描模块,替代传统注意力机制
- 仅需20轮训练即收敛,计算量降低约10倍
- 适合追求高效训练的视频理解研究者
时序建模仍是视频理解的核心挑战,尤其在序列变长时。传统依赖密集时空注意力的模型面临二次计算开销。现有方法虽采用参数高效微调(PEFT)适配图像模型,但深层插入模块导致反向传播中激活内存过高。尽管高效状态空间模型(SSM)实现线性复杂度,却破坏二维空间关系,且需大量掩码预训练恢复空间感知。为此,我们提出并行运动选择性状态空间扫描器(PKS⁴)。保留标准2D视觉骨干以捕捉空间语义,插入单一线性复杂度的PKS⁴模块,避免时序注意力与多层适配器。通过运动先验编码器提取帧间相关性与差异,捕捉局部位移和运动边界,驱动线性复杂度的SSM追踪潜在运动状态,自适应调节每步更新速度与读写策略。不进行全局扫描,而是对每个空间位置并行部署时序扫描器,保持空间结构同时降低开销。在空间密集与时序密集的动作识别基准上,PKS⁴达到当前最优性能。尤为显著的是,方法仅需20个训练周期即可收敛,训练计算量约为纯视频SSM的1/10,确立了高效视频理解的新范式。
原文摘要 · Abstract (English)
Temporal modeling remains a fundamental challenge in video understanding, particularly as sequence lengths scale. Traditional video models relying on dense spatiotemporal attention suffer from quadratic computational costs for long videos. To circumvent these costs, recent approaches adapt image models for videos via Parameter-Efficient Fine-Tuning (PEFT) methods such as adapters. However, deeply inserting these modules incurs prohibitive activation memory overhead during back-propagation. While recent efficient State Space Models (SSMs) introduce linear complexity, they disrupt 2D spatial relationships and rely on extensive masked pre-training to recover spatial awareness. To overcome these limitations, we propose Parallel Kinematic Selective State Space Scanners (PKS$^4$). We retain a standard 2D vision backbone for spatial semantics and insert a single plug-and-play PKS$^4$ module with linear-complexity temporal scanning, avoiding temporal attention and multi-layer adapters. We first extract kinematic priors via a Kinematic Prior Encoder, which captures local displacements and motion boundaries through inter-frame correlations and differences. These priors drive linear-complexity SSMs to track underlying kinematic states, adaptively modulating update speeds and read-write strategies at each time step. Instead of global scanning, we deploy parallel scanners along the temporal dimension for each spatial location, preserving spatial structures while reducing overhead. Experiments on spatial-heavy and temporal-heavy action recognition benchmarks show that PKS$^4$ achieves state-of-the-art performance. Remarkably, our method converges in merely $20$ epochs, achieving approximately $10\times$ lower training compute than pure video SSMs, establishing a new paradigm for efficient video understanding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。