用单纯形约束提升时间序列多通道建模的泛化能力
FSMLP: Modelling Channel Dependencies With Simplex Theory Based Multi-Layer Perceptions In Frequency Domain
- 在频域构建单纯形约束的MLP层,抑制极端值导致的过拟合
- 在7个数据集上实现更优精度与效率,且可扩展性强
- 适合需要稳定多通道建模的时间序列预测任务
时间序列预测在网页数据分析、能源消耗预测和天气预报等领域至关重要。尽管多层感知机(MLP)轻量高效,能捕捉时序依赖,但在建模通道间依赖时易过拟合。本文基于Rademacher复杂度理论分析发现,时间序列中的极端值会加剧该问题。为此,提出新型单纯形MLP层(Simplex-MLP),其权重被约束于标准单纯形,促使模型学习更简单模式,从而降低对极端值的过拟合。基于此,提出频率单纯形MLP(FSMLP)框架,包含两种模块:单纯形通道式MLP(SCWM)用于捕捉通道间依赖,频率时序MLP(FTM)用于提取时序信息。理论分析表明,单纯形MLP的Rademacher复杂度上界低于标准MLP。在7个基准数据集上的实验验证了方法在预测精度和效率上的显著提升,并展现出优越可扩展性。此外,单纯形MLP还可增强其他使用通道式MLP的方法,减少过拟合,提升性能。代码已开源。
原文摘要 · Abstract (English)
Time series forecasting (TSF) plays a crucial role in various domains, including web data analysis, energy consumption prediction, and weather forecasting. While Multi-Layer Perceptrons (MLPs) are lightweight and effective for capturing temporal dependencies, they are prone to overfitting when used to model inter-channel dependencies. In this paper, we investigate the overfitting problem in channel-wise MLPs using Rademacher complexity theory, revealing that extreme values in time series data exacerbate this issue. To mitigate this issue, we introduce a novel Simplex-MLP layer, where the weights are constrained within a standard simplex. This strategy encourages the model to learn simpler patterns and thereby reducing overfitting to extreme values. Based on the Simplex-MLP layer, we propose a novel \textbf{F}requency \textbf{S}implex \textbf{MLP} (FSMLP) framework for time series forecasting, comprising of two kinds of modules: \textbf{S}implex \textbf{C}hannel-\textbf{W}ise MLP (SCWM) and \textbf{F}requency \textbf{T}emporal \textbf{M}LP (FTM). The SCWM effectively leverages the Simplex-MLP to capture inter-channel dependencies, while the FTM is a simple yet efficient temporal MLP designed to extract temporal information from the data. Our theoretical analysis shows that the upper bound of the Rademacher Complexity for Simplex-MLP is lower than that for standard MLPs. Moreover, we validate our proposed method on seven benchmark datasets, demonstrating significant improvements in forecasting accuracy and efficiency, while also showcasing superior scalability. Additionally, we demonstrate that Simplex-MLP can improve other methods that use channel-wise MLP to achieve less overfitting and improved performance. Code are available \href{https://github.com/FMLYD/FSMLP}{\textcolor{red}{here}}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。