arXiv reaDer
不安定な特徴を転送することによる安定した分類器の学習
Learning Stable Classifiers by Transferring Unstable Features
疑似相関の存在下で転移学習を研究します。ソースタスクで学習した安定した特徴抽出器を直接転送しても、ターゲットタスクのこれらのバイアスを排除できない可能性があることを実験的に示します。ただし、ソースタスクの不安定な機能とターゲットタスクの不安定な機能は直接関連していると仮定します。ソースタスクの不安定な機能をターゲット分類子に明示的に通知することで、ターゲットタスクのバイアスを正規化できます。具体的には、ソースタスクのさまざまなデータ環境を対比することにより、不安定な機能をエンコードする表現を導き出します。ターゲットタスクでは、この表現からデータをクラスター化し、すべてのクラスターで最悪の場合のリスクを最小限に抑えることで堅牢性を実現します。テキストと画像の両方の分類でメソッドを評価します。経験的結果は、私たちのアルゴリズムがターゲットタスクの堅牢性を維持でき、12の転送設定にわたって絶対精度で最高のベースラインを22.9%上回っていることを示しています。私たちのコードはhttps://github.com/YujiaBao/Tofuで入手できます。
We study transfer learning in the presence of spurious correlations. We experimentally demonstrate that directly transferring the stable feature extractor learned on the source task may not eliminate these biases for the target task. However, we hypothesize that the unstable features in the source task and those in the target task are directly related. By explicitly informing the target classifier of the source task's unstable features, we can regularize the biases in the target task. Specifically, we derive a representation that encodes the unstable features by contrasting different data environments in the source task. On the target task, we cluster data from this representation, and achieve robustness by minimizing the worst-case risk across all clusters. We evaluate our method on both text and image classifications. Empirical results demonstrate that our algorithm is able to maintain robustness on the target task, outperforming the best baseline by 22.9% in absolute accuracy across 12 transfer settings. Our code is available at https://github.com/YujiaBao/Tofu.
updated: Tue Jun 15 2021 02:41:12 GMT+0000 (UTC)
published: Tue Jun 15 2021 02:41:12 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アソシエイト