過去の流れから未来の数値を予測するAI公開、主要テストで首位独占
要点 3行
- 過去のデータから将来を予測する基盤モデル群「Timestral」が発表され、主要ベンチマークで最高水準を達成した。
- 500万から2億900万パラメータの4モデル構成で、予測処理の分離や合成データを活用した2段階学習を採用した。
- 開発された4モデルすべての重みがHugging Faceで、推論コードがGitHubでそれぞれ一般公開された。
研究チームは、確率的な時系列予測に特化した基盤モデル群「Timestral」を発表した。学習で調整される膨大な数値。「7B」は70億個の意味で、多いほど賢くなる傾向があるが動かすコストも増える。">パラメータ数は500万、2800万、1億2300万、2億900万の4種類が用意されている。性能評価では「GIFT-Eval」「TIME」「fev-bench」の各ベンチマークにおいて最高水準を記録し、上位2モデルがすべての指標で1位と2位を占めたという。
Timestralは、予測対象と文脈を分離して長期予測の学習効率を高める手法や、将来の外部要因を因果構造内でネイティブに扱う枠組みを取り入れた。学習面では、大規模な合成データのみで事前学習を実施したのち、実データと合成データを組み合わせた長文脈データで事後学習を行う2段階の手法を採用している。
なお、4種類すべてのモデルの重みはHugging Face上で、推論用のコードはGitHub上でそれぞれ公開されている。
この記事は arXiv cs.AIの発表を AI が日本語に要約したものです。正確な内容は原文をご確認ください。
アプリで開く