arXiv reaDer
分布的にロバストな最適化によるラベルシフトへの対処
Coping with Label Shift via Distributionally Robust Optimisation
ラベルシフトの問題は、トレインとテストのラベル分布が一致しない教師あり学習設定を指します。ラベルシフトに対処する既存の作業は、通常、ラベルのないテストサンプルへのアクセスを前提としています。このサンプルは、テストラベルの分布を推定し、適切に再重み付けされた分類器をトレーニングするために使用できます。このアイデアを使用したアプローチは効果的であることが証明されていますが、ターゲットドメインにアクセスすることが常に実行可能であるとは限らないため、その範囲は限られています。さらに、モデルを複数のテスト環境に展開する場合は、再トレーニングを繰り返す必要があります。代わりに、幅広いファミリーからの任意のラベルシフトに対してロバストな単一の分類器を学ぶことができますか?この論文では、分布ロバスト最適化(DRO)に基づいて目的を最小化するモデルを提案することにより、この質問に答えます。次に、提案された目的を最適化するために、大規模な問題に合わせて調整された勾配降下近位ミラー上昇アルゴリズムを設計および分析します。 %、およびその収束を確立します。最後に、CIFAR-100とImageNetでの実験を通じて、ラベルシフトが存在する設定で、私たちの手法が多くのベースラインにわたってパフォーマンスを大幅に改善できることを示します。
The label shift problem refers to the supervised learning setting where the train and test label distributions do not match. Existing work addressing label shift usually assumes access to an unlabelled test sample. This sample may be used to estimate the test label distribution, and to then train a suitably re-weighted classifier. While approaches using this idea have proven effective, their scope is limited as it is not always feasible to access the target domain; further, they require repeated retraining if the model is to be deployed in multiple test environments. Can one instead learn a single classifier that is robust to arbitrary label shifts from a broad family? In this paper, we answer this question by proposing a model that minimises an objective based on distributionally robust optimisation (DRO). We then design and analyse a gradient descent-proximal mirror ascent algorithm tailored for large-scale problems to optimise the proposed objective. %, and establish its convergence. Finally, through experiments on CIFAR-100 and ImageNet, we show that our technique can significantly improve performance over a number of baselines in settings where label shift is present.
updated: Tue Aug 17 2021 05:36:25 GMT+0000 (UTC)
published: Fri Oct 23 2020 08:33:04 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アソシエイト