arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.05830cs.LGcs.AI

Sparse-VQ Transformer:一种基于向量量化的无前馈网络框架,用于增强时间序列预测

Sparse-VQ Transformer: An FFN-Free Framework with Vector Quantization for Enhanced Time Series Forecasting

Alibaba Group(阿里巴巴集团);Meta Platforms, Inc.(元平台公司)

Yanjun Zhao, Tian Zhou, Chao Chen, Liang Sun, Yi Qian, Rong Jin

AI总结:

本文提出Sparse-VQ Transformer,用稀疏向量量化替代FFN层并结合RevIN,减少噪声影响和参数,在十个基准上MAE降低7.84%和4.17%,可集成于现有模型提升预测性能。

摘要:

时间序列分析对众多应用至关重要,而Transformer在该领域已日益突出。主流方法从NLP和CV中定制Transformer架构,利用分块(patching)技术将连续信号转换为片段。然而,时间序列数据由于显著的分布偏移和固有的噪声水平而具有独特的挑战性。为解决这两个挑战,我们引入了稀疏向量量化无前馈网络Transformer(Sparse-VQ)。我们的方法利用稀疏向量量化技术,结合反向实例归一化(RevIN),以减少噪声影响并捕获用于预测的充分统计量,作为Transformer架构中前馈层(FFN)的替代方案。我们的无FFN方法削减了参数数量,提高了计算效率并减少了过拟合。通过在十个基准数据集(包括新引入的CAISO数据集)上的评估,Sparse-VQ在单变量和多变量时间序列预测中分别以7.84%和4.17%的MAE降低超越了领先模型。此外,它可以无缝集成到现有的基于Transformer的模型中,以提升其性能。

英文摘要:

Time series analysis is vital for numerous applications, and transformers have become increasingly prominent in this domain. Leading methods customize the transformer architecture from NLP and CV, utilizing a patching technique to convert continuous signals into segments. Yet, time series data are uniquely challenging due to significant distribution shifts and intrinsic noise levels. To address these two challenges,we introduce the Sparse Vector Quantized FFN-Free Transformer (Sparse-VQ). Our methodology capitalizes on a sparse vector quantization technique coupled with Reverse Instance Normalization (RevIN) to reduce noise impact and capture sufficient statistics for forecasting, serving as an alternative to the Feed-Forward layer (FFN) in the transformer architecture. Our FFN-free approach trims the parameter count, enhancing computational efficiency and reducing overfitting. Through evaluations across ten benchmark datasets, including the newly introduced CAISO dataset, Sparse-VQ surpasses leading models with a 7.84% and 4.17% decrease in MAE for univariate and multivariate time series forecasting, respectively. Moreover, it can be seamlessly integrated with existing transformer-based models to elevate their performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑