An Implicit Adaptive Fourier Neural Operator for Long-term Predictions of Three-dimensional Turbulence
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866912681863151616 |
|---|---|
| author | Jiang, Yuchi Li, Zhijie Wang, Yunpeng Yang, Huiyu Wang, Jianchun |
| author_facet | Jiang, Yuchi Li, Zhijie Wang, Yunpeng Yang, Huiyu Wang, Jianchun |
| contents | Long-term prediction of three-dimensional (3D) turbulent flows is one of the most challenging problems for machine learning approaches. Although some existing machine learning approaches such as implicit U-net enhanced Fourier neural operator (IUFNO) have been proven to be capable of achieving stable long-term predictions for turbulent flows, their computational costs are usually high. In this paper, we use the adaptive Fourier neural operator (AFNO) as the backbone to construct a model that can predict 3D turbulence. Furthermore, we employ the implicit iteration to our constructed AFNO and propose the implicit adaptive Fourier neural operator (IAFNO). IAFNO is systematically tested in three types of 3D turbulence, including forced homogeneous isotropic turbulence (HIT), temporally evolving turbulent mixing layer and turbulent channel flow. The numerical results demonstrate that IAFNO is more accurate than IUFNO and the traditional large-eddy simulation using dynamic Smagorinsky model (DSM), while exhibiting greater stability compared to IUFNO. Meanwhile, the AFNO model exhibits instability in numerical simulations. Moreover, the training efficiency of IAFNO is 4 times higher than that of IUFNO, and the number of parameters and GPU memory occupation of IAFNO are only 1/80 and 1/3 of IUFNO, respectively in HIT. In other tests, the improvements are slightly lower but still considerable. These improvements mainly come from patching and self-attention in 3D space. Besides, the well-trained IAFNO is significantly faster than the DSM. The code (and datasets) of the current work can be accessed at https://github.com/yuchi-richard-jiang/IAFNO. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2501_12740 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | An Implicit Adaptive Fourier Neural Operator for Long-term Predictions of Three-dimensional Turbulence Jiang, Yuchi Li, Zhijie Wang, Yunpeng Yang, Huiyu Wang, Jianchun Fluid Dynamics Long-term prediction of three-dimensional (3D) turbulent flows is one of the most challenging problems for machine learning approaches. Although some existing machine learning approaches such as implicit U-net enhanced Fourier neural operator (IUFNO) have been proven to be capable of achieving stable long-term predictions for turbulent flows, their computational costs are usually high. In this paper, we use the adaptive Fourier neural operator (AFNO) as the backbone to construct a model that can predict 3D turbulence. Furthermore, we employ the implicit iteration to our constructed AFNO and propose the implicit adaptive Fourier neural operator (IAFNO). IAFNO is systematically tested in three types of 3D turbulence, including forced homogeneous isotropic turbulence (HIT), temporally evolving turbulent mixing layer and turbulent channel flow. The numerical results demonstrate that IAFNO is more accurate than IUFNO and the traditional large-eddy simulation using dynamic Smagorinsky model (DSM), while exhibiting greater stability compared to IUFNO. Meanwhile, the AFNO model exhibits instability in numerical simulations. Moreover, the training efficiency of IAFNO is 4 times higher than that of IUFNO, and the number of parameters and GPU memory occupation of IAFNO are only 1/80 and 1/3 of IUFNO, respectively in HIT. In other tests, the improvements are slightly lower but still considerable. These improvements mainly come from patching and self-attention in 3D space. Besides, the well-trained IAFNO is significantly faster than the DSM. The code (and datasets) of the current work can be accessed at https://github.com/yuchi-richard-jiang/IAFNO. |
| title | An Implicit Adaptive Fourier Neural Operator for Long-term Predictions of Three-dimensional Turbulence |
| topic | Fluid Dynamics |
| url | https://arxiv.org/abs/2501.12740 |