跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

統一零樣本時間序列預測:基於 Darts 的基模型框架

文章摘要

Darts 自2020年發佈以來,已成為廣泛使用的開源時間序列分析庫。近期,一系列基模型在零樣本預測上取得了精度提升,預示着從定製模型訓練向預訓練通用預測器的範式轉變。然而,這些模型常以孤立包的形式發佈,接口碎片化且與常用工具互操作性有限。Darts 開發了統一的 FoundationModel 類集合(支持 Chronos-2、TimesFM 2.5、TiRex、PatchTST-FM),提供了標準化的全週期預測接口,最小化外部依賴,將基模型集成到生態系統中。現有 Darts 管道只需更改名稱即可使用基模型;新管道可實現零樣本或微調預測、不確定性估計和回測,結合數據處理與評估工具,全部在統一框架內完成。

來源arXiv Machine Learning作者: Zhihao Dai, Dennis Bader, Alain Gysi
統一零樣本時間序列預測:基於 Darts 的基模型框架
報告錯誤

更正渠道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

自2020年首次發佈以來,Darts 已成為廣泛使用的開源 Python 時間序列分析庫。它提供了豐富的模型和工具,幫助研究人員和工程師處理預測任務。近年來,時間序列基礎模型(如 Chronos-2、TimesFM 2.5、TiRex 和 PatchTST-FM)在零樣本預測中取得了顯著進展,聲稱在多個基準上超越了傳統的定製模型。這些模型通過在大規模數據集上進行預訓練,能夠直接應用於未見過的序列,無需針對特定任務重新訓練,從而大大降低了使用門檻。

然而,這些基礎模型通常以獨立的 Python 包形式發佈,每個模型都有自己的 API 和依賴項,導致接口碎片化。例如,用户可能需要學習不同的數據格式、預處理步驟和預測調用方式。此外,這些模型與現有工具(如數據加載、評估和可視化)的互操作性有限,使得在完整管線中進行聯合評估和集成變得困難。這阻礙了研究人員和從業者充分比較和利用這些模型。

為了解決這一挑戰,Darts 團隊開發了一個統一的 FoundationModel 類集合。該集合將上述四個基礎模型封裝在標準化接口下,提供了完整的預測生命週期支持,包括數據預處理、模型加載、預測、不確定性估計和回測。關鍵是,所有操作都在 Darts 的統一框架內完成,外部依賴極少,用户只需安裝 Darts 即可開始使用。

現在,現有的 Darts 管道僅需將模型名稱從 'ARIMA' 或 'Transformer' 更改為 'Chronos-2' 等,即可切換到基礎模型,無需修改其他代碼。新管道則可以利用基礎模型進行零樣本預測,或通過微調適應特定數據集。此外,不確定性估計和回測功能也直接可用,結合 Darts 內置的數據處理和評估工具,形成了完整的預測工作流。

這一進展對時間序列分析和預測領域具有重要意義。它降低了基礎模型的使用門檻,促進了更廣泛的實驗和比較。研究人員可以輕鬆評估不同模型在相同條件下的表現,而從業者可以快速將最先進的預測能力集成到現有系統中。隨着基礎模型數量的增加,這種統一框架將成為推動該領域標準化和效率提升的關鍵基礎設施。

展開要點與分析

文章情報

投資人進階

要點

  • Darts 推出統一接口,支持 Chronos-2、TimesFM 2.5、TiRex、PatchTST-FM
  • 實現零樣本預測、微調、不確定性估計和回測
  • 現有管道只需更改名稱即可切換至基模型,無需重構代碼

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。