Skip to yearly menu bar Skip to main content


Poster
in
Workshop: HiLD: High-dimensional Learning Dynamics Workshop

Learning Stochastic Dynamical Systems as an Implicit Regularization with Graph Neural Network

Jin Guo · Ting Gao · Yufu Lan · Peng Zhang · Sikun Yang · Jinqiao Duan


Abstract:

We provide a stochastic Gumbel Graph Network (S-GGN) for the long-term prediction of high-dimensional time series data with complex noise and spatial correlations. Specifically, by training the drift and diffusion terms of a stochastic differential equation (SDE) with a Gumbel Matrix embedding, we can capture both the randomness and underlying spatial dynamics of input data. This framework allows us to investigate the implicit regularization effect of noise in S-GGN. First, we derive the difference between the two corresponding loss functions in a small neighborhood of weight. Then, we employ Kuramoto's model to generate data for comparing the spectral density from the Hessian Matrix of the two loss functions. Moreover, we also apply our model to real-world scenarios such as wireless communication data. Experiment results demonstrate that S-GGN exhibits better convergence, robustness, and generalization.

Chat is not available.