Saved in:
Bibliographic Details
Main Authors: Long, Da, Xu, Zhitong, Yuan, Qiwei, Yang, Yin, Zhe, Shandian
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.11722
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913821341253632
author Long, Da
Xu, Zhitong
Yuan, Qiwei
Yang, Yin
Zhe, Shandian
author_facet Long, Da
Xu, Zhitong
Yuan, Qiwei
Yang, Yin
Zhe, Shandian
contents Fourier Neural Operator (FNO) is a powerful and popular operator learning method. However, FNO is mainly used in forward prediction, yet a great many applications rely on solving inverse problems. In this paper, we propose an invertible Fourier Neural Operator (iFNO) for jointly tackling the forward and inverse problems. We developed a series of invertible Fourier blocks in the latent channel space to share the model parameters, exchange the information, and mutually regularize the learning for the bi-directional tasks. We integrated a variational auto-encoder to capture the intrinsic structures within the input space and to enable posterior inference so as to mitigate challenges of illposedness, data shortage, noises that are common in inverse problems. We proposed a three-step process to combine the invertible blocks and the VAE component for effective training. The evaluations on seven benchmark forward and inverse tasks have demonstrated the advantages of our approach.
format Preprint
id arxiv_https___arxiv_org_abs_2402_11722
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Invertible Fourier Neural Operators for Tackling Both Forward and Inverse Problems
Long, Da
Xu, Zhitong
Yuan, Qiwei
Yang, Yin
Zhe, Shandian
Machine Learning
Fourier Neural Operator (FNO) is a powerful and popular operator learning method. However, FNO is mainly used in forward prediction, yet a great many applications rely on solving inverse problems. In this paper, we propose an invertible Fourier Neural Operator (iFNO) for jointly tackling the forward and inverse problems. We developed a series of invertible Fourier blocks in the latent channel space to share the model parameters, exchange the information, and mutually regularize the learning for the bi-directional tasks. We integrated a variational auto-encoder to capture the intrinsic structures within the input space and to enable posterior inference so as to mitigate challenges of illposedness, data shortage, noises that are common in inverse problems. We proposed a three-step process to combine the invertible blocks and the VAE component for effective training. The evaluations on seven benchmark forward and inverse tasks have demonstrated the advantages of our approach.
title Invertible Fourier Neural Operators for Tackling Both Forward and Inverse Problems
topic Machine Learning
url https://arxiv.org/abs/2402.11722