Universal Functional Regression with Neural Operator Flows

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Shi, Yaozhong, Gao, Angela F., Ross, Zachary E., Azizzadenesheli, Kamyar
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909405870555136
author Shi, Yaozhong
Gao, Angela F.
Ross, Zachary E.
Azizzadenesheli, Kamyar
author_facet Shi, Yaozhong
Gao, Angela F.
Ross, Zachary E.
Azizzadenesheli, Kamyar
contents Regression on function spaces is typically limited to models with Gaussian process priors. We introduce the notion of universal functional regression, in which we aim to learn a prior distribution over non-Gaussian function spaces that remains mathematically tractable for functional regression. To do this, we develop Neural Operator Flows (OpFlow), an infinite-dimensional extension of normalizing flows. OpFlow is an invertible operator that maps the (potentially unknown) data function space into a Gaussian process, allowing for exact likelihood estimation of functional point evaluations. OpFlow enables robust and accurate uncertainty quantification via drawing posterior samples of the Gaussian process and subsequently mapping them into the data function space. We empirically study the performance of OpFlow on regression and generation tasks with data generated from Gaussian processes with known posterior forms and non-Gaussian processes, as well as real-world earthquake seismograms with an unknown closed-form distribution.
format Preprint
id arxiv_https___arxiv_org_abs_2404_02986
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Universal Functional Regression with Neural Operator Flows
Shi, Yaozhong
Gao, Angela F.
Ross, Zachary E.
Azizzadenesheli, Kamyar
Machine Learning
Regression on function spaces is typically limited to models with Gaussian process priors. We introduce the notion of universal functional regression, in which we aim to learn a prior distribution over non-Gaussian function spaces that remains mathematically tractable for functional regression. To do this, we develop Neural Operator Flows (OpFlow), an infinite-dimensional extension of normalizing flows. OpFlow is an invertible operator that maps the (potentially unknown) data function space into a Gaussian process, allowing for exact likelihood estimation of functional point evaluations. OpFlow enables robust and accurate uncertainty quantification via drawing posterior samples of the Gaussian process and subsequently mapping them into the data function space. We empirically study the performance of OpFlow on regression and generation tasks with data generated from Gaussian processes with known posterior forms and non-Gaussian processes, as well as real-world earthquake seismograms with an unknown closed-form distribution.
title Universal Functional Regression with Neural Operator Flows
topic Machine Learning
url https://arxiv.org/abs/2404.02986