用小波分解提升时间序列生成质量,更真实还原多尺度结构。
WaveletDiff: Multilevel Wavelet Diffusion For Time Series Generation
- 在小波系数上训练扩散模型,融合多分辨率信息
- 跨层级注意力机制实现时频尺度的自适应信息交换
- 保持频谱保真度,生成结果在5项指标上显著优于基线
时间序列广泛应用于医疗、金融、音频处理和气候科学等领域的预测、分类与因果推断任务,但高质量的时间序列数据集仍稀缺。合成生成可缓解此问题,但现有模型局限于时域或频域,难以还原真实时间序列固有的多尺度特性。我们提出WaveletDiff,一种直接在小波系数上训练的扩散模型框架,利用时间序列的多分辨率结构。模型在每一层分解上使用专用变压器,并引入跨层级注意力机制,通过自适应门控实现时频尺度间的选通信息交互;同时基于帕塞瓦尔定理对各层施加能量守恒约束,确保扩散过程中的频谱保真度。在能源、金融与神经科学领域的六个真实数据集上进行的全面测试表明,WaveletDiff在短时与长时序列上均持续优于当前最优的时域与频域生成方法,在五种不同评估指标中表现突出。例如,其判别得分与上下文FID得分平均比第二优基线低3倍。
原文摘要 · Abstract (English)
Time series are ubiquitous in many applications that involve forecasting, classification and causal inference tasks, such as healthcare, finance, audio signal processing and climate sciences. Still, large, high-quality time series datasets remain scarce. Synthetic generation can address this limitation; however, current models confined either to the time or frequency domains struggle to reproduce the inherently multi-scaled structure of real-world time series. We introduce WaveletDiff, a novel framework that trains diffusion models directly on wavelet coefficients to exploit the inherent multi-resolution structure of time series data. The model combines dedicated transformers for each decomposition level with cross-level attention mechanisms that enable selective information exchange between temporal and frequency scales through adaptive gating. It also incorporates energy preservation constraints for individual levels based on Parseval's theorem to preserve spectral fidelity throughout the diffusion process. Comprehensive tests across six real-world datasets from energy, finance, and neuroscience domains demonstrate that WaveletDiff consistently outperforms state-of-the-art time-domain and frequency-domain generative methods on both short and long time series across five diverse performance metrics. For example, WaveletDiff achieves discriminative scores and Context-FID scores that are $3\times$ smaller on average than the second-best baseline across all datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。