Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modeling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Niu, Ruijia, Wu, Dongxia, Kim, Kai, Ma, Yi-An, Watson-Parris, Duncan, Yu, Rose
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917702523682816
author Niu, Ruijia
Wu, Dongxia
Kim, Kai
Ma, Yi-An
Watson-Parris, Duncan
Yu, Rose
author_facet Niu, Ruijia
Wu, Dongxia
Kim, Kai
Ma, Yi-An
Watson-Parris, Duncan
Yu, Rose
contents Multi-fidelity surrogate modeling aims to learn an accurate surrogate at the highest fidelity level by combining data from multiple sources. Traditional methods relying on Gaussian processes can hardly scale to high-dimensional data. Deep learning approaches utilize neural network based encoders and decoders to improve scalability. These approaches share encoded representations across fidelities without including corresponding decoder parameters. This hinders inference performance, especially in out-of-distribution scenarios when the highest fidelity data has limited domain coverage. To address these limitations, we propose Multi-fidelity Residual Neural Processes (MFRNP), a novel multi-fidelity surrogate modeling framework. MFRNP explicitly models the residual between the aggregated output from lower fidelities and ground truth at the highest fidelity. The aggregation introduces decoders into the information sharing step and optimizes lower fidelity decoders to accurately capture both in-fidelity and cross-fidelity information. We show that MFRNP significantly outperforms state-of-the-art in learning partial differential equations and a real-world climate modeling task. Our code is published at: https://github.com/Rose-STL-Lab/MFRNP
format Preprint
id arxiv_https___arxiv_org_abs_2402_18846
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modeling
Niu, Ruijia
Wu, Dongxia
Kim, Kai
Ma, Yi-An
Watson-Parris, Duncan
Yu, Rose
Machine Learning
Multi-fidelity surrogate modeling aims to learn an accurate surrogate at the highest fidelity level by combining data from multiple sources. Traditional methods relying on Gaussian processes can hardly scale to high-dimensional data. Deep learning approaches utilize neural network based encoders and decoders to improve scalability. These approaches share encoded representations across fidelities without including corresponding decoder parameters. This hinders inference performance, especially in out-of-distribution scenarios when the highest fidelity data has limited domain coverage. To address these limitations, we propose Multi-fidelity Residual Neural Processes (MFRNP), a novel multi-fidelity surrogate modeling framework. MFRNP explicitly models the residual between the aggregated output from lower fidelities and ground truth at the highest fidelity. The aggregation introduces decoders into the information sharing step and optimizes lower fidelity decoders to accurately capture both in-fidelity and cross-fidelity information. We show that MFRNP significantly outperforms state-of-the-art in learning partial differential equations and a real-world climate modeling task. Our code is published at: https://github.com/Rose-STL-Lab/MFRNP
title Multi-Fidelity Residual Neural Processes for Scalable Surrogate Modeling
topic Machine Learning
url https://arxiv.org/abs/2402.18846