用专家混合架构让时间序列模型突破百亿参数,高效预测更准。
Time-MoE: Billion-Scale Time Series Foundation Models with Mixture of Experts
- 采用稀疏专家混合设计,只激活部分网络提升效率
- 在3000亿时间点数据上训练出24亿参数模型,精度显著提升
- 适合需要高精度、低成本长时序预测的工业场景
时间序列预测的深度学习虽有长足进展,但预训练模型规模小、成本高,制约了实际应用中大模型的发展。为此,我们提出Time-MoE,一种可扩展的统一架构,能在保持高模型容量的同时降低推理开销。通过稀疏专家混合(MoE)机制,每个预测仅激活部分子网络,实现计算效率与模型能力的平衡。Time-MoE为仅解码器的Transformer模型,支持自回归生成和灵活的预测时长。我们在新构建的大规模数据集Time-300B上进行预训练,覆盖9个领域,包含超过3000亿个时间点。首次将时间序列基础模型扩展至24亿参数,显著提升预测精度。实验验证了时间序列领域的缩放定律:训练样本量与模型规模之间存在正向关系。在相同激活参数数或等量计算预算下,我们的模型均大幅优于密集模型。这些成果使Time-MoE成为具备卓越性能、效率与灵活性的领先解决方案。
原文摘要 · Abstract (English)
Deep learning for time series forecasting has seen significant advancements over the past decades. However, despite the success of large-scale pre-training in language and vision domains, pre-trained time series models remain limited in scale and operate at a high cost, hindering the development of larger capable forecasting models in real-world applications. In response, we introduce Time-MoE, a scalable and unified architecture designed to pre-train larger, more capable forecasting foundation models while reducing inference costs. By leveraging a sparse mixture-of-experts (MoE) design, Time-MoE enhances computational efficiency by activating only a subset of networks for each prediction, reducing computational load while maintaining high model capacity. This allows Time-MoE to scale effectively without a corresponding increase in inference costs. Time-MoE comprises a family of decoder-only transformer models that operate in an auto-regressive manner and support flexible forecasting horizons with varying input context lengths. We pre-trained these models on our newly introduced large-scale data Time-300B, which spans over 9 domains and encompassing over 300 billion time points. For the first time, we scaled a time series foundation model up to 2.4 billion parameters, achieving significantly improved forecasting precision. Our results validate the applicability of scaling laws for training tokens and model size in the context of time series forecasting. Compared to dense models with the same number of activated parameters or equivalent computation budgets, our models consistently outperform them by large margin. These advancements position Time-MoE as a state-of-the-art solution for tackling real-world time series forecasting challenges with superior capability, efficiency, and flexibility.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。