apache / apache/iotdb

[AINode] Integrate PatchTST-FM-R1 as builtin forecasting model

オープン
#17,301 コメント 3 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Java
スター
6.4k
フォーク
1.2k
平均マージ
1日 23時間
マージ済み PR(30日)
115

説明

### Search before asking

- [x] I searched in the [issues](https://github.com/apache/iotdb/issues) and found nothing similar.

### Motivation

Description:

Summary: This issue tracks the integration of IBM Research's [PatchTST-FM-R1](https://huggingface.co/ibm-research/patchtst-fm-r1) into IoTDB AINode's built-in model registry, following the contribution guide documented [here](https://iotdb.apache.org/UserGuide/latest-Table/AI-capability/AINode_Upgrade_apache.html#_6-contributing-open-source-time-series-large-models-to-iotdb-ainode).

Motivation: PatchTST-FM-R1 is a SOTA patch-based Transformer foundation model for time series forecasting. Adding it expands the range of zero-shot forecasting options available to IoTDB users without requiring model training.

Implementation Plan:

Create iotdb-core/ainode/iotdb/ainode/core/model/patchtst/ directory with a PatchTSTFMPipeline extending

ForecastPipeline
Register the model in BUILTIN_HF_TRANSFORMERS_MODEL_MAP in

model_info.py
Add integration test
Reference: PR #16903 (Chronos2 integration) as the implementation pattern.

### Solution

_No response_

### Alternatives

_No response_

### Are you willing to submit a PR?

- [x] I'm willing to submit a PR!

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

Chronos2の統合パターンについてはまずPR #16903から始め、次にAINode contribution guideを読み、model_info.pyとForecastPipelineのエントリポイントを調べます。PatchTSTディレクトリとパイプラインを追加し、モデルをBUILTIN_HF_TRANSFORMERS_MODEL_MAPに登録して、統合テストがパスすることを確認します。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
huggingface
領域
machine-learning
issue の種類
機能追加
難易度
4/5
見積もり時間
3〜5日
活発さ
静か
明瞭さ
おおむね明確
初心者へのやさしさ
55/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。