arXiv reaDer
分布外予測のための分布シフト反転
Distribution Shift Inversion for Out-of-Distribution Prediction
機械学習社会は、統一された予測子または不変特徴表現を検索することによって、トレーニング分布とテスト分布の間の分布シフトに対処する、無数の分布外 (OoD) アルゴリズムの出現を目の当たりにしてきました。ただし、目に見えないテスト セットの分布シフトを直接緩和するタスクは、トレーニング段階ではテスト分布が利用できないため、トレーニングとテスト分布の間の分布トランスレーター マッピングをトレーニングすることが不可能であるため、ほとんど調査されていません。このペーパーでは、ディストリビューション トランスレータのトレーニングのためにディストリビューションをテストするという要件を回避し、ディストリビューションの変換を OoD 予測に役立てる方法を検討します。私たちは、ポータブルな分布シフト反転アルゴリズムを提案します。このアルゴリズムでは、予測モデルに入力される前に、OoD テスト サンプルが最初に追加のガウス ノイズと線形結合され、次にソース分布のみでトレーニングされた拡散モデルを使用してトレーニング分布に戻されます。 。理論的分析により、私たちの方法の実現可能性が明らかになります。マルチドメイン汎化データセットと単一ドメイン汎化データセットの両方に関する実験結果は、私たちの方法が一般的に使用される広範囲の OoD アルゴリズムにプラグインされた場合に全体的なパフォーマンスの向上をもたらすことを示しています。
Machine learning society has witnessed the emergence of a myriad of Out-of-Distribution (OoD) algorithms, which address the distribution shift between the training and the testing distribution by searching for a unified predictor or invariant feature representation. However, the task of directly mitigating the distribution shift in the unseen testing set is rarely investigated, due to the unavailability of the testing distribution during the training phase and thus the impossibility of training a distribution translator mapping between the training and testing distribution. In this paper, we explore how to bypass the requirement of testing distribution for distribution translator training and make the distribution translation useful for OoD prediction. We propose a portable Distribution Shift Inversion algorithm, in which, before being fed into the prediction model, the OoD testing samples are first linearly combined with additional Gaussian noise and then transferred back towards the training distribution using a diffusion model trained only on the source distribution. Theoretical analysis reveals the feasibility of our method. Experimental results, on both multiple-domain generalization datasets and single-domain generalization datasets, show that our method provides a general performance gain when plugged into a wide range of commonly used OoD algorithms.
updated: Wed Jun 14 2023 08:00:49 GMT+0000 (UTC)
published: Wed Jun 14 2023 08:00:49 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アソシエイト