新闻中心

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

2024-01-22
浏览次数:
返回列表

Transformer在大模型领域的地位无可撼动。然而,随着模型规模的扩展和序列长度的增加,传统的Transformer架构的局限性开始凸显。幸运的是,Mamba的问世正在迅速改变这一现状。它出色的性能立即引起了AI界的轰动。Mamba的出现为大规模模型的训练和序列处理带来了巨大的突破。它的优势在AI界迅速蔓延,为未来的研究和应用带来了巨大的希望。

上周四, Vision Mamba(Vim)的提出已经展现了它成为视觉基础模型的下一代骨干的巨大潜力。仅隔一天,中国科学院、华为、鹏城实验室的研究人员提出了 VMamba:一种具有全局感受野、线性复杂度的视觉 Mamba 模型。这项工作标志着视觉 Mamba 模型 Swin 时刻的来临。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba


  • 论文标题:VMamba: Visual State Space Model
  • 论文地址: https://arxiv.org/abs/2401.10166
  • 代码地址: https://github.com/MzeroMiko/VMamba

CNN 和视觉 Transformer(ViT)是当前最主流的两类基础视觉模型。尽管 CNN 具有线性复杂度,ViT 具有更为强大的数据拟合能力,然而代价是计算复杂较高。研究者认为 ViT 之所以拟合能力强,是因为其具有全局感受野和动态权重。受 Mamba 模型的启发,研究者设计出一种在线性复杂度下同时具有这两种优秀性质的模型,即 Visual State Space Model(VMamba)。大量的实验证明,VMamba 在各种视觉任务中表现卓越。如下图所示,VMamba-S 在 ImageNet-1K 上达到 83.5% 的正确率,比 Vim-S 高 3.2%,比 Swin-S 高 0.5%。

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

方法介绍

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

VMamba 的成功关键在于采用了 S6 模型,这个模型最初是为了解决自然语言处理(NLP)任务而设计的。与 ViT 的注意力机制不同,S6 模型通过将 1D 向量中的每个元素与之前的扫描信息进行交互,有效地将二次复杂度降低为线性。这种交互方式使得 VMamba 在处理大规模数据时更加高效。因此,S6 模型的引入为 VMamba 的成功打下了坚实的基础。

然而,由于视觉信号(如图像)不像文本序列那样具有天然的有序性,因此无法在视觉信号上简单地对 S6 中的数据扫描方法进行直接应用。为此研究者设计了 Cross-Scan 扫描机制。Cross-Scan 模块(CSM)采用四向扫描策略,即从特征图的四个角同时扫描(见上图)。该策略确保特征中的每个元素都以不同方向从所有其他位置整合信息,从而形*局感受野,又不增加线性计算复杂度。

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

在 CSM 的基础上,作者设计了 2D-selective-scan(SS2D)模块。如上图所示,SS2D 包含了三个步骤:

  • scan expand 将一个 2D 特征沿 4 个不同方向(左上、右下、左下、右上)展平为 1D 向量。
  • S6 block 独立地将上步得到的 4 个 1D 向量送入 S6 操作。
  • scan merge 将得到的 4 个 1D 向量融合为一个 2D 特征输出。

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

上图为本文提出的 VMamba 结构图。VMamba 的整体框架与主流的视觉模型类似,其主要区别在于基本模块(VSS block)中采用的算子不同。VSS block 采用了上述介绍的 2D-selective-scan 操作,即 SS2D。SS2D 保证了 VMamba 在线性复杂度的代价下实现全局感受野。

实验结果

imagenet 分类

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

通过对比实验结果不难看出,在相似的参数量和 FLOPs 下:

  • VMamba-T 取得了 82.2% 的性能,超过 RegNetY-4G 达 2.2%、DeiT-S 达 2.4%、Swin-T 达 0.9%。
  • VMamba-S 取得了 83.5% 的性能,超过 RegNetY-8G 达 1.8%,Swin-S 达 0.5%。
  • VMamba-B 取得了 83.2% 的性能(有 bug,正确结果将尽快在 Github 页面更新),比 RegNetY 高 0.3%。

这些结果远高于 Vision Mamba (Vim) 模型,充分验证了 VMamba 的潜力。

COCO 目标检测

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

在 COOCO 数据集上,VMamba 也保持卓越性能:在 fine-tune 12 epochs 的情况下,VMamba-T/S/B 分别达到 46.5%/48.2%/48.5% mAP,超过了 Swin-T/S/B 达 3.8%/3.6%/1.6% mAP,超过 ConvNeXt-T/S/B 达 2.3%/2.8%/1.5% mAP。这些结果验证了 VMamba 在视觉下游实验中完全 work,展示出了能平替主流基础视觉模型的潜力。

ADE20K 语义分割

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

在 ADE20K 上,VMamba 也表现出卓越性能。VMamba-T 模型在 512 × 512 分辨率下实现 47.3% 的 mIoU,这个分数超越了所有竞争对手,包括 ResNet,DeiT,Swin 和 ConvNeXt。这种优势在 VMamba-S/B 模型下依然能够保持。

分析实验

有效感受野

刺鸟创客 刺鸟创客

一款专业高效稳定的AI内容创作平台

刺鸟创客 110 查看详情 刺鸟创客

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

VMamba 具有全局的有效感受野,其他模型中只有 DeiT 具有这个特性。但是值得注意的是,DeiT 的代价是平方级的复杂度,而 VMamaba 是线性复杂度。

输入尺度缩放

视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba

  • 上图(a)显示,VMamba 在不同输入图像尺寸下展现出最稳定的性能(不微调)。有意思的是,随着输入尺寸从 224 × 224 增加到 384 × 384,只有 VMamba 表现出性能明显上升的趋势(VMamba-S 从 83.5% 上升到 84.0%),突显了其对输入图像大小变化的稳健性。
  • 上图(b)显示,VMamba 系列模型随着输入变大,复杂性呈线性增长,这与 CNN 模型是一致的。

最后,让我们期待更多基于 Mamba 的视觉模型被提出,并列于 CNNs 和 ViTs,为基础视觉模型提供第三种选择。

以上就是视觉Mamba模型的Swin时刻,中国科学院、华为等推出VMamba的详细内容,更多请关注其它相关文章!


# 数据  # 所示  # 采用了  # 带来了  # 表现出  # 取得了  # 将于  # 三大  # 的是  # 华为  # 中国科学院  # 模型  # 兰州网站推广建设  # 西藏seo排名多少钱  # 温州营销推广报名网站  # 快手推广网站哪个便宜  # 综合商城网站如何推广  # 溪木源营销推广  # 新网站建设游戏  # 长春新手网站建设  # 苏州seo费用  # 英文版展会营销推广计划 


相关栏目: 【 行业资讯67740 】 【 技术百科0 】 【 网络运营39195


相关推荐: typescript用在哪里  开机如何运行dos命令提示符  平板键盘nfc功能是什么意思  j*a怎么创建json数组  苹果16有哪些自带配件  j*a数组对象怎么取  征信不好如何恢复正常 征信不好要怎么样才能恢复正常教程  j*a 怎么清空数组元素  哪里要用typescript  如何区别固态硬盘  电动车power灯亮是什么意思  51单片机怎么用flash  新装固态硬盘如何安装  typescript怎么拼接  单片机怎么储存和显示  索尼type-c接口是什么  怎么更新typescript  宝马x5仪表盘上边有power是什么意思  linux如何安装yum命令  typescript中如何定义json  酷我音乐怎么改每日推荐 酷我音乐每日推荐修改方法  满射为什么没有逆映射  充电器上的power是什么意思  苹果16都有哪些亮点  华为的nfc功能是什么意思  j*a怎么复制数组中  硬件如何执行命令  怎么在typescript定义集合  苹果16哪些型号好用  单片机速度怎么看  vue怎么连接typescript  为什么学typescript  买的5g手机但是没有5g网络怎么办  win10电脑如何使用命令提示符  市盈率ttm写的亏损是什么意思  广东春运抢票怎么抢不到  问一下市盈率是什么意思  丰田type-c接口是什么  移动固态硬盘如何使用  市盈率底下 18A 19E 是什么意思  什么网址不能域名解析  焊机上power指示灯亮是什么意思  如何提高import命令的性能  抖音GMV是什么_抖音GMV是什么意思  液位传感器power是什么意思  grep命令的是如何实现  typescript如何使用  datediff函数怎么用视频  unix时间戳是什么意思  夸克绑定设备是什么意思 

搜索