arXiv reaDer
ワンショットメタ学習におけるマルチタスク重み最適化のための同時摂動法
Simultaneous Perturbation Method for Multi-Task Weight Optimization in One-Shot Meta-Learning
メタ学習法は、データ量の少ない環境で新しいタスクにすばやく適応できる学習アルゴリズムを構築することを目的としています。このようなアルゴリズムの最も難しいベンチマークの 1 つは、ワンショット学習問題です。この設定では、多くのアルゴリズムが限られた量のトレーニング サンプルに関連する不確実性に直面し、オーバーフィッティングにつながる可能性があります。この問題は、モデルに追加情報を提供することで解決できます。これを行う最も効率的な方法の 1 つは、マルチタスク学習です。この論文では、標準的なメタ学習パイプラインの変更について調査します。提案された方法は、共通の損失関数でいくつかのメタトレーニングタスクからの情報を同時に利用します。損失関数におけるこれらのタスクの影響は、タスクごとの重みによって制御されます。重みの適切な最適化は、トレーニングとモデルの最終的な品質に大きな影響を与える可能性があります。メタトレイン タスクの重みを最適化するために、同時摂動確率近似 (SPSA) ファミリーのメソッドの使用を提案し、調査します。また、勾配ベースの方法と比較して、確率的近似の優位性を示します。提案されたマルチタスク修正は、ほとんどすべてのメタ学習方法に適用できます。 CIFAR-FS、FC100、miniImageNet、および tieredImageNet のワンショット学習ベンチマークで、モデルに依存しないメタ学習およびプロトタイプ ネットワーク アルゴリズムに対するこの変更の適用を研究します。これらの実験中に、Multi-Task Modification は元の方法よりも改善されていることが実証されました。マルチタスクの重みの最適化のためにこの論文で最初に採用された SPSA 追跡アルゴリズムは、最先端のメタ学習方法に匹敵する最大の精度向上を示しています。私たちのコードはオンラインで入手できます。
Meta-learning methods aim to build learning algorithms capable of quickly adapting to new tasks in low-data regime. One of the most difficult benchmarks of such algorithms is a one-shot learning problem. In this setting many algorithms face uncertainties associated with limited amount of training samples, which may result in overfitting. This problem can be resolved by providing additional information to the model. One of the most efficient ways to do this is multi-task learning. In this paper we investigate the modification of a standard meta-learning pipeline. The proposed method simultaneously utilizes information from several meta-training tasks in a common loss function. The impact of these tasks in the loss function is controlled by a per task weight. Proper optimization of the weights can have big influence on training and the final quality of the model. We propose and investigate the use of methods from the family of Simultaneous Perturbation Stochastic Approximation (SPSA) for optimization of meta-train tasks weights. We also demonstrate superiority of stochastic approximation in comparison to gradient-based method. The proposed Multi-Task Modification can be applied to almost all meta-learning methods. We study applications of this modification on Model-Agnostic Meta-Learning and Prototypical Network algorithms on CIFAR-FS, FC100, miniImageNet and tieredImageNet one-shot learning benchmarks. During these experiments Multi-Task Modification has demonstrated improvement over original methods. SPSA-Tracking algorithm first adapted in this paper for multi-task weight optimization shows the largest accuracy boost that is competitive to the state-of-the-art meta-learning methods. Our code is available online.
updated: Sun Oct 02 2022 19:39:39 GMT+0000 (UTC)
published: Mon Oct 25 2021 18:11:49 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト