本研究では、時系列予測モデルの性能を向上させるための新しいデータ拡張手法AutoTSAugを提案している。
まず、複数の事前学習済み予測モデルからなるモデルZooを用いて、訓練データ内の「境界サンプル」を特定する。これらの境界サンプルは、モデルZoo内の各モデルの予測精度のばらつきが大きいデータであり、モデルの学習が不安定になりやすい。
次に、変分マスクオートエンコーダ(V-MAE)を拡張モデルとして用い、強化学習(REINFORCE)アルゴリズムを適用することで、これらの境界サンプルを効果的に拡張する。具体的には、拡張データの予測精度のばらつきを報酬関数とすることで、モデルZoo全体の予測精度の安定性を高めるような拡張を行う。
実験の結果、AutoTSAugは従来の固定的なデータ拡張手法と比べて、大幅な性能向上を示した。特に、訓練データが不足している場合でも、AutoTSAugは顕著な効果を発揮することが確認された。
Başka Bir Dile
kaynak içeriğinden
arxiv.org
Önemli Bilgiler Şuradan Elde Edildi
by Haochen Yuan... : arxiv.org 09-11-2024
https://arxiv.org/pdf/2409.06282.pdfDaha Derin Sorular