超越传统卷积!快速傅里叶卷积突破计算极限,加速高达7.93倍

本文主要是介绍超越传统卷积!快速傅里叶卷积突破计算极限,加速高达7.93倍,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

通过结合快速傅里叶变换的优势,快速傅里叶卷积(FFC)提供了一种高效的方式来执行卷积操作,特别是在需要全局上下文信息和跨尺度特征融合的场景中。

这种独特的频域操作不仅能提高特征提取的效率和质量,还可以加速计算过程。在实际应用中帮助我们提高模型性能,同时保持较低的计算成本。比如:

  • 英伟达等团队提出的基于傅里叶的神经网络预测模型FourCastNet,在节点小时(node-hour)基础上比传统 NWP 模型快约 45,000 倍。

  • 斯坦福团队引入FlashFFTConv来优化机器学习中长序列的FFT卷积,可将FFT卷积加速高达7.93倍。

为帮助同学们深入了解FFC原理,获取科研灵感,我整理了10个快速傅里叶卷积创新方案,来源文章以及开源代码也列上了,方便同学们复现。

论文和代码需要的同学看文末

FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores

方法:作者提出了一种新的系统FlashFFTConv,通过Monarch FFT分解对FFT卷积进行优化。该分解将FFT表示为一系列矩阵乘法操作,可以有效地映射到硬件上。通过简单的GPU成本模型,作者展示了如何根据序列长度调整分解的顺序p,以平衡FLOP成本和I/O成本。

FlashFFTConv还利用实值FFT算法将FFT操作的长度减半,并在输入进行零填充时有选择地跳过矩阵乘法操作的部分。此外,作者还提出了部分卷积和频率稀疏卷积的架构扩展,这些扩展可以减少内存占用和运行时间。

创新点:

  • FlashFFTConv是一种优化FFT卷积的新系统,通过Monarch分解FFT来提高卷积的效率和性能。

  • FlashFFTConv通过优化FFT卷积算法,提高了卷积序列模型的质量和效率。

  • FlashFFTConv还引入了部分卷积和频率稀疏卷积两种新的卷积算法,可以进一步减少内存占用和提高计算速度。

FOURCASTNET: A GLOBAL DATA-DRIVEN HIGH-RESOLUTION WEATHER MODEL USING ADAPTIVE FOURIER NEURAL OPERATORS

方法:FourCastNet是一种全球数据驱动的深度学习(DL)天气预报模型,基于Fourier神经算子(FNO)和自适应Fourier神经算子(AFNO)。FourCastNet能够以很高的准确性生成高分辨率的风速和降水预报,预测速度更快、计算成本更低,并且能够生成大规模的集合预报。

创新点:

  • FourCastNet 的分辨率比现有的最先进基于DL的全球天气模型高出八倍,使其能够准确解析极端事件,如热带气旋和大气河。

  • 在最长一周的预报时效内,FourCastNet 的预测结果可以与传统的数值天气预报(NWP)模型IFS相媲美,展现了数据驱动模型在补充甚至最终替代NWP方面的潜力。

  • FourCastNet 能够在数秒内生成非常大的集合预报,提高了对极端天气事件预警的可靠性,并能够快速评估其影响。

  • FourCastNet 比传统的NWP模型快大约45,000倍,并且能耗降低了12,000倍,使其成为天气预报的一种更高效、更经济的解决方案。

Fast-ParC: Capturing Position Aware Global Feature for ConvNets and ViTs

方法:本文提出了一种名为ParC(Positional Aware Circular Convolution)的新型插入式操作,将Transformer和ConvNet的优点相结合。ParC通过使用全局卷积核和循环卷积来捕捉全局特征,并使用位置编码保持位置敏感性。与MHA相比,ParC将全局操作的时间复杂度从O(n^2)降低到O(n^3/2)。通过使用FFT(Fast Fourier Transform),Fast-ParC将ParC的复杂度进一步降低到O(n log n)。

创新点:

  • ParC操作
    • ParC操作是一种新颖的插入式操作,结合了ViT中使用的自注意力机制和纯卷积操作,具有全局感受野。

    • ParC操作可以更方便地在不同硬件平台上支持,并提高网络在分类任务上的性能。

    • 基于ParC的模型在下游任务上也表现出优越性能。

  • Fast-ParC操作
    • Fast-ParC是ParC的一种基于FFT的加速版本,用于处理高分辨率输入特征。

    • Fast-ParC操作能够在高输入分辨率的情况下保持较低的计算预算,使ParC成为大多数计算机视觉任务的竞争性通用选择。

    • Fast-ParC操作进一步拓宽了ParC的应用场景。

ADAPTIVE FOURIER NEURAL OPERATORS: EFFICIENT TOKEN MIXERS FOR TRANSFORMERS

方法:论文研究了运算符学习和高分辨率令牌混合的有效变换器之间的关系。通过建立了运算符学习和高分辨率令牌混合之间的联系,并将FNO从PDEs中进行了适应,提出了一种具有准线性复杂度的高效混合器AFNO。

创新点:

  • 将运算符学习与高分辨率令牌混合相结合,将PDE中的FNO改编为具有准线性复杂度的高效混合器。

  • 通过引入块对角结构、自适应权重共享和稀疏性,以有原则的方式设计了AFNO,以提高其表达能力和泛化能力。

关注下方《学姐带你玩AI》🚀🚀🚀

回复“FFC模块”获取全部论文+代码

码字不易,欢迎大家点赞评论收藏

这篇关于超越传统卷积!快速傅里叶卷积突破计算极限,加速高达7.93倍的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/833926

相关文章

电脑桌面文件删除了怎么找回来?别急,快速恢复攻略在此

在日常使用电脑的过程中,我们经常会遇到这样的情况:一不小心,桌面上的某个重要文件被删除了。这时,大多数人可能会感到惊慌失措,不知所措。 其实,不必过于担心,因为有很多方法可以帮助我们找回被删除的桌面文件。下面,就让我们一起来了解一下这些恢复桌面文件的方法吧。 一、使用撤销操作 如果我们刚刚删除了桌面上的文件,并且还没有进行其他操作,那么可以尝试使用撤销操作来恢复文件。在键盘上同时按下“C

黑神话,XSKY 星飞全闪单卷性能突破310万

当下,云计算仍然是企业主要的基础架构,随着关键业务的逐步虚拟化和云化,对于块存储的性能要求也日益提高。企业对于低延迟、高稳定性的存储解决方案的需求日益迫切。为了满足这些日益增长的 IO 密集型应用场景,众多云服务提供商正在不断推陈出新,推出具有更低时延和更高 IOPS 性能的云硬盘产品。 8 月 22 日 2024 DTCC 大会上(第十五届中国数据库技术大会),XSKY星辰天合正式公布了基于星

poj 1113 凸包+简单几何计算

题意: 给N个平面上的点,现在要在离点外L米处建城墙,使得城墙把所有点都包含进去且城墙的长度最短。 解析: 韬哥出的某次训练赛上A出的第一道计算几何,算是大水题吧。 用convexhull算法把凸包求出来,然后加加减减就A了。 计算见下图: 好久没玩画图了啊好开心。 代码: #include <iostream>#include <cstdio>#inclu

uva 1342 欧拉定理(计算几何模板)

题意: 给几个点,把这几个点用直线连起来,求这些直线把平面分成了几个。 解析: 欧拉定理: 顶点数 + 面数 - 边数= 2。 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <cmath>#inc

uva 11178 计算集合模板题

题意: 求三角形行三个角三等分点射线交出的内三角形坐标。 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <cmath>#include <stack>#include <vector>#include <

XTU 1237 计算几何

题面: Magic Triangle Problem Description: Huangriq is a respectful acmer in ACM team of XTU because he brought the best place in regional contest in history of XTU. Huangriq works in a big compa

hdu 4565 推倒公式+矩阵快速幂

题意 求下式的值: Sn=⌈ (a+b√)n⌉%m S_n = \lceil\ (a + \sqrt{b}) ^ n \rceil\% m 其中: 0<a,m<215 0< a, m < 2^{15} 0<b,n<231 0 < b, n < 2^{31} (a−1)2<b<a2 (a-1)^2< b < a^2 解析 令: An=(a+b√)n A_n = (a +

v0.dev快速开发

探索v0.dev:次世代开发者之利器 今之技艺日新月异,开发者之工具亦随之进步不辍。v0.dev者,新兴之开发者利器也,迅速引起众多开发者之瞩目。本文将引汝探究v0.dev之基本功能与优势,助汝速速上手,提升开发之效率。 何谓v0.dev? v0.dev者,现代化之开发者工具也,旨在简化并加速软件开发之过程。其集多种功能于一体,助开发者高效编写、测试及部署代码。无论汝为前端开发者、后端开发者

音视频入门基础:WAV专题(10)——FFmpeg源码中计算WAV音频文件每个packet的pts、dts的实现

一、引言 从文章《音视频入门基础:WAV专题(6)——通过FFprobe显示WAV音频文件每个数据包的信息》中我们可以知道,通过FFprobe命令可以打印WAV音频文件每个packet(也称为数据包或多媒体包)的信息,这些信息包含该packet的pts、dts: 打印出来的“pts”实际是AVPacket结构体中的成员变量pts,是以AVStream->time_base为单位的显

利用Django框架快速构建Web应用:从零到上线

随着互联网的发展,Web应用的需求日益增长,而Django作为一个高级的Python Web框架,以其强大的功能和灵活的架构,成为了众多开发者的选择。本文将指导你如何从零开始使用Django框架构建一个简单的Web应用,并将其部署到线上,让世界看到你的作品。 Django简介 Django是由Adrian Holovaty和Simon Willison于2005年开发的一个开源框架,旨在简