arXiv reaDer
HyperTransformer:教師ありおよび半教師ありの少数ショット学習のためのモデル生成
HyperTransformer: Model Generation for Supervised and Semi-Supervised Few-Shot Learning
この作業では、サポートサンプルから直接畳み込みニューラルネットワーク(CNN)の重みを生成する、数ショット学習用のトランスベースモデルであるHyperTransformerを提案します。生成された小さなCNNモデルの特定のタスクへの依存は、大容量のトランスモデルによってエンコードされるため、大きなタスクスペースの複雑さを個々のタスクの複雑さから効果的に切り離します。私たちの方法は、固定されたユニバーサルタスクに依存しない埋め込みの学習が最適ではなく、タスクに関する情報がすべてのモデルパラメータを変調できる場合にパフォーマンスが向上する、小さなターゲットCNNアーキテクチャに特に効果的です。より大きなモデルの場合、最後のレイヤーを生成するだけで、エンドツーエンドの微分可能でありながら、最先端の方法で得られる結果よりも競争力のある、またはより良い結果を生み出すことができることがわかります。最後に、サポートセット内のラベルのないサンプルを利用し、数ショットのパフォーマンスをさらに向上させる半教師ありレジームにアプローチを拡張します。
In this work we propose a HyperTransformer, a transformer-based model for few-shot learning that generates weights of a convolutional neural network (CNN) directly from support samples. Since the dependence of a small generated CNN model on a specific task is encoded by a high-capacity transformer model, we effectively decouple the complexity of the large task space from the complexity of individual tasks. Our method is particularly effective for small target CNN architectures where learning a fixed universal task-independent embedding is not optimal and better performance is attained when the information about the task can modulate all model parameters. For larger models we discover that generating the last layer alone allows us to produce competitive or better results than those obtained with state-of-the-art methods while being end-to-end differentiable. Finally, we extend our approach to a semi-supervised regime utilizing unlabeled samples in the support set and further improving few-shot performance.
updated: Sat Jan 15 2022 00:56:22 GMT+0000 (UTC)
published: Tue Jan 11 2022 20:15:35 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アソシエイト