今天小编分享的科学经验:SIGGRAPH Asia 2023最佳论文,欢迎阅读。
NFM 团队 投稿
量子位 | 公众号 QbitAI
机器学习让计算机图形学(CG)仿真更真实了!
方法名为神经流向图(Neural Flow Maps,NFM),四个涡旋的烟雾也能精确模拟的那种:
更为复杂的也能轻松实现:
要知道,在这个 AI 应用满天飞的时代,CG 物理仿真仍然是传统数值算法的天下。
△NFM 模拟 " 蛙跳 "
尽管神经网络应用在 CG 能创造目眩神迷的视觉效果,它却无法严格、鲁棒地描述物理性质。
△NFM 模拟 " 墨滴 "
也正是因此,基于神经网络的物理仿真至今还处于概念验证(proof of concept)的阶段,所生成的效果也远非 SOTA。
基于这个难题,来自达特茅斯学院、佐治亚理工学院以及斯坦福大学的研究团队提出了神经流向图这一新方法,通过将神经网络的优异性质与先进的物理模型相结合,同时达到了前所未有的视觉效果和物理精确性。
该论文发表于图形学顶刊 ACM Transactions on Graphics(TOG),并获SIGGRAPH Asia 2023 最佳论文。
NFM 长啥样?
研究团队的核心观点是:想利用 AI 去更好地解决物理问题,就不能局限地将可学习模块(learnable modules)嵌入已有的方法框架(例如 SPH,stable fluids)中。
现有方法是针对传统数值方法的能力范畴量身定制的,也正因为如此,机器学习的发展所提出的一系列的全新的能力(例如 NeRF 对时空信号的紧凑表达),往往在已有的框架中找不到用武之地。
因此,研究人员认为与其套用 AI 在现有的框架中,不如基于 AI 提出的新能力,来设计数学和数值的新框架,从而最大化这些能力的价值。
物理模型
基于上述思路,研究人员通过对物理和 AI 进行协同设计(co-design),构建一个超越 SOTA 的流体模拟器。
物理部分,NFM 首先使用了一套基于冲量的(impulse-based)流体方程,通过对常见的欧拉方程进行度规变换(gauge transformation),确立了速度场与流向图(flow map)以及其空间导数的关系。
换言之,只要可以得到精确的 flow map 数值解,那么演化的速度场就可以被精确的重构出来。
为了最精确地计算 flow map,NFM 提出了一个精心设计的 " 双向行进 "(bidirectional marching)数值算法。
该算法比已有算法的精度高出 3 至 5 个数量级,但它同时也要求存储长期的时空(spatiotemporal)速度场。
对大规模 3D 模拟来说,存储单帧的速度场尚且存在挑战,存储数十上百帧的速度场则全然不可行。因此 " 双向行进 " 的算法尽管精准,但用传统的手段却无法实现。
神经网络存储
NFM 巧妙地结合了基于流向图的物理模型对于存储高精度速度场的需求,和隐式神经表示(implicit neural representation,或 INR)进行高质量时空信号压缩的能力,让上述高度精确但无法实现的模拟方法变得可行。
INR 通常对每个场景只需训练一次,但 NFM 却把它用作一个中间变量在模拟的过程中不断的进行更新,这也对 INR 的性能提出了更苛刻的要求。
针对这点,NFM 提出了一种称作 SSNF 的新型高性能 INR。
通过自动规划空间稀疏存储中每个格点的开启状态,以及一个基于 Lagrange 多项式的时间处理方案,SSNF 达到了比 Instant-NGP、KPlanes 等方法更快的收敛速度,更高的压缩比,以及更高的存储精度。
拿下最新 SOTA
实验结果表明,作为一个基于 AI 的模拟器,NFM 显著地超越了 SOTA 方法:bimocq、covector fluids 以及 MC+R。
在 2D 点涡(point vortex)保持的实验中,NFM 的平均绝对误差对比其余三者减少了最少 14,最多 308 倍。
在 3D 蛙跳(leapfrogging vortices)实验中,NFM 也显著提升了能量守恒的能力。
同时,这种数值能力体现为对自然现象的更好模拟:根据物理定律,蛙跳中的两对涡管将永不融和,而 NFM 的两个涡管在完成 5 次蛙跳后仍然保持分离,对比的方法至多在 3 次之后就完全融和。
最后,文章还通过一系列算例(如固体互動,瑞利泰勒不稳定性,涡管重连等)展示了 NFM 在创作复杂视觉效果上的优越性。
在这个层面上值得注意的是,尽管都是利用 AI 赋予流体更多的细节,现有的 AI 超分辨率算法只能提升画面细节,但 NFM 却突破性地以物理的方式提升了动力学细节,从而根本性地提高了流体模拟的真实度。
项目链接:https://yitongdeng-projects.github.io/neural_flow_maps_webpage/
— 完 —
点这里关注我,记得标星哦~
一键三连「分享」、「点赞」和「在看」
科技前沿进展日日相见 ~
>