arXiv reaDer
画像分類タスクのための実用的な転送可能性の推定
Practical Transferability Estimation for Image Classification Tasks
転送可能性の見積もりは、ソースモデル(またはソースタスク)をターゲットタスクに転送するときのパフォーマンスがどれだけ優れているかを予測するための転送学習における重要な問題です。最近の分析転送可能性メトリックは、ソースモデルの選択とマルチタスク学習に広く使用されています。主な課題は、クロスドメインクロスタスク設定の下で転送可能性の見積もりを堅牢にする方法です。最近提案されたOTCEスコアは、効率のオーバーヘッドを引き起こす補助タスクの転送エクスペリエンスの助けを借りて、ドメインとタスクの両方の違いを考慮することによってこの問題を解決します。この作業では、OTCEでのタスク差推定のロバスト性を劇的に改善し、補助タスクの必要性を排除する、JC-NCEスコアと呼ばれる実用的な転送可能性メトリックを提案します。具体的には、サンプル距離とラベル距離の両方を考慮した地上コストで最適な輸送問題を解決することにより、ソースデータとターゲットデータ間の共同対応を構築し、一致したラベルの負の条件付きエントロピーとして転送可能性スコアを計算します。データセット内およびデータセット間転送設定での広範な検証により、JC-NCEスコアがOTCEの補助タスクフリーバージョンよりもそれぞれ7%および12%優れており、他の既存の転送可能性メトリックよりも平均して堅牢であることが示されています。 。
Transferability estimation is an essential problem in transfer learning to predict how good the performance is when transferring a source model (or source task) to a target task. Recent analytical transferability metrics have been widely used for source model selection and multi-task learning. A major challenge is how to make transfereability estimation robust under the cross-domain cross-task settings. The recently proposed OTCE score solves this problem by considering both domain and task differences, with the help of transfer experiences on auxiliary tasks, which causes an efficiency overhead. In this work, we propose a practical transferability metric called JC-NCE score that dramatically improves the robustness of the task difference estimation in OTCE, thus removing the need for auxiliary tasks. Specifically, we build the joint correspondences between source and target data via solving an optimal transport problem with a ground cost considering both the sample distance and label distance, and then compute the transferability score as the negative conditional entropy of the matched labels. Extensive validations under the intra-dataset and inter-dataset transfer settings demonstrate that our JC-NCE score outperforms the auxiliary-task free version of OTCE for 7% and 12%, respectively, and is also more robust than other existing transferability metrics on average.
updated: Wed Jun 30 2021 10:26:37 GMT+0000 (UTC)
published: Sat Jun 19 2021 11:59:11 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アソシエイト