首页IT科技fasternet网络结构图(FasterNet)

fasternet网络结构图(FasterNet)

时间2025-05-04 15:51:22分类IT科技浏览3200
导读:更高FLOPS才是更快更强的底气,作者重新审视了现有的操作符,特别是DWConv的计算速度——FLOPS。作者发现导致低FLOPS问题的主要原因是频繁的内存访问。然后,作者提出了PConv作为一种竞争性替代方案,它减少了计算冗余以及内存访问的数量。...

更高FLOPS才是更快更强的底气,作者重新审视了现有的操作符          ,特别是DWConv的计算速度——FLOPS         。作者发现导致低FLOPS问题的主要原因是频繁的内存访问              。然后             ,作者提出了PConv作为一种竞争性替代方案    ,它减少了计算冗余以及内存访问的数量    。

论文链接:https://paperswithcode.com/paper/run-don-t-walk-chasing-higher-flops-for

为了设计快速神经网络          ,许多工作都集中在减少浮点运算(FLOPs)的数量上     。然而              ,作者观察到FLOPs的这种减少不一定会带来延迟的类似程度的减少              。这主要源于每秒低浮点运算(FLOPS)效率低下         。

为了实现更快的网络    ,作者重新回顾了FLOPs的运算符     ,并证明了如此低的FLOPS主要是由于运算符的频繁内存访问              ,尤其是深度卷积     。因此         ,本文提出了一种新的partial convolution(PConv)     ,通过同时减少冗余计算和内存访问可以更有效地提取空间特征              。

基于PConv进一步提出FasterNet              ,这是一个新的神经网络家族         ,它在广泛的设备上实现了比其他网络高得多的运行速度,而不影响各种视觉任务的准确性         。例如              ,在ImageNet-1k上小型FasterNet-T0在GPU         、CPU和ARM处理器上分别比MobileVitXXS快3.1倍              、3.1倍和2.5倍             ,同时准确度提高2.9%。

大模型FasterNet-L实现了令人印象深刻的83.5%的TOP-1精度,与Swin-B不相上下          ,同时GPU上的推理吞吐量提高了49%             ,CPU上的计算时间也节省了42%              。

神经网络在图像分类    、检测和分割等各种计算机视觉任务中经历了快速发展              。尽管其令人印象深刻的性能为许多应用程序提供了动力    ,但一个巨大的趋势是追求具有低延迟和高吞吐量的快速神经网络          ,以获得良好的用户体验     、即时响应和安全原因等。

如何快速?研究人员和从业者不需要更昂贵的计算设备              ,而是倾向于设计具有成本效益的快速神经网络    ,降低计算复杂度     ,主要以浮点运算(FLOPs)的数量来衡量         。

MobileNet              、ShuffleNet和GhostNet等利用深度卷积(DWConv)和/或组卷积(GConv)来提取空间特征              。然而              ,在减少FLOPs的过程中         ,算子经常会受到内存访问增加的副作用的影响    。MicroNet进一步分解和稀疏网络     ,将其FLOPs推至极低水平         。尽管这种方法在FLOPs方面有所改进              ,但其碎片计算效率很低              。此外         ,上述网络通常伴随着额外的数据操作,如级联         、Shuffle和池化              ,这些操作的运行时间对于小型模型来说往往很重要    。

除了上述纯卷积神经网络(CNNs)之外             ,人们对使视觉Transformer(ViTs)和多层感知器(MLP)架构更小更快也越来越感兴趣     。例如,MobileViT和MobileFormer通过将DWConv与改进的注意力机制相结合          ,降低了计算复杂性              。然而             ,它们仍然受到DWConv的上述问题的困扰    ,并且还需要修改的注意力机制的专用硬件支持         。使用先进但耗时的标准化和激活层也可能限制其在设备上的速度     。

所有这些问题一起导致了以下问题:这些“快速         ”的神经网络真的很快吗?为了回答这个问题          ,作者检查了延迟和FLOPs之间的关系              ,这由

其中FLOPS是每秒浮点运算的缩写    ,作为有效计算速度的度量              。虽然有许多减少FLOPs的尝试     ,但都很少考虑同时优化FLOPs以实现真正的低延迟         。为了更好地理解这种情况              ,作者比较了Intel CPU上典型神经网络的FLOPS。 

图2中的结果表明         ,许多现有神经网络的FLOPS较低     ,其FLOPS通常低于流行的ResNet50              。由于FLOPS如此之低              ,这些“快速              ”的神经网络实际上不够快              。它们的FLOPs减少不能转化为延迟的确切减少量。在某些情况下         ,没有任何改善,甚至会导致更糟的延迟         。例如              ,CycleMLP-B1具有ResNet50的一半FLOPs             ,但运行速度较慢(即CycleMLPB1与ResNet50:111.9ms与69.4ms)              。

请注意,FLOPs与延迟之间的差异在之前的工作中也已被注意到          ,但由于它们采用了DWConv/GConv和具有低FLOPs的各种数据处理             ,因此部分问题仍未解决    。人们认为没有更好的选择         。

本文旨在通过开发一种简单     、快速              、有效的运算符来消除这种差异    ,该运算符可以在减少FLOPs的情况下保持高FLOPS              。具体来本文旨在通过开发一种简单         、快速、有效的运算符来消除这种差异          ,该运算符可以在减少FLOPs的情况下保持高FLOPS    。

具体来说              ,作者重新审视了现有的操作符    ,特别是DWConv的计算速度——FLOPS     。作者发现导致低FLOPS问题的主要原因是频繁的内存访问              。然后     ,作者提出了PConv作为一种竞争性替代方案              ,它减少了计算冗余以及内存访问的数量         。

图1说明了PConv的设计     。它利用了特征图中的冗余         ,并系统地仅在一部分输入通道上应用规则卷积(Conv)     ,而不影响其余通道              。本质上              ,PConv的FLOPs低于常规Conv         ,而FLOPs高于DWConv/GConv         。换句话说,PConv更好地利用了设备上的计算能力。PConv在提取空间特征方面也很有效              ,这在本文后面的实验中得到了验证              。

作者进一步引入PConv设计了FasterNet作为一个在各种设备上运行速度非常快的新网络家族              。特别是             ,FasterNet在分类              、检测和分割任务方面实现了最先进的性能,同时具有更低的延迟和更高的吞吐量。例如          ,在GPU              、CPU和ARM处理器上             ,小模型FasterNet-T0分别比MobileVitXXS快3.1倍、3.1倍和2.5倍    ,而在ImageNet-1k上的准确率高2.9%         。大模型FasterNet-L实现了83.5%的Top-1精度          ,与Swin-B不相上下              ,同时在GPU上提供了49%的高吞吐量    ,在CPU上节省了42%的计算时间              。

总之     ,贡献如下:

指出了实现更高FLOPS的重要性              ,而不仅仅是为了更快的神经网络而减少FLOPs    。

引入了一种简单但快速且有效的卷积PConv         ,它很有可能取代现有的选择DWConv         。

推出FasterNet     ,它在GPU         、CPU和ARM处理器等多种设备上运行良好且普遍快速              。

对各种任务进行了广泛的实验              ,并验证了PConv和FasterNet的高速性和有效性    。

PConv和FasterNet的设计

原理

 PConv作为一个基本的算子

在下面演示了通过利用特征图的冗余度可以进一步优化成本     。如图3所示         ,特征图在不同通道之间具有高度相似性              。许多其他著作也涵盖了这种冗余,但很少有人以简单而有效的方式充分利用它         。 

请注意              ,保持其余通道不变             ,而不是从特征图中删除它们     。这是因为它们对后续PWConv层有用,PWConv允许特征信息流经所有通道              。

PConv之后是PWConv

为了充分有效地利用来自所有通道的信息          ,进一步将逐点卷积(PWConv)附加到PConv         。它们在输入特征图上的有效感受野看起来像一个T形Conv             ,与均匀处理补丁的常规Conv相比    ,它更专注于中心位置          ,如图5所示。为了证明这个T形感受野的合理性              ,首先通过计算位置的Frobenius范数来评估每个位置的重要性              。

作者认为一个显著位置是具有最大Frobenius范数的位置              。然后    ,在预训练的ResNet18中集体检查每个过滤器     ,找出它们的显著位置              ,并绘制显著位置的直方图。图6中的结果表明         ,中心位置是过滤器中最常见的突出位置         。换句话说     ,中心位置的权重比周围的更重              。这与集中于中心位置的T形计算一致    。 

FasterNet作为Backbone

鉴于新型PConv和现成的PWConv作为主要的算子              ,进一步提出FasterNet         ,这是一个新的神经网络家族,运行速度非常快              ,对许多视觉任务非常有效         。作者的目标是使体系结构尽可能简单             ,使其总体上对硬件友好              。

在图4中展示了整体架构    。它有4个层次级,每个层次级前面都有一个嵌入层(步长为4的常规4×4卷积)或一个合并层(步长为2的常规2×2卷积)          ,用于空间下采样和通道数量扩展     。每个阶段都有一堆FasterNet块              。作者观察到             ,最后两个阶段中的块消耗更少的内存访问    ,并且倾向于具有更高的FLOPS          ,如表1中的经验验证         。因此              ,放置了更多FasterNet块    ,并相应地将更多计算分配给最后两个阶段     。每个FasterNet块有一个PConv层     ,后跟2个PWConv(或Conv 1×1)层              。它们一起显示为倒置残差块              ,其中中间层具有扩展的通道数量         ,并且放置了Shorcut以重用输入特征         。

除了上述算子     ,标准化和激活层对于高性能神经网络也是不可或缺的。然而              ,许多先前的工作在整个网络中过度使用这些层         ,这可能会限制特征多样性,从而损害性能              。它还可以降低整体计算速度              。相比之下              ,只将它们放在每个中间PWConv之后             ,以保持特征多样性并实现较低的延迟。

此外,使用批次归一化(BN)代替其他替代方法         。BN的优点是          ,它可以合并到其相邻的Conv层中             ,以便更快地进行推断    ,同时与其他层一样有效              。对于激活层          ,根据经验选择了GELU用于较小的FasterNet变体              ,而ReLU用于较大的FasterNet变体    ,同时考虑了运行时间和有效性    。最后三个层     ,即全局平均池化              、卷积1×1和全连接层              ,一起用于特征转换和分类         。

为了在不同的计算预算下提供广泛的应用         ,提供FasterNet的Tiny模型    、Small模型         、Medium模型和Big模型变体     ,分别称为FasterNetT0/1/2              、FasterNet-S    、FasterNet-M和FasterNet-L              。它们具有相似的结构              ,但深度和宽度不同    。

架构规范如下:

代码实现

实验

PConv的快速性与高Flops

PConv与PWConv一起有效

FasterNet on ImageNet-1k 

 FasterNet在下游任务的表现

1     、目标检测

消融实验

whaosoft aiot http://143ai.com

创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!

展开全文READ MORE
如何让百度收录文章(掌握百度收录的14个方法,让你的网站瞬间爆红)