Constrained Posterior Sampling: Time Series Generation with Hard Constraints

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Narasimhan, Sai Shankar, Agarwal, Shubhankar, Rout, Litu, Shakkottai, Sanjay, Chinchali, Sandeep P.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909876885651456
author Narasimhan, Sai Shankar
Agarwal, Shubhankar
Rout, Litu
Shakkottai, Sanjay
Chinchali, Sandeep P.
author_facet Narasimhan, Sai Shankar
Agarwal, Shubhankar
Rout, Litu
Shakkottai, Sanjay
Chinchali, Sandeep P.
contents Generating realistic time series samples is crucial for stress-testing models and protecting user privacy by using synthetic data. In engineering and safety-critical applications, these samples must meet certain hard constraints that are domain-specific or naturally imposed by physics or nature. Consider, for example, generating electricity demand patterns with constraints on peak demand times. This can be used to stress-test the functioning of power grids during adverse weather conditions. Existing approaches for generating constrained time series are either not scalable or degrade sample quality. To address these challenges, we introduce Constrained Posterior Sampling (CPS), a diffusion-based sampling algorithm that aims to project the posterior mean estimate into the constraint set after each denoising update. Notably, CPS scales to a large number of constraints ($\sim100$) without requiring additional training. We provide theoretical justifications highlighting the impact of our projection step on sampling. Empirically, CPS outperforms state-of-the-art methods in sample quality and similarity to real time series by around 70\% and 22\%, respectively, on real-world stocks, traffic, and air quality datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2410_12652
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Constrained Posterior Sampling: Time Series Generation with Hard Constraints
Narasimhan, Sai Shankar
Agarwal, Shubhankar
Rout, Litu
Shakkottai, Sanjay
Chinchali, Sandeep P.
Machine Learning
Artificial Intelligence
Signal Processing
Generating realistic time series samples is crucial for stress-testing models and protecting user privacy by using synthetic data. In engineering and safety-critical applications, these samples must meet certain hard constraints that are domain-specific or naturally imposed by physics or nature. Consider, for example, generating electricity demand patterns with constraints on peak demand times. This can be used to stress-test the functioning of power grids during adverse weather conditions. Existing approaches for generating constrained time series are either not scalable or degrade sample quality. To address these challenges, we introduce Constrained Posterior Sampling (CPS), a diffusion-based sampling algorithm that aims to project the posterior mean estimate into the constraint set after each denoising update. Notably, CPS scales to a large number of constraints ($\sim100$) without requiring additional training. We provide theoretical justifications highlighting the impact of our projection step on sampling. Empirically, CPS outperforms state-of-the-art methods in sample quality and similarity to real time series by around 70\% and 22\%, respectively, on real-world stocks, traffic, and air quality datasets.
title Constrained Posterior Sampling: Time Series Generation with Hard Constraints
topic Machine Learning
Artificial Intelligence
Signal Processing
url https://arxiv.org/abs/2410.12652