网易首页 > 网易号 > 正文 申请入驻

用上傅里叶变换,很快啊,AI几秒钟就能解出偏微分方程

0
分享至

边策 明敏 发自 凹非寺
量子位 报道 | 公众号 QbitAI

最近的气温真是忽高忽高、让人琢磨不定,但所幸天气预报都还很准确,没有和大家开玩笑。

不过,你知道这些准确的气温预测,是通过解方程算出来的吗?

不仅如此,靠解方程还能模拟飞机空气动力、疾病传播模型!

是什么方程这么厉害?我学过吗?

它就是偏微分方程(PDE),在我们的世界中无处不在。

但在实际应用中,用计算机求解偏微分方程的难度很大,往往为了求出一个解而需要大型机器运行一个月

并且,随着科研中遇到问题的复杂度、运算量逐渐增加,也就更需要高效快速的求解方法。

最近,来自加州理工大学的一个研究团队就用AI来解决这一难题,他们开发了一种新的神经网络,比传统的PDE求解快几个数量级,并且在理论上适用于任何偏微分方程。

甚至连流体力学里的“老大难”:N-S方程也不在话下!

对于简单方程的求解,这种方法只需几秒就能解出答案,而传统方法需要18个小时!

训练神经网络=求解PDE

神经网络的本质是逼近一个函数,函数是从一个变量到另一个变量的映射。

比如图像识别网络,就是把输入的图像数据,与最后的分类结果之间建立映射关系。

训练神经网络其实就是尽可能逼近这个函数,这和数值求解PDE本质是一样的。

2016年,人们开始研究图像识别神经网络如何用于求解PDE,用成对的生成数据来训练神经网络,比如计算平面上不同基本形状(如三角形、四边形)物体周围的空气流速场。

训练数据集的输入是物体几何形状和的初始条件信息,输出是相应的二维几何物体。训练过程等于建立输入和输出之间的相关性。

训练后的神经网络,可以用于预测其他情况(比如汽车形状)的流速场,它只和与传统数值求解器的结果略有不同,但求解速度更快。

然而,对于专门研究PDE的人来说,这种方法还远远不够。

因为上面的方法精度一般达不到要求,如果想要实现更高的精度,所需的数据量和网络大小将爆炸式增长,失去了原本快速求解的意义。

从函数到算子

所以,人们想到了一种新方法,求助于“算子”。算子是一种从函数到函数的映射。

函数:数→数
算子:函数→函数

比如,正弦算子(sin)把线性函数x变成三角函数sinx,微分(求导)算子(d/dx)把三次函数x³变成二次函数3x²。

2019年,来自布朗大学和中科院的学者开发了一种“深度算子网络”(DeepONet),就是用算子的方法求解PDE。

DeepONet的特殊之处在于其分叉架构,它以两个并行网络处理数据,一个是“分支”和一个“主干”。

“分支网络”学习生成算子,也就是对输入端函数进行近似,“主干网络”负责对输出端函数进行同样操作。然后,DeepONet结合两个网络的输出,得到PDE的解。

虽然DeepONet相比PDE数值求解器速度惊人,但是它需要在训练期间进行密集计算。当必须使用大量数据训练使算子越来越精确时,可能会存在问题。

那么神经算子能加速PDE求解吗?

傅里叶变换

后来,加州理工大学与普渡大学的团队,开发了另一种新的方法——“傅里叶神经算子”(FNO)

FNO比DeepONet在计算上要更加简单,因为DeepONet还要浅层网络去近似模拟算子,而FNO有现成的方法可用,就是傅里叶变换。

有些音乐软件显示的频谱图,其实就是一种傅里叶变换,它把连续变化声音信号转换到频率空间中。

经过傅里叶变换,原来函数之间的卷积,在频率空间中变成了更简单的乘积。

FNO的核心是傅里叶层。

傅里叶层会在训练数据推送到神经网络的单层之前,对其进行傅里叶变换,然后用一个线性变换R过滤掉其中的高频部分,再通过傅里叶逆变换的得到原空间的函数。

FNO可以显著提高求解PDE的速度。在一个求解NS方程的例子中,需要3万次模拟,FNO求解时间不到一秒,DeepONet耗时2.5秒,而在这种情况下,传统的求解器需要18个小时。

团队介绍

该研究的第一作者是Zongyi Li,是加州理工计算机和数学系的一名博士二年级学生。研究方向为机器学习、理论计算机科学和应用数学。

他的导师是英伟达著名女科学家Anima Anandkumar

作为AI界的风云女性,Anima不仅在研究成果上战绩斐然,在学术圈也是怼天怼地

此前,她曾在博客公开喊话,强烈反对发论文不给代码行为,呼吁学术会议强制要求投稿同时必须公开代码,以对造假论文进行追责、利于行业公平竞争。

并且为了对抗AI学术界对女性的歧视和调侃,她曾在Twitter上大战LeCun。以一己之力让顶级学术会议NIPS改了名字,避免了女性参会者的尴尬。

Anima希望学术界更关注女性的学术成就,而不是长相。当有人在她的讲座视频下夸她漂亮时,她的做法是——删评

参考链接:
[1]https://www.quantamagazine.org/new-neural-networks-solve-hardest-equations-faster-than-ever-20210419/
[2]https://arxiv.org/abs/2010.08895
[3]https://arxiv.org/abs/1910.03193
[4]https://github.com/zongyi-li/fourier_neural_operator

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
心脏病还没出现这4种情况,可以松一口气了,问题不大!

心脏病还没出现这4种情况,可以松一口气了,问题不大!

今日养生之道
2024-11-28 20:08:33
中美贸易战打出默契,直接搞垮3个工业强国,谁会是最后赢家?

中美贸易战打出默契,直接搞垮3个工业强国,谁会是最后赢家?

奇点使者
2024-11-28 17:25:02
CBA最新消息!丁彦雨航加入广州,顾全赛季报销,同曦签约佩莱

CBA最新消息!丁彦雨航加入广州,顾全赛季报销,同曦签约佩莱

体坛瞎白话
2024-11-28 14:25:48
【旧事】东北:那年今日,辽阜新舞厅烧死233人,只因17岁青年装B

【旧事】东北:那年今日,辽阜新舞厅烧死233人,只因17岁青年装B

年之父
2024-11-27 06:10:02
广东一男子洁身自好,却感染艾滋病,只因生活中的一个坏习惯

广东一男子洁身自好,却感染艾滋病,只因生活中的一个坏习惯

凯裕说故事
2024-09-17 15:25:35
银行取钱新规已开始执行,存款人死亡后,亲属取款限额提高

银行取钱新规已开始执行,存款人死亡后,亲属取款限额提高

老鹈爱历史
2024-11-28 11:51:42
范冰冰饭局照流出,喝酒还吐舌头,素颜温婉大方,和妆后判若两人

范冰冰饭局照流出,喝酒还吐舌头,素颜温婉大方,和妆后判若两人

南城无双
2024-10-25 23:54:18
谁敢动普京!俄罗斯的两个新帮手浮出水面,中国大使出面镇场子

谁敢动普京!俄罗斯的两个新帮手浮出水面,中国大使出面镇场子

千里持剑
2024-11-28 14:07:26
65岁大爷为降糖每天吃红薯,半年后检查,结果让他喜出望外!

65岁大爷为降糖每天吃红薯,半年后检查,结果让他喜出望外!

荷兰豆爱健康
2024-11-28 17:00:33
哭灵24年!成全村首富的李美珍,一场3000每天三场,给儿子买房车

哭灵24年!成全村首富的李美珍,一场3000每天三场,给儿子买房车

原广工业
2024-11-28 22:03:46
广东队CBDL大名单出炉!三张领衔 杜锋选人意图明确

广东队CBDL大名单出炉!三张领衔 杜锋选人意图明确

胖子喷球
2024-11-28 21:06:13
40岁保姆色诱70岁教授,“性猝死”背后真相是如何露馅的?

40岁保姆色诱70岁教授,“性猝死”背后真相是如何露馅的?

苏大强专栏
2024-07-26 11:50:51
勘探队又立功了!塔克拉玛干沙漠中挖出21个湖泊,沙漠有望变绿洲

勘探队又立功了!塔克拉玛干沙漠中挖出21个湖泊,沙漠有望变绿洲

比利
2024-11-28 15:05:58
管晨辰就评论吴柳芳事件道歉,因看不惯体操"擦边",网友不买账!

管晨辰就评论吴柳芳事件道歉,因看不惯体操"擦边",网友不买账!

你食不食油饼
2024-11-26 23:21:28
欧冠争八全乱了:第6到第24差4分,3豪门预定出线,皇马6分或翻盘

欧冠争八全乱了:第6到第24差4分,3豪门预定出线,皇马6分或翻盘

体育知多少
2024-11-28 09:42:38
苏州静思园第三次流拍:起拍价1.05亿元,曾被称为“江南第一私家园林”

苏州静思园第三次流拍:起拍价1.05亿元,曾被称为“江南第一私家园林”

澎湃新闻
2024-11-28 18:06:32
为什么打工人都去广东而不是浙江?网友们说出了根本原因

为什么打工人都去广东而不是浙江?网友们说出了根本原因

据说说娱乐
2024-11-28 20:01:28
曝特斯拉新款Model Y明年1月发布 外观大改 保留拨杆

曝特斯拉新款Model Y明年1月发布 外观大改 保留拨杆

手机中国
2024-11-28 15:20:07
晚上20点!中超准入重大突破,足协或提前公布名单,半数球队过关

晚上20点!中超准入重大突破,足协或提前公布名单,半数球队过关

负角度的球
2024-11-28 21:25:23
为什么说苏57战斗机和F35对比,完全不是一个时代的产物?

为什么说苏57战斗机和F35对比,完全不是一个时代的产物?

凯撒谈兵
2024-11-28 20:26:37
2024-11-29 02:59:02
量子位
量子位
追踪人工智能动态
9675文章数 175717关注度
往期回顾 全部

科技要闻

讽刺谁?特斯拉称供应链付款周期缩至90天

头条要闻

大众宣布将出售其在新疆的业务 外交部回应

头条要闻

大众宣布将出售其在新疆的业务 外交部回应

体育要闻

道心破碎的姆巴佩,交出一张负分答卷

娱乐要闻

冯绍峰新恋情曝出!女方第一时间发文辟谣

财经要闻

洪灏:不要误解增量政策 不是数越大越好

汽车要闻

ID. CODE概念车/探岳L领衔 大众汽车携25款车亮相

态度原创

游戏
时尚
房产
本地
军事航空

《小丑牌》登Steam Deck榜单第四!紧随星露谷

蛐蛐权志龙,成为权志龙

房产要闻

湾区黄金枢纽之上,有灵魂与底蕴的「世界庄园」广佛新世界,打版第四代高质量住宅!

本地新闻

云游中国|来伦布夏果感受充满Passion的人生

军事要闻

涉嫌严重违纪 中央军委委员苗华被停职检查

无障碍浏览 进入关怀版