Saved in:
Bibliographic Details
Main Authors: Abdellatif, Alhasan, Menke, Hannah P., Doster, Florian, Singh, Kamaljit, Elsheikh, Ahmed H.
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2511.20543
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915702865133568
author Abdellatif, Alhasan
Menke, Hannah P.
Doster, Florian
Singh, Kamaljit
Elsheikh, Ahmed H.
author_facet Abdellatif, Alhasan
Menke, Hannah P.
Doster, Florian
Singh, Kamaljit
Elsheikh, Ahmed H.
contents The UNet-enhanced Fourier Neural Operator (UFNO) extends the Fourier Neural Operator (FNO) by incorporating a parallel UNet pathway, enabling the retention of both high- and low-frequency components. While UFNO improves predictive accuracy over FNO, it inefficiently treats scalar inputs (e.g., temperature, injection rate) as spatially distributed fields by duplicating their values across the domain. This forces the model to process redundant constant signals within the frequency domain. Additionally, its standard loss function does not account for spatial variations in error sensitivity, limiting performance in regions of high physical importance. We introduce UFNO-FiLM, an enhanced architecture that incorporates two key innovations. First, we decouple scalar inputs from spatial features using a Feature-wise Linear Modulation (FiLM) layer, allowing the model to modulate spatial feature maps without introducing constant signals into the Fourier transform. Second, we employ a spatially weighted loss function that prioritizes learning in critical regions. Our experiments on subsurface multiphase flow demonstrate a 21\% reduction in gas saturation Mean Absolute Error (MAE) compared to UFNO, highlighting the effectiveness of our approach in improving predictive accuracy.
format Preprint
id arxiv_https___arxiv_org_abs_2511_20543
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Feature-Modulated UFNO for Improved Prediction of Multiphase Flow in Porous Media
Abdellatif, Alhasan
Menke, Hannah P.
Doster, Florian
Singh, Kamaljit
Elsheikh, Ahmed H.
Machine Learning
The UNet-enhanced Fourier Neural Operator (UFNO) extends the Fourier Neural Operator (FNO) by incorporating a parallel UNet pathway, enabling the retention of both high- and low-frequency components. While UFNO improves predictive accuracy over FNO, it inefficiently treats scalar inputs (e.g., temperature, injection rate) as spatially distributed fields by duplicating their values across the domain. This forces the model to process redundant constant signals within the frequency domain. Additionally, its standard loss function does not account for spatial variations in error sensitivity, limiting performance in regions of high physical importance. We introduce UFNO-FiLM, an enhanced architecture that incorporates two key innovations. First, we decouple scalar inputs from spatial features using a Feature-wise Linear Modulation (FiLM) layer, allowing the model to modulate spatial feature maps without introducing constant signals into the Fourier transform. Second, we employ a spatially weighted loss function that prioritizes learning in critical regions. Our experiments on subsurface multiphase flow demonstrate a 21\% reduction in gas saturation Mean Absolute Error (MAE) compared to UFNO, highlighting the effectiveness of our approach in improving predictive accuracy.
title Feature-Modulated UFNO for Improved Prediction of Multiphase Flow in Porous Media
topic Machine Learning
url https://arxiv.org/abs/2511.20543