arXiv reaDer
ソースモデルアンサンブルを選択するための転送可能性メトリック
Transferability Metrics for Selecting Source Model Ensembles
転移学習におけるアンサンブル選択の問題に対処します。ソースモデルのプールが大きい場合、ターゲットトレーニングセットを微調整した後、ターゲットテストセットで最高のパフォーマンスが得られるモデルのアンサンブルを選択します。考えられるすべてのアンサンブルを微調整することは計算上禁止されているため、計算効率の高い転送可能性メトリックを使用して、ターゲットデータセットのパフォーマンスを予測することを目指しています。このタスク用に設計されたいくつかの新しい転送可能性メトリックを提案し、セマンティックセグメンテーションの挑戦的で現実的な転送学習セットアップでそれらを評価します。2つの異なるさまざまな画像ドメインをカバーする17のソースデータセットを検討することにより、ソースモデルの大規模で多様なプールを作成します。アーキテクチャ、および2つの事前トレーニングスキーム。このプールが与えられると、サブセットを自動的に選択して、特定のターゲットデータセットで適切に機能するアンサンブルを形成します。メソッドによって選択されたアンサンブルを、単一のソースモデルを選択する2つのベースラインと比較します。(1)メソッドと同じプールから。または(2)それぞれがアンサンブルと同様の容量を持つ大規模なソースモデルを含むプールから。 17のターゲットデータセットで平均すると、これらのベースラインをそれぞれ6.0%と2.5%の相対平均IoUで上回っています。
We address the problem of ensemble selection in transfer learning: Given a large pool of source models we want to select an ensemble of models which, after fine-tuning on the target training set, yields the best performance on the target test set. Since fine-tuning all possible ensembles is computationally prohibitive, we aim at predicting performance on the target dataset using a computationally efficient transferability metric. We propose several new transferability metrics designed for this task and evaluate them in a challenging and realistic transfer learning setup for semantic segmentation: we create a large and diverse pool of source models by considering 17 source datasets covering a wide variety of image domain, two different architectures, and two pre-training schemes. Given this pool, we then automatically select a subset to form an ensemble performing well on a given target dataset. We compare the ensemble selected by our method to two baselines which select a single source model, either (1) from the same pool as our method; or (2) from a pool containing large source models, each with similar capacity as an ensemble. Averaged over 17 target datasets, we outperform these baselines by 6.0% and 2.5% relative mean IoU, respectively.
updated: Thu Nov 25 2021 10:43:29 GMT+0000 (UTC)
published: Thu Nov 25 2021 10:43:29 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アソシエイト