用专家混合机制自动识别时间序列模式,提升预测模型泛化能力
Moirai-MoE: Empowering Time Series Foundation Models with Sparse Mixture of Experts
- 引入稀疏专家混合架构,让模型按令牌级别自适应处理不同模式
- 在39个数据集上实现比现有模型更高的零样本预测准确率
- 摆脱人为频率划分,适合研究时间序列建模与通用预测的学者
时间序列基础模型在零样本预测中表现优异,但实现统一训练仍具挑战。现有方法通过人为设定频率层级的模型专精来应对时间序列的高度异质性,例如Moirai使用多个输入/输出投影层以适配不同频率,TimesFM则依赖频率嵌入词典。然而,频率并非可靠的时间序列模式指标:不同频率可能呈现相似模式,相同频率也可能差异显著。此外,真实世界时间序列普遍存在非平稳性,在短时窗内分布即变化,频率级专精过于粗粒度,难以捕捉这种多样性。为此,本文提出Moirai-MoE,仅使用单一输入/输出投影层,将多样模式建模任务交由Transformer中的稀疏专家混合(MoE)机制完成。该设计降低对人工先验的依赖,实现令牌级别的自动专精。在39个数据集上的大量实验表明,Moirai-MoE在分布内与零样本场景下均优于现有基础模型。此外,研究还进行了深入的模型分析,揭示时间序列MoE基础模型的内在机制,为未来研究提供关键洞见。
原文摘要 · Abstract (English)
Time series foundation models have demonstrated impressive performance as zero-shot forecasters. However, achieving effectively unified training on time series remains an open challenge. Existing approaches introduce some level of model specialization to account for the highly heterogeneous nature of time series data. For instance, Moirai pursues unified training by employing multiple input/output projection layers, each tailored to handle time series at a specific frequency. Similarly, TimesFM maintains a frequency embedding dictionary for this purpose. We identify two major drawbacks to this human-imposed frequency-level model specialization: (1) Frequency is not a reliable indicator of the underlying patterns in time series. For example, time series with different frequencies can display similar patterns, while those with the same frequency may exhibit varied patterns. (2) Non-stationarity is an inherent property of real-world time series, leading to varied distributions even within a short context window of a single time series. Frequency-level specialization is too coarse-grained to capture this level of diversity. To address these limitations, this paper introduces Moirai-MoE, using a single input/output projection layer while delegating the modeling of diverse time series patterns to the sparse mixture of experts (MoE) within Transformers. With these designs, Moirai-MoE reduces reliance on human-defined heuristics and enables automatic token-level specialization. Extensive experiments on 39 datasets demonstrate the superiority of Moirai-MoE over existing foundation models in both in-distribution and zero-shot scenarios. Furthermore, this study conducts comprehensive model analyses to explore the inner workings of time series MoE foundation models and provides valuable insights for future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。