Neural McKean-Vlasov Processes: Distributional Dependence in Diffusion Processes

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Haoming, Hasan, Ali, Ng, Yuting, Tarokh, Vahid
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916205498990592
author Yang, Haoming
Hasan, Ali
Ng, Yuting
Tarokh, Vahid
author_facet Yang, Haoming
Hasan, Ali
Ng, Yuting
Tarokh, Vahid
contents McKean-Vlasov stochastic differential equations (MV-SDEs) provide a mathematical description of the behavior of an infinite number of interacting particles by imposing a dependence on the particle density. As such, we study the influence of explicitly including distributional information in the parameterization of the SDE. We propose a series of semi-parametric methods for representing MV-SDEs, and corresponding estimators for inferring parameters from data based on the properties of the MV-SDE. We analyze the characteristics of the different architectures and estimators, and consider their applicability in relevant machine learning problems. We empirically compare the performance of the different architectures and estimators on real and synthetic datasets for time series and probabilistic modeling. The results suggest that explicitly including distributional dependence in the parameterization of the SDE is effective in modeling temporal data with interaction under an exchangeability assumption while maintaining strong performance for standard Itô-SDEs due to the richer class of probability flows associated with MV-SDEs.
format Preprint
id arxiv_https___arxiv_org_abs_2404_09402
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Neural McKean-Vlasov Processes: Distributional Dependence in Diffusion Processes
Yang, Haoming
Hasan, Ali
Ng, Yuting
Tarokh, Vahid
Machine Learning
Artificial Intelligence
McKean-Vlasov stochastic differential equations (MV-SDEs) provide a mathematical description of the behavior of an infinite number of interacting particles by imposing a dependence on the particle density. As such, we study the influence of explicitly including distributional information in the parameterization of the SDE. We propose a series of semi-parametric methods for representing MV-SDEs, and corresponding estimators for inferring parameters from data based on the properties of the MV-SDE. We analyze the characteristics of the different architectures and estimators, and consider their applicability in relevant machine learning problems. We empirically compare the performance of the different architectures and estimators on real and synthetic datasets for time series and probabilistic modeling. The results suggest that explicitly including distributional dependence in the parameterization of the SDE is effective in modeling temporal data with interaction under an exchangeability assumption while maintaining strong performance for standard Itô-SDEs due to the richer class of probability flows associated with MV-SDEs.
title Neural McKean-Vlasov Processes: Distributional Dependence in Diffusion Processes
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2404.09402