arXiv:2410.03174cs.CV2024-10被引 1

用新型状态空间模型提升人体姿态估计的高分辨率表征效率

Efficient High-Resolution Visual Representation Learning with State Space Model for Human Pose Estimation

  • 设计动态视觉状态空间块,融合多尺度卷积与可变形操作增强局部感知
  • 在人体姿态估计等任务上达到领先性能,且计算量仅为ViT的1/4
  • 特别适合移动端等资源受限场景下的高精度视觉理解

在人体姿态估计等密集预测任务中,捕捉长程依赖关系并保持高分辨率视觉表示至关重要。视觉变换器(ViTs)虽通过自注意力机制实现全局建模,但其计算复杂度随标记数呈二次增长,限制了在高分辨率输入下的效率与可扩展性,尤其在移动和资源受限设备上表现不佳。状态空间模型(SSMs)如Mamba,通过线性复杂度实现全局感受野,具备高效可扩展的优势。然而现有视觉SSMs在应用于密集预测任务时存在三大问题:空间归纳偏置弱、隐藏状态衰减导致长程遗忘、输出分辨率低影响精细定位。为此,我们提出动态视觉状态空间(DVSS)模块,引入多尺度卷积以增强局部空间表示并强化空间归纳偏置;通过架构探索与理论分析,发现可变形操作能通过输入相关的自适应空间采样,有效提升语义聚合并缓解长程遗忘。我们将DVSS嵌入多分支高分辨率架构,构建了名为HRVMamba的新模型,用于高效高分辨率表征学习。在人体姿态估计、图像分类与语义分割上的大量实验表明,HRVMamba性能媲美主流的CNN、ViT及SSM基线模型。代码已开源:https://github.com/zhanghao5201/PoseVMamba。

原文摘要 · Abstract (English)

Capturing long-range dependencies while preserving high-resolution visual representations is crucial for dense prediction tasks such as human pose estimation. Vision Transformers (ViTs) have advanced global modeling through self-attention but suffer from quadratic computational complexity with respect to token count, limiting their efficiency and scalability to high-resolution inputs, especially on mobile and resource-constrained devices. State Space Models (SSMs), exemplified by Mamba, offer an efficient alternative by combining global receptive fields with linear computational complexity, enabling scalable and resource-friendly sequence modeling. However, when applied to dense prediction tasks, existing visual SSMs face key limitations: weak spatial inductive bias, long-range forgetting from hidden state decay, and low-resolution outputs that hinder fine-grained localization. To address these issues, we propose the Dynamic Visual State Space (DVSS) block, which augments visual state space models with multi-scale convolutional operations to enhance local spatial representations and strengthen spatial inductive biases. Through architectural exploration and theoretical analysis, we incorporate deformable operation into the DVSS block, identifying it as an efficient and effective mechanism to enhance semantic aggregation and mitigate long-range forgetting via input-dependent, adaptive spatial sampling. We embed DVSS into a multi-branch high-resolution architecture to build HRVMamba, a novel model for efficient high-resolution representation learning. Extensive experiments on human pose estimation, image classification, and semantic segmentation show that HRVMamba performs competitively against leading CNN-, ViT-, and SSM-based baselines. Code is available at https://github.com/zhanghao5201/PoseVMamba.

人体姿态估计状态空间模型高分辨率轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。