CVPR 2024 | 微软新作StarNet:超强轻量级Backbone
点击下方卡片,关注“CVer”公众号
点击下方卡片,关注“CVer”公众号
AI/CV重磅干货,第一时间送达
AI/CV重磅干货,第一时间送达
添加微信:CVer5555,小助手会拉你进群!
扫描下方二维码,加入CVer学术星球!可以获得最新顶会/顶刊上的论文idea和CV从入门到精通资料,及最前沿应用!发论文/搞科研/涨薪,强烈推荐!
论文揭示了star operation
(元素乘法)在无需加宽网络下,将输入映射到高维非线性特征空间的能力。基于此提出了StarNet
,在紧凑的网络结构和较低的能耗下展示了令人印象深刻的性能和低延迟
转载自:晓飞的算法工程笔记
论文: Rewrite the Stars
论文地址:https://arxiv.org/abs/2403.19967 论文代码:https://github.com/ma-xu/Rewrite-the-Stars 作者的介绍:神经网络中,element-wise mutiplication为什么效果好?CVPR’24
Introduction
最近,通过元素乘法融合不同的子空间特征的学习范式越来越受到关注,论文将这种范例称为star operation
(由于元素乘法符号类似于星形)。
为了便于说明,论文构建了一个用于图像分类的demo block
,如图 1 左侧所示。通过在stem
层后堆叠多个demo block
,论文构建了一个名为DemoNet
的简单模型。保持所有其他因素不变,论文观察到逐元素乘法(star operation
)在性能上始终优于求和,如图 1 右侧所示。
在这项工作中,论文证明star operation
具有将输入映射到极高维的非线性特征空间的能力,从而解释star operation
的强表达能力。论文不依赖直观或假设的高级解释,而是深入研究star operation
的细节。通过重写和重新表述star operation
计算过程,论文发现这个看似简单的运算实际可以生成一个新的特征空间,含大约 线性独立维度。
与增加网络宽度(又称通道数)的传统神经网络不同,star operation
类似于在不同通道上进行成对特征乘法的核函数,特别是多项式核函数。当应用到神经网络中并通过多层堆叠时,每一层都会带来隐式维度复杂性的指数增长。只需几层,star operation
就可以在紧凑的特征空间内实现几乎无限的维度。在紧凑的特征空间内计算,同时受益于隐含的高维度,这就是star operation
的独特魅力所在。
根据上述见解,论文推断star operation
本质上更适合高效、紧凑的网络,而不是常规使用的大型模型。为了验证这一点,论文提出了一种概念验证的高效网络StarNet
,其特点是简洁和高效。StarNet
非常简单,缺乏复杂的设计和微调的超参数。在设计理念上,StarNet
与现有网络明显不同,如表 1 所示。利用star operation
,StarNet
甚至可以超越各种精心设计的高效模型,如MobileNetv3
、EdgeViT
、FasterNet
等。这些结果不仅从经验上验证了论文对恒星运行的见解,而且强调了其在实际应用中的实用价值。
论文简要总结并强调这项工作的主要贡献如下:
证明了 star operation
的有效性,如图 1 所示,揭示了star operation
具有将特征投影到极高维隐式特征空间的能力,类似于多项式核函数。从分析中汲取灵感,确定了 star operation
在高效网络领域的实用性,并提出了概念验证模型StarNet
。无需复杂的设计或精心选择的超参数即可实现高性能,超越了许多高效的设计。基于 star operation
存在大量未探索的可能性,论文的分析可以作为指导框架,引导研究人员远离随意的网络设计尝试。
Rewrite the Stars
Star Operation in One layer
在单层神经网络中,star operation
通常写为 ,通过逐元素乘法融合两个线性变换的特征。为了方便起见,将权重矩阵和偏置合并为一个实体 ,同样地,通过 ,得到star operation
。
为了简化分析,论文重点关注涉及单输出通道转换和单元素输入的场景。具体来说,定义 ,其中 为输入通道数。这可以随时进行 扩展以适应多个输出通道,以及处理多元素输入 。
一般来说,可以通过以下方式重写star operation
:
其中 作为通道下标, 为个子项的系数:
重写star operation
后,可以将其展开为 个不同子项的组合,如等式 4 所示。值得注意的是,除了 的每个子项(这里是 偏置项)都与 呈非线性关联,表明它们是单独的隐式维度。
因此,在 维度空间中使用计算效率高的star operation
,可以得到()的隐式维度特征空间。从而在显著放大特征维度的同时,不会在单层内产生任何额外的计算开销,这个突出的属性与内核函数有着相似的理念。
Generalized to multiple layers
通过堆叠多个层,可以递归地将隐式维度以指数方式增加到几乎无限。
对于宽度为 的初始网络层,应用一次star operation
(),可得到 的隐式特征空间内。
让 表示第 个star operation
的输出,可得:
也就是说,通过堆叠 层可以隐式获得 维特征空间。例如,给定一个宽度为 128 的 10 层网络,通过star operation
获得的隐式特征维数近似为 ,相当于无限维度。因此,通过堆叠多个层,即使只有几个层,star operation
也可以以指数方式大幅放大隐式维度。
Special Cases
实际上,并非所有star operation
都遵循公式 1 那样,两个分支都进行变换。例如,VAN
和SENet
包含一个identity
分支,而GENet-
无需任何需学习的变换(池化、最近邻插值后乘回原特征)即可运行。
Case I: Non-Linear Nature of and/or
在实际场景中,大量研究(例如Conv2Former
、FocalNet
等)通过合并激活函数将变换函数 和/或 变为非线性。尽管如此,最重要的其实是看通道间的处理是否像公式 2 那样实现,是则其隐式维度仍然保持不变(大约为 )。
Case II:
当移除 变换时,隐式维度从大约 减少到 。
Case III:
在这种情况下,star operation
将特征从特征空间 转换为 的新特征空间。
有几个值得注意的方面需要考虑:
star operation
及其特殊情况通常会(尽管不一定)与空间交互集成,比如通过池化或卷积实现线性变换。但许多这些方法只强调扩大感受野带来的好处,往往忽视隐式高维空间赋予的优势。组合这些特殊情况是可行的,如 Conv2Former
合并了Case I
和Case II
,以及GENet-
混合了Case I
和Case III
。虽然 Case II
和Case III
可能不会显著增加单层的隐式维度,但使用线性层(主要用于通道通信)和skip
连接依然可以通过堆叠多个层来实现高隐式维度。
Proof-of-Concept: StarNet
鉴于star operation
的独特优势—在低维空间中计算的同时产生高维特征,论文确定了其在高效网络架构领域的实用性。因此,论文提出StarNet
作为概念验证模型,特点是极其简约的设计和显著减少的人为干预。尽管StarNet
很简单,但它展示了卓越的性能,强调了star operation
的功效。
StarNet Architecture
StarNet
采用 4 级分层架构,利用卷积层进行下采样,并使用修改后的demo block
进行特征提取。为了满足效率的要求,将Layer Normalization
替换为Batch Normalization
,并将其放置在深度卷积之后(可以在推理时融合)。受到MobileNeXt
的启发,论文在每个块的末尾加入了一个深度卷积。通道扩展因子始终设置为 4,网络宽度在每个阶段加倍。遵循MobileNetv2
设计,demo block
中的GELU
激活被替换为ReLU6
。
Experimental
Star Operation
StarNet
何恺明在MIT授课的课件PPT下载
何恺明在MIT授课的课件PPT下载
在CVer公众号后台回复:何恺明,即可下载本课程的所有566页课件PPT!赶紧学起来!
CVPR 2024 论文和代码下载
CVPR 2024 论文和代码下载
在CVer公众号后台回复:CVPR2024,即可下载CVPR 2024论文和代码开源的论文合集
Mamba、多模态和扩散模型交流群成立
扫描下方二维码,或者添加微信:CVer5555,即可添加CVer小助手微信,便可申请加入CVer-Mamba、多模态学习或者扩散模型微信交流群。另外其他垂直方向已涵盖:目标检测、图像分割、目标跟踪、人脸检测&识别、OCR、姿态估计、超分辨率、SLAM、医疗影像、Re-ID、GAN、NAS、深度估计、自动驾驶、强化学习、车道线检测、模型剪枝&压缩、去噪、去雾、去雨、风格迁移、遥感图像、行为识别、视频理解、图像融合、图像检索、论文投稿&交流、PyTorch、TensorFlow和Transformer、NeRF、3DGS、Mamba等。
一定要备注:研究方向+地点+学校/公司+昵称(如Mamba、多模态学习或者扩散模型+上海+上交+卡卡),根据格式备注,可更快被通过且邀请进群
▲扫码或加微信号: CVer5555,进交流群
CVer计算机视觉(知识星球)来了!想要了解最新最快最好的CV/DL/AI论文速递、优质实战项目、AI行业前沿、从入门到精通学习教程等资料,欢迎扫描下方二维码,加入CVer计算机视觉(知识星球),已汇集近万人!
▲扫码加入星球学习
▲点击上方卡片,关注CVer公众号
整理不易,请点赞和在看
▲扫码或加微信号: CVer5555,进交流群
CVer计算机视觉(知识星球)来了!想要了解最新最快最好的CV/DL/AI论文速递、优质实战项目、AI行业前沿、从入门到精通学习教程等资料,欢迎扫描下方二维码,加入CVer计算机视觉(知识星球),已汇集近万人!
▲扫码加入星球学习
▲点击上方卡片,关注CVer公众号
整理不易,请点赞和在看
微信扫码关注该文公众号作者