arXiv · ML· Vignesh Gopakumar, Zhuoyuan Wang, Ander Gray, Valentin Duruisseaux, Josh Williams, William Michael Edward Ellis, Nuno Nobre, Matthew Field, James Panayis, Alexander I. Blair, Andrew Davis, Stanislas Pamela, Robert Akers, Marc Peter Deisenroth, Anima Anandkumar·· 8 小时前AI 评分23
L-SplineFNO:用于非结构化神经算子的可学习样条
Learnable Splines for Unstructured Neural Operators
AI 导读
研究提出 L-SplineFNO,一种将 FNO 骨干与可学习张量积 B 样条耦合的内存高效神经算子,适用于非结构化网格。 该模型在 6 个三维工程基准上实现高预测精度,训练可扩展至 2660 万节点,推理达 4600 万节点,单卡 A100 即可运行。
来源:arXiv · ML · arxiv.org