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

fasternet网络结构图(FasterNet)

时间2025-08-05 07:44:25分类IT科技浏览4269
导读:更高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
德化seo方法的技巧(德化seo的窍门)