arXiv reaDer
FiT: パーソナライズされたフェデレーテッド画像分類のためのパラメーター効率の高い少数ショット転送学習
FiT: Parameter Efficient Few-shot Transfer Learning for Personalized and Federated Image Classification
最新の深層学習システムは、パーソナライゼーションや連合学習など、i) 少量のデータでの学習、および ii) 通信効率の高い分散トレーニング プロトコルをサポートする必要がある状況でますます展開されています。この作業では、転送学習 (固定された事前トレーニング済みバックボーンと微調整された FiLM アダプター レイヤー) とメタ学習 (自動構成された Naive Bayes 分類器とエピソードトレーニング) を使用して、ローショットで優れた分類精度を持つパラメーター効率の高いモデルを生成します。結果として得られるパラメーター効率は、少数ショットの学習、パーソナライズのための安価なモデルの更新、および通信効率の高いフェデレーテッド ラーニングを実現するための鍵となります。幅広いダウンストリーム データセットで FiT を実験し、ローショットで主要な Big Transfer (BiT) アルゴリズムよりも優れた分類精度を達成し、困難な VTAB-1k ベンチマークで最先端の精度を達成することを示します。更新可能なパラメータの 1% 未満です。最後に、モデルの更新サイズが重要なパフォーマンス指標であるモデルのパーソナライズやフェデレーテッド ラーニングなど、分散ローショット アプリケーションにおける FiT のパラメーター効率と優れた精度を示します。
Modern deep learning systems are increasingly deployed in situations such as personalization and federated learning where it is necessary to support i) learning on small amounts of data, and ii) communication efficient distributed training protocols. In this work, we develop FiLM Transfer (FiT) which fulfills these requirements in the image classification setting by combining ideas from transfer learning (fixed pretrained backbones and fine-tuned FiLM adapter layers) and meta-learning (automatically configured Naive Bayes classifiers and episodic training) to yield parameter efficient models with superior classification accuracy at low-shot. The resulting parameter efficiency is key for enabling few-shot learning, inexpensive model updates for personalization, and communication efficient federated learning. We experiment with FiT on a wide range of downstream datasets and show that it achieves better classification accuracy than the leading Big Transfer (BiT) algorithm at low-shot and achieves state-of-the art accuracy on the challenging VTAB-1k benchmark, with fewer than 1% of the updateable parameters. Finally, we demonstrate the parameter efficiency and superior accuracy of FiT in distributed low-shot applications including model personalization and federated learning where model update size is an important performance metric.
updated: Thu Feb 02 2023 20:22:18 GMT+0000 (UTC)
published: Fri Jun 17 2022 10:17:20 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アソシエイト