apache / apache/iotdb

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

Đang mở
#17,301 3 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Java
Star
6.4k
Fork
1.2k
Merge trung bình
1 ngày 23 giờ
Pull request đã merge (30 ngày)
115

Mô tả

### 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!

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Hướng nghiên cứu

Bắt đầu với PR #16903 để tìm hiểu mẫu tích hợp Chronos2, sau đó đọc hướng dẫn đóng góp của AINode và kiểm tra model_info.py cùng entry point ForecastPipeline. Thêm thư mục và pipeline PatchTST, đăng ký model trong BUILTIN_HF_TRANSFORMERS_MODEL_MAP và xác minh rằng integration test vượt qua.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Đánh giá

Công nghệ
huggingface
Lĩnh vực
machine-learning
Loại issue
Tính năng
Độ khó
4/5
Thời gian dự kiến
3-5 ngày
Mức độ hoạt động
Ít trao đổi
Độ rõ ràng
Khá rõ ràng
Mức phù hợp với người mới
55/100

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.