跳转至

OLIVIA: Harmonizing Time Series Foundation Models with Power Spectral Density

会议: ICML 2026
arXiv: 2605.17340
代码: 待确认
领域: 时间序列 / 基础模型
关键词: 功率谱密度, 时间序列基础模型, 域适应, 注意力机制

一句话总结

OLIVIA 通过引入功率谱密度(PSD)驱动的协调机制——Harmonizer(基于 Householder 反射的正交二阶协调)和 HarmonicAttention(共鸣器低维交互)——显著改进了时间序列基础模型在异质数据上的预训练,在 TSLib 零样本 + GIFT-Eval + GluonTS 多基准上实现 SOTA。

研究背景与动机

领域现状:时间序列基础模型通过多领域大规模数据集合上的预训练学习统一通用表示——这个范式在 NLP 和 CV 已被证明有效。然而现有模型在处理异质时间序列时面临严峻挑战。

现有痛点:不同领域的时间序列展现显著不同的时间模式(周期结构、长期依赖)。这种多样性虽然是学习广泛适用时间知识的前提,但也为预训练带来困难——(1)优化层面:联合训练具有不同时间特征的数据通常导致收敛缓慢和次优;(2)表示学习层面:模型需要同时适应不兼容的时间结构,难以形成统一可转移的表示。

核心矛盾:现有基础模型通过架构模块化或容量专业化(混合专家、频率感知分块)实现域适应,但没有显式解决时间分布中的根本差异——即用信号处理中的 PSD 概念诊断和调和跨域谱差异。

本文目标:(1)以原则性方式理解和量化跨域时间异质性;(2)在大规模预训练中高效实现 PSD 一致性而不陷入直接的、不稳定的发散度最小化。

切入角度:normalized PSD 是数据集级描述符,通过捕捉频率方向上的时间变化分布反映底层二阶时间相关性结构;PSD 对全局时间平移不变、对局部时间错位相对鲁棒,是比较异质条件下采集信号的理想表示。

核心 idea:通过谱域中协调各数据集的 PSD 减少不匹配——从不可实现的直接发散度最小化重新表述为基于二阶时间相关性共享重参数化的结构性协调方法。

方法详解

整体框架

编码器-解码器架构融合两个核心创新——Harmonizer(PSD 驱动的变换模块,通过共享重参数化二阶相关性隐式协调谱特征,编码前对齐输入、解码后恢复原域)+ HarmonicFormer(Transformer 衍生主干,采用 HarmonicAttention 替代稠密 token 交互,在低维谐波交互空间执行自注意力)。

信息流:原始时间序列 → Harmonizer-Aligner(对齐到共享谱空间)→ HarmonicFormer 编码-解码 → Harmonizer-Restorer(恢复到原域)→ 预测输出。

关键设计

  1. Harmonizer:基于 Householder 反射的正交二阶协调:

    • 功能:通过学习正交矩阵 \(Q\) 将时间序列投影到共享规范谱空间,隐式协调跨数据集 PSD 一致性而无需显式发散度计算。
    • 核心思路:直接最小化 PSD 的 JSD 在大规模预训练中不可行(梯度噪声大、训练不稳定)。从 命题 1(存在共享正交矩阵 \(Q\) 使其前 \(r\) 列张成的子空间对所有数据集的二阶矩矩阵都不变,将协方差矩阵块对角化)出发,以共享重参数化二阶相关性实现 PSD 协调。Aligner \(\mathcal{X} = X Q^\top\)\(Q\) 通过 \(K\) 个 Householder 反射的积 \(Q = \prod_k H_k\)\(H_k = I - 2 V_k V_k^\top\) 参数化,保证 \(Q\) 始终在正交群上。Restorer 执行逆映射 \(Y = \mathcal{Y} Q\)
    • 设计动机:避免直接的不稳定发散度最小化,改为通过投影到共享子空间的结构对齐实现;双向设计保持信号完整性和能量守恒;Householder 反射的逐步构造使梯度流平稳。
  2. HarmonicAttention:通过共鸣器的低维谐波交互:

    • 功能:把标准 Transformer 的 \(\mathcal{O}(L^2 P)\) 稠密 token 交互降到 \(\mathcal{O}(L M P + M^2 P)\),其中 \(M\) 个共鸣器(\(M \ll L\))作紧凑中间体高效传导全局依赖。
    • 核心思路:命题 2 指出 Harmonizer 对齐后的二阶矩矩阵具有块对角结构 \(\Sigma_\mathcal{X} = \text{diag}(\Lambda, \Phi)\),token 的 Gram 矩阵可分解为主导低秩项 + 有界余差——为用紧凑的谐波模式近似稠密依赖提供理论依据。三阶段:(1)token 到共鸣器聚合 \(R^{(h)} = (A^{(h)})^\top \tilde{Z}^{(h)}\);(2)共鸣器间交互 \(\text{ResAct}(R^{(h)}) = \text{Softmax}_{\text{res}}(R^{(h)} (R^{(h)})^\top / \sqrt{P}) R^{(h)}\);(3)全局共鸣器投影 \(\text{Head}^{(h)} = A^{(h)} \text{ResAct}(R^{(h)})\)
    • 设计动机:通过共鸣器瓶颈调解交互,避免稠密 \(L \times L\) 注意力矩阵,实现长序列线性可扩展;共鸣器充当紧凑谐波模式集,自然编码共享子空间中的主导能量。
  3. HarmonicFormer + 训练策略:

    • 功能:在 Harmonizer 对齐基础上堆叠 HarmonicAttention 编码-解码器,形成可扩展且表达力强的预训练主干。
    • 核心思路:从 Harmonizer 输出的 PSD 一致表示出发,构建 Transformer 风格编码-解码框架,全部用 HarmonicAttention 替代标准多头自注意力。预训练采用多任务学习 + 两阶段策略(含掩码建模、回归等多信号)。
    • 设计动机:结构化低维谱交互 + Transformer 表达深度,使 Harmonizer 的理论优雅性能够通过 HarmonicFormer 的高效交互机制转化为实际性能收益。

实验关键数据

主实验(TSLib 零样本)

基准 指标 Olivia SEMPO Time-MoE_B Time-MoE_L Moirai_B
ETTh1 MSE 0.399 0.410 0.445 0.435 0.433
ETTh1 MAE 0.421 0.430 0.449 0.449 0.431
Weather MSE 0.247 0.248 0.279 0.318 0.312
Electricity MSE 0.188 0.196 0.207

消融与效率

配置 ETTh1 MSE 推理 (s) 模型大小 (M)
HarmonicAttention 0.399 43.051 5.1
w/o Harmonizer 0.472
Full Attention 0.472
Linear Attention 0.412
Nyström Attention 0.488

关键发现

  • Olivia 在 TSLib 零样本相比 SEMPO 实现平均 MSE 降低 2.7%,相比 Time-MoE 系列平均 MSE 降低 26.3%。
  • GluonTS 上对 SEMPO 11.6-32% NRMSE 改进,对 Time-MoE 86%+ 改进。
  • Harmonizer 消融移除后 MSE 显著恶化(0.399 → 0.472),验证 PSD 协调的核心价值。
  • HarmonicAttention 性能收益源于与 PSD 一致表示的结构匹配,而非单纯注意力容量。
  • Olivia 参数最小(5.1M vs SEMPO 6.5M、Time-MoE_B 113M)。

亮点与洞察

  • PSD 作为根本诊断工具:把功率谱密度系统引入时间序列基础模型的异质性诊断,相比泛泛而谈的"域适应"更具操作意义。
  • 二阶矩结构协调的优雅转化:通过命题 1 / 2 的双重理论,把看似无法直接优化的 PSD 发散度问题巧妙转化为二阶统计量的块对角化问题——展示了深度理论思考如何指导模型设计。
  • 可复用的低维交互范式:HarmonicAttention 通过"共鸣器瓶颈"实现高效全局依赖建模,对任何需要在长序列上应用自注意力的领域都有潜在价值。
  • 表示学习与计算效率的统一:通过 PSD 协调同时优化两者——Harmonizer 改善表示学习,HarmonicAttention 改善计算效率,相辅相成而非 trade-off。

局限与展望

  • 推理延迟略高(Householder 反射构造正交矩阵引入额外开销,43s vs SEMPO 8s);可探索更高效的正交参数化(QR 分解、Cayley 变换)。
  • 共鸣器数量超参数 \(M\) 与信号真实秩 \(r\) 的关系理论上有界,但实践中如何对齐、如何在不同数据集间变化未充分讨论。
  • 对异质下游任务(分类、异常检测)的适用性仍待验证。

相关工作与启发

  • vs SEMPO / Time-MoE / Moirai:通过架构模块化实现域泛化但缺乏对跨域根本性谱差异的显式处理;Olivia 通过 PSD 一致性约束显式对齐。
  • vs ROSE:ROSE 用谱掩蔽和自适应寄存器隔离域特定特征;Olivia 相反——通过协调使域特定变化聚集在正交补子空间。
  • vs 一般低秩注意力:Linear / Nyström 是通用的低秩近似;HarmonicAttention 的共鸣器由 PSD 对齐结构原则上推导,对时间序列特定结构更敏感。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 把 PSD 作为诊断工具系统引入基础模型设计;HarmonicAttention 与 Harmonizer 的整合原理深刻。
  • 实验充分度: ⭐⭐⭐⭐⭐ 覆盖两个大规模基准 + 额外 6 个 GluonTS 数据集 + 全面消融 + 效率分析,结果一致显著。
  • 写作质量: ⭐⭐⭐⭐ 论文架构清晰,两个命题与方法设计对应关系论证充分;偶有细节略过。
  • 价值: ⭐⭐⭐⭐⭐ 在时间序列基础模型这一前沿方向上提供理论驱动的突破;PSD 协调思想可迁移到其他多源异质数据场景。