3 个月前

SOFTS:基于序列核心融合的高效多变量时间序列预测

SOFTS:基于序列核心融合的高效多变量时间序列预测

摘要

多变量时间序列预测在金融、交通管理、能源以及医疗等多个领域中发挥着至关重要的作用。近年来的研究表明,通道独立性有助于抵御分布漂移,但忽略了通道间的相关性,从而限制了性能的进一步提升。尽管已有若干方法通过引入注意力机制或混合器(mixer)等结构来捕捉通道相关性,但这些方法往往导致模型复杂度显著增加,或过度依赖通道间相关性,难以在存在分布漂移、尤其是通道数量较多的情况下取得理想效果。为解决上述问题,本文提出了一种高效的基于多层感知机(MLP)的模型——系列核心融合时间序列预测器(Series-cOre Fused Time Series forecaster, SOFTS),其核心创新在于引入了一种新型的STar Aggregate-Redistribute(STAR)模块。与传统方法通过分布式结构(如注意力机制)处理通道间交互不同,STAR采用集中式策略,显著提升了计算效率,并降低了对单个通道质量的依赖。该模块首先将所有时间序列聚合为一个全局核心表示,随后将该核心信息分发并融合至各通道的独立表示中,从而高效实现通道间的交互建模。实验结果表明,SOFTS在保持线性时间复杂度的前提下,显著优于现有最先进的方法。此外,本文还通过实验证明了STAR模块在多种不同预测模型中的广泛适用性。为促进后续研究与开发,我们已将代码公开发布于:https://github.com/Secilia-Cxy/SOFTS。

代码仓库

secilia-cxy/softs
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
time-series-forecasting-on-etth1-336-1SOFTS
MAE: 0.452
MSE: 0.480

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SOFTS:基于序列核心融合的高效多变量时间序列预测 | 论文 | HyperAI超神经