HyperAIHyperAI

Command Palette

Search for a command to run...

HRFormer:用于密集预测的高分辨率Transformer

Yuhui Yuan Rao Fu Lang Huang Weihong Lin Chao Zhang Xilin Chen Jingdong Wang

摘要

我们提出了一种高分辨率Transformer(HRFormer),用于学习密集预测任务中的高分辨率表征,这与原始视觉Transformer(Vision Transformer)生成低分辨率表征且具有高内存和计算开销的特点形成对比。我们借鉴了高分辨率卷积网络(HRNet)中提出的多分辨率并行结构,并结合局部窗口自注意力机制(local-window self-attention),该机制在不重叠的小图像窗口上执行自注意力操作,从而提升内存与计算效率。此外,我们在前馈网络(FFN)中引入卷积操作,以在彼此分离的图像窗口之间实现信息交互。实验表明,HRFormer在人体姿态估计与语义分割任务上均表现出色。例如,在COCO姿态估计任务中,HRFormer以50%更少的参数量和30%更少的浮点运算次数(FLOPs),实现了比Swin Transformer高出1.3 AP的性能。代码已开源,地址为:https://github.com/HRNet/HRFormer


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供