google-research / torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Apache License 2.0
1.51k stars 194 forks source link

Latent SDE failed to generate longer samples #142

Closed xdong99 closed 6 months ago

xdong99 commented 7 months ago

Hi,

Thank you for this great work! I've been playing with and testing the latent_sde_lorenz example. The original model you provided, trained with 2 seconds of data sampling form the L63 system, fails to generate samples for a bit longer time. Below is the x trajectory of one generated sample comparing to one real sample.

smpx_traj1

Might this due to the insufficient training of the drift function "h"? Could you please provide some insights on how to design the latent SDE model so it can have higher generative capacity? Thank you!