跳到主要內容
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
  • 實現零樣本預測、微調、不確定性估計和回測
  • 現有管道只需更改名稱即可切換至基模型,無需重構程式碼

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