arXiv:2502.13721cs.LGcs.CL2025-02被引 1

自动搜索更适合时间序列预测的Transformer结构,效果更优且训练高效。

Learning Novel Transformer Architecture for Time-series Forecasting

  • 用可微架构搜索方法自动寻找最优注意力与激活设计
  • 在多个时间序列数据集上超越现有模型精度
  • 适合需要高精度时序预测的研究与工业应用

尽管基于Transformer的模型在时间序列预测(TSP)任务中表现优异,但现有Transformer架构仍存在局限,且缺乏对替代架构的系统探索。为此,我们提出AutoFormer-TS,一个针对TSP任务定制的新型框架,采用全面的架构搜索空间。该框架引入一种改进的可微神经架构搜索方法(AB-DARTS),提升了对最优操作的识别能力。AutoFormer-TS系统性地探索了替代注意力机制、激活函数和编码操作,突破传统Transformer设计。大量实验表明,AutoFormer-TS在多个TSP基准测试中持续优于现有最先进模型,在保持合理训练效率的同时实现了更高预测精度。

原文摘要 · Abstract (English)

Despite the success of Transformer-based models in the time-series prediction (TSP) tasks, the existing Transformer architecture still face limitations and the literature lacks comprehensive explorations into alternative architectures. To address these challenges, we propose AutoFormer-TS, a novel framework that leverages a comprehensive search space for Transformer architectures tailored to TSP tasks. Our framework introduces a differentiable neural architecture search (DNAS) method, AB-DARTS, which improves upon existing DNAS approaches by enhancing the identification of optimal operations within the architecture. AutoFormer-TS systematically explores alternative attention mechanisms, activation functions, and encoding operations, moving beyond the traditional Transformer design. Extensive experiments demonstrate that AutoFormer-TS consistently outperforms state-of-the-art baselines across various TSP benchmarks, achieving superior forecasting accuracy while maintaining reasonable training efficiency.

时间序列Transformer架构搜索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。