Saved in:
Bibliographic Details
Main Authors: Samota, Sandeep Kumar, Gupta, Reema, Chakraverty, Snehashish
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2604.00669
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912995059171328
author Samota, Sandeep Kumar
Gupta, Reema
Chakraverty, Snehashish
author_facet Samota, Sandeep Kumar
Gupta, Reema
Chakraverty, Snehashish
contents This study examines the challenges of modeling complex and noisy data related to socioeconomic factors over time, with a focus on data from various districts in Odisha, India. Traditional time-series models struggle to capture both trends and variations together in this type of data. To tackle this, a Variational Neural Stochastic Differential Equation (V-NSDE) model is designed that combines the expressive dynamics of Neural SDEs with the generative capabilities of Variational Autoencoders (VAEs). This model uses an encoder and a decoder. The encoder takes the initial observations and district embeddings and translates them into a Gaussian distribution, which determines the mean and log-variance of the first latent state. Then the obtained latent state initiates the Neural SDE, which utilize neural networks to determine the drift and diffusion functions that govern continuous-time latent dynamics. These governing functions depend on the time index, latent state, and district embedding, which help the model learn the unique characteristics specific to each district. After that, using a probabilistic decoder, the observations are reconstructed from the latent trajectory. The decoder outputs a mean and log-variance for each time step, which follows the Gaussian likelihood. The Evidence Lower Bound (ELBO) training loss improves by adding a KL-divergence regularization term to the negative log-likelihood (nll). The obtained results demonstrate the effective learning of V-NSDE in recognizing complex patterns over time, yielding realistic outcomes that include clear trends and random fluctuations across different areas.
format Preprint
id arxiv_https___arxiv_org_abs_2604_00669
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Embedded Variational Neural Stochastic Differential Equations for Learning Heterogeneous Dynamics
Samota, Sandeep Kumar
Gupta, Reema
Chakraverty, Snehashish
Machine Learning
Dynamical Systems
This study examines the challenges of modeling complex and noisy data related to socioeconomic factors over time, with a focus on data from various districts in Odisha, India. Traditional time-series models struggle to capture both trends and variations together in this type of data. To tackle this, a Variational Neural Stochastic Differential Equation (V-NSDE) model is designed that combines the expressive dynamics of Neural SDEs with the generative capabilities of Variational Autoencoders (VAEs). This model uses an encoder and a decoder. The encoder takes the initial observations and district embeddings and translates them into a Gaussian distribution, which determines the mean and log-variance of the first latent state. Then the obtained latent state initiates the Neural SDE, which utilize neural networks to determine the drift and diffusion functions that govern continuous-time latent dynamics. These governing functions depend on the time index, latent state, and district embedding, which help the model learn the unique characteristics specific to each district. After that, using a probabilistic decoder, the observations are reconstructed from the latent trajectory. The decoder outputs a mean and log-variance for each time step, which follows the Gaussian likelihood. The Evidence Lower Bound (ELBO) training loss improves by adding a KL-divergence regularization term to the negative log-likelihood (nll). The obtained results demonstrate the effective learning of V-NSDE in recognizing complex patterns over time, yielding realistic outcomes that include clear trends and random fluctuations across different areas.
title Embedded Variational Neural Stochastic Differential Equations for Learning Heterogeneous Dynamics
topic Machine Learning
Dynamical Systems
url https://arxiv.org/abs/2604.00669