Saved in:
Bibliographic Details
Main Authors: Yang, Minglei, Liu, Yanfang, del-Castillo-Negrete, Diego, Cao, Yanzhao, Zhang, Guannan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.15990
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909698421161984
author Yang, Minglei
Liu, Yanfang
del-Castillo-Negrete, Diego
Cao, Yanzhao
Zhang, Guannan
author_facet Yang, Minglei
Liu, Yanfang
del-Castillo-Negrete, Diego
Cao, Yanzhao
Zhang, Guannan
contents Simulating stochastic differential equations (SDEs) in bounded domains, presents significant computational challenges due to particle exit phenomena, which requires accurate modeling of interior stochastic dynamics and boundary interactions. Despite the success of machine learning-based methods in learning SDEs, existing learning methods are not applicable to SDEs in bounded domains because they cannot accurately capture the particle exit dynamics. We present a unified hybrid data-driven approach that combines a conditional diffusion model with an exit prediction neural network to capture both interior stochastic dynamics and boundary exit phenomena. Our ML model consists of two major components: a neural network that learns exit probabilities using binary cross-entropy loss with rigorous convergence guarantees, and a training-free diffusion model that generates state transitions for non-exiting particles using closed-form score functions. The two components are integrated through a probabilistic sampling algorithm that determines particle exit at each time step and generates appropriate state transitions. The performance of the proposed approach is demonstrated via three test cases: a one-dimensional simplified problem for theoretical verification, a two-dimensional advection-diffusion problem in a bounded domain, and a three-dimensional problem of interest to magnetically confined fusion plasmas.
format Preprint
id arxiv_https___arxiv_org_abs_2507_15990
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Generative AI Models for Learning Flow Maps of Stochastic Dynamical Systems in Bounded Domains
Yang, Minglei
Liu, Yanfang
del-Castillo-Negrete, Diego
Cao, Yanzhao
Zhang, Guannan
Machine Learning
Simulating stochastic differential equations (SDEs) in bounded domains, presents significant computational challenges due to particle exit phenomena, which requires accurate modeling of interior stochastic dynamics and boundary interactions. Despite the success of machine learning-based methods in learning SDEs, existing learning methods are not applicable to SDEs in bounded domains because they cannot accurately capture the particle exit dynamics. We present a unified hybrid data-driven approach that combines a conditional diffusion model with an exit prediction neural network to capture both interior stochastic dynamics and boundary exit phenomena. Our ML model consists of two major components: a neural network that learns exit probabilities using binary cross-entropy loss with rigorous convergence guarantees, and a training-free diffusion model that generates state transitions for non-exiting particles using closed-form score functions. The two components are integrated through a probabilistic sampling algorithm that determines particle exit at each time step and generates appropriate state transitions. The performance of the proposed approach is demonstrated via three test cases: a one-dimensional simplified problem for theoretical verification, a two-dimensional advection-diffusion problem in a bounded domain, and a three-dimensional problem of interest to magnetically confined fusion plasmas.
title Generative AI Models for Learning Flow Maps of Stochastic Dynamical Systems in Bounded Domains
topic Machine Learning
url https://arxiv.org/abs/2507.15990