arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.03174cs.CV

基于状态空间模型的高效高分辨率视觉表征学习用于人体姿态估计

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

National Key Laboratory of Human-Machine Hybrid Augmented Intelligence(人机混合增强智能国家级重点实验室);National Engineering Research Center for Visual Information and Applications(视觉信息与应用国家工程研究中心);Institute of Artificial Intelligence and Robotics(人工智能与机器人研究院);Xi’an Jiaotong University(西安交通大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Hao Zhang, Yongqiang Ma, Wenqi Shao, Ping Luo, Nanning Zheng, Kaipeng Zhang

AI总结:

针对现有视觉状态空间模型在密集预测中空间归纳偏置弱和长距离遗忘等问题,提出DVSS模块结合多尺度卷积与可变形操作,并构建HRVMamba模型实现高效高分辨率表征学习,在人体姿态估计等任务上表现优异。

摘要:

捕获长距离依赖同时保留高分辨率视觉表征对于人体姿态估计等密集预测任务至关重要。Vision Transformers (ViTs) 通过自注意力机制推进了全局建模,但其在 token 数量上存在二次计算复杂度,限制了其对高分辨率输入的效率和可扩展性,尤其是在移动和资源受限设备上。以 Mamba 为代表的 State Space Models (SSMs) 通过将全局感受野与线性计算复杂度相结合,提供了一种高效的替代方案,实现了可扩展且资源友好的序列建模。然而,应用于密集预测任务时,现有的视觉 SSMs 面临关键局限:空间归纳偏置弱、隐藏状态衰减导致的长距离遗忘,以及阻碍细粒度定位的低分辨率输出。为解决这些问题,我们提出了 Dynamic Visual State Space (DVSS) 模块,通过多尺度卷积操作增强视觉状态空间模型,以提升局部空间表征并强化空间归纳偏置。通过架构探索和理论分析,我们将可变形操作引入 DVSS 模块,指出其作为一种高效机制,能通过依赖输入的自适应空间采样来增强语义聚合并缓解长距离遗忘。我们将 DVSS 嵌入多分支高分辨率架构,构建了 HRVMamba,一种用于高效高分辨率表征学习的新型模型。在人体姿态估计、图像分类和语义分割上的大量实验表明,HRVMamba 与领先的基于 CNN、ViT 和 SSM 的基线模型相比表现出竞争力。代码可在 https://github.com/zhanghao5201/PoseVMamba 获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑