cs.LGJul 9, 2026

Deep Learning Method for Stationary Distribution of Reflected Brownian Motion

Authors: Jim DaiZhanhao Zhang

Organizations: Operations Research and Information Engineering, Cornell University

Abstract

The stationary distribution of reflected Brownian motion (RBM) plays an important role in the analysis of high-dimensional stochastic systems, yet closed-form solutions are known only for a few special cases. Computing important performance metrics, such as tail probabilities, is even more intractable, despite their practical relevance. In this paper, we develop a deep learning approach that accurately and efficiently learns the Laplace transform of high-dimensional RBMs based on the basic adjoint relationship (BAR). Our framework combines a careful design of the loss function, training data sampling procedure, and neural network architecture. We evaluate the proposed method on RBM instances with known ground-truth tail probabilities and demonstrate near-perfect prediction in high-dimensional settings, highlighting its potential as a general tool for analyzing stochastic systems beyond analytically tractable regimes. Our code can be found at https://github.com/zhangz73/NN4MGF.

Explore similar work

CardsList
  1. Reflected Schrödinger Bridge Matching

    Jul 3, 2026Marcus Häggbom, Viktor Nilsson, Pierre Nyquist +1Schrödinger BridgesGenerative Models

  2. Don't Stop Me Yet: Sampling Loss Minima via Dissipative Riemannian Mechanics

    May 14, 2026Albert Kjøller Jacobsen, Leo Uhre Jakobsen, Johanna Marie Gegenfurtner +1SamplersReparameterization