Computer Science ›› 2025, Vol. 52 ›› Issue (9): 144-151.doi: 10.11896/jsjkx.240700122

• High Performance Computing • Previous Articles     Next Articles

Partial Differential Equation Solving Method Based on Locally Enhanced Fourier NeuralOperators

LUO Chi1, LU Lingyun2, LIU Fei1   

  1. 1 School of Software Engineering,South China University of Technology,Guangzhou 510006,China
    2 Nanjing Research Institute of Electronics Engineering,Nanjing 210007,China
  • Received:2024-07-19 Revised:2024-10-22 Online:2025-09-15 Published:2025-09-11
  • About author:LUO Chi,born in 1997,postgraduate.His main research interest is deep learning.
    LIU Fei,born in 1976,Ph.D,professor,Ph.D supervisor,is a member of CCF(No.B9231M).His main research interests include modeling and simulation,and artificial intelligence.
  • Supported by:
    National Natural Science Foundation of China(62273153) and Guangdong Basic and Applied Basic Research Foundation(2024A1515010900).

Abstract: Partial differential equations(PDEs) are crucial mathematical tools for describing real-world systems,and solving them is key for predicting and analyzing system behavior.Analytical solutions for PDEs are often difficult to obtain,and numerical methods are typically used for approximate solutions.However,numerical solutions for parameterized PDEs can be inefficient.In recent years,the use of deep learning for solving PDEs has shown its advantages in addressing these issues,and particularly the Fourier Neural Operator(FNO) has proven effective.However,FNO only captures global information through convolution in the frequency domain and struggles with multi-scale information of PDEs.To address this challenge,a locally-enhanced FNO model is proposed,incorporating a parallel multi-size convolution module in the Fourier layer to enhance the model’s capability to capture local multi-scale information.Behind the linear layer,a multi-branch feature fusion module is introduced,enhancing the model’s ability to integrate multi-channel information by elevating the data across different channels.Experimental results demonstrate that the model reduces errors by 30.9% in solving Burgers’ equation,18.5% in Darcy Flow equations,and 5.5% in Navier-Stokes equations.

Key words: Deep learning, Partial differential equations, Fourier neural operator, Multi-size convolution, Multi-branch feature fusion, Multiscale PDE

CLC Number: 

  • TP391
[1]KARNIADAKIS G E,KEVREKIDIS I G,LU L,et al.Physics-informed machine learning[J].Nature Reviews Physics,2021,3(6):422-440.
[2]ZHANG T,LIU R Y,WANG P X,et al.Machine Learning with Perceptual Physical Priors and Its Prospects in Geospatial Intelligence Research[J].Journal of Geo-information Science,2023,25(7):1297-1311.
[3]RAISSI M,PERDIKARIS P,KARNIADAKIS G.Physics-in-formed neural networks:A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations[J].Journal of Computational Physics,2019,378:686-707.
[4]HU S,LIU M,ZHANG S,et al.Physics-informed neural network combined with characteristic-based split for solving forward and inverse problems involving Navier-Stokes equations[J].Neurocomputing,2024,573:127240.
[5]YAZDANI A,LU L,RAISSI M,et al.Systems biology informed deep learning for inferring parameters and hidden dynamics[J].PLoS Computational Biology,2020,16(11):e1007575.
[6]DIAO Y,YANG J,ZHANG Y,et al.Solving multi-materialproblems in solid mechanics using physics-informed neural networks based on domain decomposition technology[J].Computer Methods in Applied Mechanics and Engineering,2023,413:116120.
[7]ZHAO Y,GUO L,WONG P P L.Application of physics-informed neural network in the analysis of hydrodynamic lubrication[J].Friction,2023,11(7):1253-1264.
[8]LI Z,KOVACHKI N B,AZIZZADENESHELI K,et al.Fourier Neural Operator for Parametric Partial Differential Equations[C]//International Conference on Learning Representations(ICLR).2021:1-12.
[9]KURTH T,SUBRAMANIAN S,HARRINGTON P,et al.Fourcastnet:Accelerating global high-resolution weather forecasting using adaptive fourier neural operators[C]//The Platform for Advanced Scientific Computing Conference.2022:1-15.
[10]LI B,WANG H,FENG S,et al.Solving Seismic Wave Equations on Variable Velocity Models With Fourier Neural Operator[J].IEEE Transactions on Geoscience and Remote Sensing,2023,61:1-18.
[11]GUAN S,HSU K T,CHITNIS P V.Fourier neural operator network for fast photoacoustic wave simulations[J].Algorithms,2023,16(2):124.
[12]TRAN A,MATHEWS A,XIE L,et al.Factorized Fourier Neural Operators[C]//The Eleventh International Conference on Learning Representations(ICLR).2023:1-12.
[13]BONEV B,KURTH T,HUNDT C,et al.Spherical FourierNeural Operators:Learning Stable Dynamics on the Sphere[C]//The 40th International Conference on Machine Learning(ICML).2023:2806-2823.
[14]HELWIG J,ZHANG X,FU C,et al.Group uivariant FourierNeural Operators for Partial Differential Equations[C]//The 40th International Conference on Machine Learning(ICML).2023:12907-12930.
[15]WHITE C,TU R,KOSSAIFI J,et al.Speeding up Fourier Neural Operators via Mixed Precision[J].arXiv:2307.15034,2023.
[16]ZHAO X,SUN Y,ZHANG T,et al.Local Convolution En-hanced Global Fourier Neural Operator for Multiscale Dynamic Spaces Prediction[J].arXiv:2311.12902,2023.
[17]SZEGEDY C,LIU W,JIA Y,et al.Going Deeper with Convolutions[C]//The IEEE Conference on Computer Vision and Pattern Recognition(CVPR).2015:1-9.
[18]WU G,JIANG J,JIANG K,et al.Fully 1×1 Convolutional Network for Lightweight Image Super-Resolution[J].arXiv:2307.16140,2023.
[19]SHELHAMER E,LONG J,DARRELL T.Fully Convolutional Networks for Semantic Segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(4):640-651.
[20]RONNEBERGER O,FISCHER P,BROX T.U-Net:ConvolutionalNetworks for Biomedical Image Segmentation[C]//The International Conference on Medical Image Computing and Compu-ter-Assisted Intervention(MICCAI).Springer,2015:234-241.
[21]LU L,JIN P,PANG G,et al.Learning nonlinear operators via DeepONet based on the universal approximation theorem of ope-rators[J].Nature Machine Intelligence,2021,3(3):218-229.
[22]VINUESA R,BRUNTON S L.Enhancing computational fluid dynamics with machine learning[J].Nature Computational Science,2022,2(6):358-366,385.
[1] LIU Wei, XU Yong, FANG Juan, LI Cheng, ZHU Yujun, FANG Qun, HE Xin. Multimodal Air-writing Gesture Recognition Based on Radar-Vision Fusion [J]. Computer Science, 2025, 52(9): 259-268.
[2] YIN Shi, SHI Zhenyang, WU Menglin, CAI Jinyan, YU De. Deep Learning-based Kidney Segmentation in Ultrasound Imaging:Current Trends and Challenges [J]. Computer Science, 2025, 52(9): 16-24.
[3] ZENG Lili, XIA Jianan, LI Shaowen, JING Maike, ZHAO Huihui, ZHOU Xuezhong. M2T-Net:Cross-task Transfer Learning Tongue Diagnosis Method Based on Multi-source Data [J]. Computer Science, 2025, 52(9): 47-53.
[4] LI Yaru, WANG Qianqian, CHE Chao, ZHU Deheng. Graph-based Compound-Protein Interaction Prediction with Drug Substructures and Protein 3D Information [J]. Computer Science, 2025, 52(9): 71-79.
[5] LIU Leyuan, CHEN Gege, WU Wei, WANG Yong, ZHOU Fan. Survey of Data Classification and Grading Studies [J]. Computer Science, 2025, 52(9): 195-211.
[6] TANG Boyuan, LI Qi. Review on Application of Spatial-Temporal Graph Neural Network in PM2.5 ConcentrationForecasting [J]. Computer Science, 2025, 52(8): 71-85.
[7] LIU Zhengyu, ZHANG Fan, QI Xiaofeng, GAO Yanzhao, SONG Yijing, FAN Wang. Review of Research on Deep Learning Compiler [J]. Computer Science, 2025, 52(8): 29-44.
[8] ZHENG Cheng, YANG Nan. Aspect-based Sentiment Analysis Based on Syntax,Semantics and Affective Knowledge [J]. Computer Science, 2025, 52(7): 218-225.
[9] FAN Xing, ZHOU Xiaohang, ZHANG Ning. Review on Methods and Applications of Short Text Similarity Measurement in Social Media Platforms [J]. Computer Science, 2025, 52(6A): 240400206-8.
[10] YANG Jixiang, JIANG Huiping, WANG Sen, MA Xuan. Research Progress and Challenges in Forest Fire Risk Prediction [J]. Computer Science, 2025, 52(6A): 240400177-8.
[11] WANG Jiamin, WU Wenhong, NIU Hengmao, SHI Bao, WU Nier, HAO Xu, ZHANG Chao, FU Rongsheng. Review of Concrete Defect Detection Methods Based on Deep Learning [J]. Computer Science, 2025, 52(6A): 240900137-12.
[12] HAO Xu, WU Wenhong, NIU Hengmao, SHI Bao, WU Nier, WANG Jiamin, CHU Hongkun. Survey of Man-Machine Distance Detection Method in Construction Site [J]. Computer Science, 2025, 52(6A): 240700098-10.
[13] ZHOU Lei, SHI Huaifeng, YANG Kai, WANG Rui, LIU Chaofan. Intelligent Prediction of Network Traffic Based on Large Language Model [J]. Computer Science, 2025, 52(6A): 241100058-7.
[14] GUAN Xin, YANG Xueyong, YANG Xiaolin, MENG Xiangfu. Tumor Mutation Prediction Model of Lung Adenocarcinoma Based on Pathological [J]. Computer Science, 2025, 52(6A): 240700010-8.
[15] TAN Jiahui, WEN Chenyan, HUANG Wei, HU Kai. CT Image Segmentation of Intracranial Hemorrhage Based on ESC-TransUNet Network [J]. Computer Science, 2025, 52(6A): 240700030-9.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!