arXiv reaDer
RotoGBML: 勾配ベースのメタ学習のための分散外一般化に向けて
RotoGBML: Towards Out-of-Distribution Generalization for Gradient-Based Meta-Learning
勾配ベースのメタ学習 (GBML) アルゴリズムは、学習したメタ知識を転送することで新しいタスクにすばやく適応できますが、すべてのタスクが同じ分布 (分布内、ID) から来ると想定しています。ただし、現実の世界では、タスクが異なるディストリビューションに由来するアウト オブ ディストリビューション (OOD) 一般化の問題に悩まされることがよくあります。 OOD はタスク勾配の大きさと方向の不一致を悪化させ、GBML が各ミニバッチのタスク勾配の合計を最小化することによってメタ知識を最適化するという課題をもたらします。この問題に対処するために、OOD タスク勾配を均質化する新しいアプローチである RotoGBML を提案します。 RotoGBML は、再重み付けされたベクトルを使用して、さまざまな大きさを共通のスケールに動的にバランスさせ、回転行列を使用して、競合する方向を互いに近づけて回転させます。オーバーヘッドを削減するために、ネットワーク パラメーターではなく機能を使用して勾配を均一化します。これに基づいて、非因果的特徴 (背景など) の介入を回避するために、不変な因果的特徴 (オブジェクトの輪郭など) を抽出するための不変自己情報 (ISI) モジュールも提案します。最後に、タスクの勾配は、これらの不変の因果的特徴に基づいて均質化されます。実験では、RotoGBML がさまざまな少数ショット画像分類ベンチマークで他の最先端の方法よりも優れていることが示されています。
Gradient-based meta-learning (GBML) algorithms are able to fast adapt to new tasks by transferring the learned meta-knowledge, while assuming that all tasks come from the same distribution (in-distribution, ID). However, in the real world, they often suffer from an out-of-distribution (OOD) generalization problem, where tasks come from different distributions. OOD exacerbates inconsistencies in magnitudes and directions of task gradients, which brings challenges for GBML to optimize the meta-knowledge by minimizing the sum of task gradients in each minibatch. To address this problem, we propose RotoGBML, a novel approach to homogenize OOD task gradients. RotoGBML uses reweighted vectors to dynamically balance diverse magnitudes to a common scale and uses rotation matrixes to rotate conflicting directions close to each other. To reduce overhead, we homogenize gradients with the features rather than the network parameters. On this basis, to avoid the intervention of non-causal features (e.g., backgrounds), we also propose an invariant self-information (ISI) module to extract invariant causal features (e.g., the outlines of objects). Finally, task gradients are homogenized based on these invariant causal features. Experiments show that RotoGBML outperforms other state-of-the-art methods on various few-shot image classification benchmarks.
updated: Sun Mar 12 2023 14:50:48 GMT+0000 (UTC)
published: Sun Mar 12 2023 14:50:48 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アソシエイト