Time-series Generative Adversarial Networks

Jinsung Yoon, Daniel Jarrett, Mihaela van der Schaar

Abstract: A good generative model for time-series data should preserve temporal dynamics, in the sense that new sequences respect the original relationships between variables across time. Existing methods porting generative adversarial networks (GANs) to the sequential setting do not adequately attend to the temporal correlations unique to time-series data. At the same time, supervised models for sequence prediction that allow finer control over network dynamics are inherently deterministic. We propose a novel framework for generating realistic time-series data that combines the flexibility of the unsupervised paradigm with the control afforded by supervised training. Through a learned embedding space jointly optimized with both supervised and adversarial objectives, we encourage the network to adhere to the dynamics of the training data during sampling. Empirically, we evaluate the ability of our method to generate realistic samples using a variety of real and synthetic time-series datasets. Qualitatively and quantitatively, we find that the proposed framework consistently and significantly outperforms state-of-the-art benchmarks with respect to measures of similarity and predictive ability.

Code Link: https://bitbucket.org/mvdschaar/mlforhealthlabpub/src/master/alg/timegan/

CMT Num: 2946