atlas

Research map / Time series / Causal and generative time series

SDformer: Similarity-driven Discrete Transformer For Time Series Generation

NeurIPS 2024 · Time series › Causal and generative time series · approach: Diffusion models for time series · 3 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SDformer, a two-stage discrete token model for time series generation using a similarity-driven VQ-VAE tokenizer plus autoregressive (SDformer-ar) or masked (SDformer-m) Transformers over the tokens.

Problem

Diffusion-based time series generators are slow at inference and their generated sample quality can still be improved.

Approach

Replace distance-based codebook lookup in VQ-VAE with similarity-based (inner product on unit-normalized vectors) quantization, then model the resulting discrete time series tokens with an autoregressive or masked Transformer, using random replacement as augmentation for the AR variant.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use SDformer-ar instead of Diffusion-TStime-series generation · StocksDiscriminative Score: 0.067±.015 -> 0.010±.006Helped
use SDformer-ar instead of Diffusion-TStime-series generation · ETThContext-FID Score: 0.116±.010 -> 0.008±.001Helped
use SDformer-ar instead of Diffusion-TStime-series generation · SinesDiscriminative Score: 0.006±.007 -> 0.006±.004No clear effect
replace similarity-driven quantization with distance-driven quantization (ablation)SDformer-ar · ETThDiscriminative Score: 0.003±.001 -> 0.010±.005Hurt
replace similarity-driven quantization with distance-driven quantization (ablation)SDformer-m · StocksContext-FID Score: 0.034±.008 -> 0.123±.009Hurt
replace discrete tokens with continuous tokens (ablation)SDformer-ar · ETThDiscriminative Score: 0.003±.001 -> 0.145±.020Hurt
use continuous tokens and provide the actual first token at inference (ablation)SDformer-ar · EnergyDiscriminative Score: 0.006±.004 -> 0.277±.007Hurt
replace Transformer with LSTM (ablation)SDformer-ar · fMRIContext-FID Score: 0.015±.001 -> 0.018±.001Hurt
replace Transformer with LSTM (ablation)SDformer-ar · StocksDiscriminative Score: 0.010±.006 -> 0.008±.005Helped
increase model size from 1.4M to 44.9M (ablation)SDformer · EnergyDiscriminative Score: 0.149±.007 -> 0.006±.004Helped
use SDformer-s instead of Diffusion-TStime-series generation · StocksInference Time (s): 30.01 -> 2.59Helped
use SDformer-ar instead of Diffusion-TStime-series generation · MuJoCoDiscriminative Score: 0.008±.002 -> 0.008±.005No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: Sines Stocks ETTh MuJoCo Energy fMRI

Benchmarks: Sines Stocks ETTh MuJoCo Energy fMRI

Metrics: Discriminative Score Predictive Score Context-FID Score Inference time MSE for forecasting

Baselines: Diffusion-TS TimeGAN TimeVAE Diffwave DiffTime Cot-GAN

Builds on: VQ-VAE MaskGIT Diffusion-TS T2M-GPT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy