arXiv reaDer
Bhattacharyyaクラス分離可能性を使用した転送可能性の推定
Transferability Estimation using Bhattacharyya Class Separability
転移学習は、コンピュータービジョンで事前にトレーニングされたモデルを活用するための一般的な方法になっています。ただし、計算コストの高い微調整を実行しないと、特定のターゲットタスクに適した事前トレーニング済みのソースモデル、または逆に、事前トレーニング済みのソースモデルを簡単に適応できるタスクを定量化することは困難です。この作業では、ソースモデルとターゲットデータセット間の転送可能性を定量化するための新しい方法であるガウスバタチャリヤ係数(GBC)を提案します。最初のステップでは、すべてのターゲット画像をソースモデルで定義された特徴空間に埋め込み、クラスごとのガウス分布で表現します。次に、Bhattacharyya係数を使用してペアごとのクラス分離可能性を推定し、ソースモデルがターゲットタスクにどれだけうまく転送されるかを簡単かつ効果的に測定します。データセットとアーキテクチャの選択のコンテキストで、画像分類タスクでGBCを評価します。さらに、より複雑なセマンティックセグメンテーション転送可能性推定タスクの実験も実行します。 GBCは、セマンティックセグメンテーション設定のほとんどの評価基準で最先端の転送可能性メトリックを上回り、画像分類におけるデータセット転送可能性のトップメソッドのパフォーマンスと一致し、画像分類のアーキテクチャ選択問題で最高のパフォーマンスを発揮することを示します。
Transfer learning has become a popular method for leveraging pre-trained models in computer vision. However, without performing computationally expensive fine-tuning, it is difficult to quantify which pre-trained source models are suitable for a specific target task, or, conversely, to which tasks a pre-trained source model can be easily adapted to. In this work, we propose Gaussian Bhattacharyya Coefficient (GBC), a novel method for quantifying transferability between a source model and a target dataset. In a first step we embed all target images in the feature space defined by the source model, and represent them with per-class Gaussians. Then, we estimate their pairwise class separability using the Bhattacharyya coefficient, yielding a simple and effective measure of how well the source model transfers to the target task. We evaluate GBC on image classification tasks in the context of dataset and architecture selection. Further, we also perform experiments on the more complex semantic segmentation transferability estimation task. We demonstrate that GBC outperforms state-of-the-art transferability metrics on most evaluation criteria in the semantic segmentation settings, matches the performance of top methods for dataset transferability in image classification, and performs best on architecture selection problems for image classification.
updated: Wed Nov 24 2021 20:22:28 GMT+0000 (UTC)
published: Wed Nov 24 2021 20:22:28 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アソシエイト