IBM发布Granite时间序列集成模型,无需微调即可预测

IBM发布Granite时间序列集成模型,无需微调即可预测。该方案融合四个互补预训练基础模型,借鉴气象集合预报理念,提供开箱即用的点预测与分位数概率预测接口,有效降低单一模型选择风险。

IBM 发布 Granite 时间序列集成模型:四种架构融合,无需微调即可预测

IBM Research 团队近日在 Hugging Face 发布了 Granite Time Series Ensemble(Granite 时间序列集成模型)。该方案通过整合四个互补的预训练时间序列基础模型,提供了一套开箱即用的预测接口。开发者无需进行模型微调或手动调整组合权重,即可获得中位数点预测和分位数概率预测。

整合四模型无需微调

这一设计理念借鉴了气象预报中的“集合预报”方法(如北美多模型集合系统),旨在通过结合不同模型的视角来降低单一模型选择错误的风险。当面对工业信号、零售需求或长季节性数据等复杂场景时,单一模型往往难以兼顾所有特性,而集成模型则提供了更稳健的默认选择。

快速上手:安装与调用

用户可通过 pip 安装最新版本的 granite-tsfm 库,并直接从 Hugging Face Hub 加载集成配方。以下代码演示了如何使用 ETT-small 数据集进行预测:

pip install "granite-tsfm>=0.3.11"
import pandas as pd

from tsfm_public.models.ensemble.modeling_ensemble import QuantileEnsembleForecaster

# 加载示例数据
df = pd.read_csv(
    "https://raw.githubusercontent.com/zhouhaoyi/ETDataset/main/ETT-small/ETTh1.csv",
    parse_dates=["date"],
)

# 加载 IBM Granite 时间序列集成模型 R1
ensemble = QuantileEnsembleForecaster.from_pretrained(
    "ibm-granite/granite-timeseries-ensemble-r1",
)

# 执行预测:使用最后1024个时间点作为上下文,预测未来24步
result = ensemble(
    df.tail(1024),
    timestamp_column="date",
    target_columns=["OT"],
    context_length=1024,
    prediction_length=24,
)

print(result.prediction_outputs) # 点预测结果
print(result.quantile_outputs)   # 分位数预测结果

需要注意的是,该配方存储的是组合配置而非合并后的模型权重。首次运行时,系统会自动下载四个成员模型的检查点,因此需要预留额外的初始化时间。

输出结果包含两部分:prediction_outputs 提供中心趋势的点预测;quantile_outputs 提供默认范围为 0.1 至 0.9 的分位数预测。应用程序可利用这些分位数值构建预测区间,例如 0.1 和 0.9 分位数定义了名义覆盖率为 80% 的中心预测区间。

集成成员:四种不同的架构

默认的 Granite 时间序列集成配方结合了四个具有不同架构特性的预训练模型,以捕捉趋势、季节性和局部变化的多样性:

模型名称 架构类型
Granite PatchTST-FM-r1 Patched Transformer (带补丁的变压器)
Granite PatchTST-FM-r2 Patched Conformity (已修补的符合性架构)
Granite FlowState-r1.1 State Space Model (状态空间模型)
Granite TTM-r3 Tiny Time Mixers (小型时间混合器 / MLP Mixer)

由于各架构对数据的处理方式不同,它们的预测误差并不完全一致。根据预测组合领域的研究,结合多个互补模型可以有效降低依赖单一最强模型的风险,同时保持较高的整体精度。

两种聚合策略:概率集成与 IQR 集成

每个成员模型均返回分位数预测,集成框架通过两种主要规则将这些预测转化为最终的分布和中位数。

1. 默认推荐:概率集成 (Probabilistic Ensemble)

概率集成采用统一的概率聚合方式,赋予每个成员同等的影响力。它在概率空间中聚合各成员的量子预测,并从中提取所需的分位数。由于权重固定为相等,该方法无需针对新数据进行适配步骤,也没有需要调整的超参数。研究表明,在点预测中,简单的等重组合往往能与复杂的加权估计方法相媲美,这一理念同样适用于概率预测。

2. 自适应替代方案:IQR 集成 (IQR Ensemble)

IQR 集成根据成员预测的不确定性(四分位距)进行动态加权,使预测区间更窄的成员拥有更大的影响力。它近似计算每个成员在 25% 和 75% 百分位数周围的四分位距范围。权重因样本、目标变量和地平线步骤而异,且不依赖于历史预测错误。通过同位素回归确保合并后的分位数单调递增。

值得注意的是,更窄的区间并不一定意味着更高的准确性或更好的校准度,IQR 权重可能会偏向过度自信的成员。如需实验此方法,可通过更改子文件夹路径加载:

ensemble = QuantileEnsembleForecaster.from_pretrained(
    "ibm-granite/granite-timeseries-ensemble-r1",
    subfolder="recipes/granite-iqr",
)

性能评估:基于 GIFT-Eval 基准测试

IBM 团队在 97 个 GIFT-Eval 数据集配置上评估了该集成模型。评估过程中未对模型进行微调,也未使用测试集目标来选择或适应聚合方法。以下是截至 2026 年 10 月 20 日的排行榜成绩对比,指标包括平均绝对缩放误差 (MASE) 和连续 Ranked Probability Score (CRPS/MWQL),分数越低越好:

预测模型 MASE ↓ MASE 排名 ↓ CRPS/MWQL ↓ CRPS 排名 ↓
Granite Ensemble R1 Probabilistic 0.683 ★ 47.175 ★ 0.471 50.948
Granite Ensemble R1 IQR 0.684 48.773 0.471 47.619 ★
Granite PatchTST-FM-r2 0.685 51.371 0.467 ★ 48.918
Granite FlowState-r1.1 0.701 66.474 0.490 69.485
Granite PatchTST-FM-r1 0.717 74.794 0.488 68.134
Granite TTM-r3 0.724 83.062 0.520 88.959

★ 表示六个预测模型中的最佳值。

数据显示,两种集成配方在总体 MASE 和平均 MASE 排名上均优于其四个单独成员。其中,概率集成实现了最低的 MASE (0.683),略优于表现最好的单个成员 PatchTST-FM-r2 (0.685)。虽然与最佳单体相比差距较小,但集成的主要优势在于稳定性:开发者无需预先知道哪种架构最适合特定数据,即可获得可比甚至更好的总精度。

在 CRPS/MWQL 指标上,PatchTST-FM-r2 保持了最低总分,而 IQR 集成获得了最低的平均 CRPS 排名。这表明在不同的评估指标下,最优选择可能有所不同。

扩展性与应用建议

该集成框架不仅限于 IBM 自家的四个模型。Forecaster 通用接口允许接入兼容的外部模型。开发者可以更改成员组合或聚合方法,同时保持下游预测格式不变,无需重新训练或合并模型权重。这使得将集成模型融入现有的预测技术栈变得更加容易。

此外,仓库中还包含添加了其他 IBM Research 检查点的扩展配方。请注意,这些额外成员可能涉及非商业和研究用途的限制许可,使用前需查阅各自的模型卡。

对于实际应用,建议遵循以下评估流程:

  • 匹配用例:使用反映实际可用数据的历史预测窗口,确定所需的地平线和系列。
  • 横向对比:在同一组数据窗口上运行概率集成、IQR 集成以及一个强单体模型(如 PatchTST-FM-r2)。
  • 超越点误差:除了 MASE,还需评估分位数损失,并检查 0.1 到 0.9 区间的经验覆盖率是否符合名义上的 80%。
  • 衡量成本:记录运行时和内存使用情况,判断结合四个模型的收益是否足以抵消增加的计算开销。

目前存在的局限性包括:当前配方独立预测多变量通道,未利用跨通道信息;加载四个成员检查点会增加推理时间和内存占用;且聚合方法不保证在分布偏移下的校准性或鲁棒性,部署后仍需持续监控。

评论 0

0/500

评论需审核后展示,请文明发言

💬
还没有评论,来说两句

相关阅读