Non-adversarial training of Neural SDEs with signature kernel scores

TL;DR

This paper introduces a non-adversarial training method for Neural SDEs using signature kernel scores, improving stability and performance in generating irregular time series and spatiotemporal data.

Contribution

It proposes a novel class of pathspace scoring rules based on signature kernels, providing theoretical guarantees and an efficient training framework for Neural SDEs.

Findings

Outperforms adversarial training methods on various tasks

Enables stable, memory-efficient training of Neural SDEs

Successfully generates complex spatiotemporal data

Abstract

Neural SDEs are continuous-time generative models for sequential data. State-of-the-art performance for irregular time series generation has been previously obtained by training these models adversarially as GANs. However, as typical for GAN architectures, training is notoriously unstable, often suffers from mode collapse, and requires specialised techniques such as weight clipping and gradient penalty to mitigate these issues. In this paper, we introduce a novel class of scoring rules on pathspace based on signature kernels and use them as objective for training Neural SDEs non-adversarially. By showing strict properness of such kernel scores and consistency of the corresponding estimators, we provide existence and uniqueness guarantees for the minimiser. With this formulation, evaluating the generator-discriminator pair amounts to solving a system of linear path-dependent PDEs which allows for memory-efficient adjoint-based backpropagation. Moreover, because the proposed kernel scores are well-defined for paths with values in infinite dimensional spaces of functions, our framework can be easily extended to generate spatiotemporal data. Our procedure permits conditioning on a rich variety of market conditions and significantly outperforms alternative ways of training Neural SDEs on a variety of tasks including the simulation of rough volatility models, the conditional probabilistic forecasts of real-world forex pairs where the conditioning variable is an observed past trajectory, and the mesh-free generation of limit order book dynamics.