IBM, 오픈소스 시계열 모델 PatchTST-FM-r2 공개

IBM Research가 최신 시계열 파운데이션 모델 Granite Time Series PatchTST-FM-r2를 공개했다. 이 모델은 약 385M 파라미터 규모로, Apache 2.0 및 OpenMDW 1.0 이중 라이선스를 채택해 상업적 사용이 용이하다.

아키텍처 및 성능

기존 r1 버전 대비 내부 구조를 재설계하여 장단기 시간적 구조 포착 능력을 강화했다. 표준 Transformer 대신 Conformer 블록(Multi-head self-attention + Temporal convolution)을 도입해 단기 패턴은 컨볼루션으로, 장기 의존성은 어텐션으로 처리한다. 최대 8,192 스텝의 컨텍스트를 지원하며, 99개 퀀타일 기반의 확률적 예측과 결측치 보간 기능을 제공한다.

GIFT-Eval 벤치마크 결과, 재현 가능한 zero-shot 모델 중 CRPS 0.467, MASE 0.6846을 기록하며 오픈소스 라이선스 모델 중 1위를 차지했다. 일부 사전 학습된 대형 모델보다 작은 크기임에도 Chronos-2, Timer-S1 등 경쟁 모델보다 우수한 성능을 보였다.

배포 및 활용

모델 가중치, 아키텍처, 추론 파이프라인 코드가 모두 공개되었으며, granite-tsfm 라이브러리를 통해 Hugging Face Hub에서 쉽게 로드할 수 있다. 파인튜닝 없이 최근 시계열 이력만 입력하면 즉시 예측값과 불확실성 구간을 생성할 수 있다. IBM은 Confluent Cloud와 협력하여 Apache Flink 기반 스트리밍 애플리케이션에서 해당 모델을 직접 추론할 수 있는 Early Access 프로그램도 함께 제공한다.