[AINode] Integrate PatchTST-FM-R1 as builtin forecasting model
- 主要言語
- 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