arXiv reaDer
シフトにとらわれない重みの正則化と最も近いソースプロトタイプによるテスト時間の適応の改善
Improving Test-Time Adaptation via Shift-agnostic Weight Regularization and Nearest Source Prototypes
このホワイトペーパーでは、ターゲットドメインからのラベルのないオンラインデータのみを使用してソースドメインで事前トレーニングされたモデルを調整し、ソースドメインとターゲットドメイン間の分散シフトによるパフォーマンスの低下を軽減する、新しいテスト時間適応戦略を提案します。ラベルのないオンラインデータを使用してモデルパラメータ全体を適応させることは、教師なしの目的からの誤った信号のために有害である可能性があります。この問題を軽減するために、テスト時間の適応中に、分布シフトに敏感なモデルパラメータを大幅に更新し、シフトに敏感でないモデルパラメータをわずかに更新することを奨励する、シフトにとらわれない重み正則化を提案します。この正則化により、高い学習率の利点を利用して、パフォーマンスを低下させることなく、モデルをターゲットドメインにすばやく適応させることができます。さらに、ソースとターゲットの機能を調整するための最も近いソースプロトタイプに基づく補助タスクを提示します。これにより、分散シフトが削減され、パフォーマンスがさらに向上します。私たちの方法は、さまざまな標準ベンチマークで最先端のパフォーマンスを示し、監視対象のメソッドよりも優れていることを示しています。
This paper proposes a novel test-time adaptation strategy that adjusts the model pre-trained on the source domain using only unlabeled online data from the target domain to alleviate the performance degradation due to the distribution shift between the source and target domains. Adapting the entire model parameters using the unlabeled online data may be detrimental due to the erroneous signals from an unsupervised objective. To mitigate this problem, we propose a shift-agnostic weight regularization that encourages largely updating the model parameters sensitive to distribution shift while slightly updating those insensitive to the shift, during test-time adaptation. This regularization enables the model to quickly adapt to the target domain without performance degradation by utilizing the benefit of a high learning rate. In addition, we present an auxiliary task based on nearest source prototypes to align the source and target features, which helps reduce the distribution shift and leads to further performance improvement. We show that our method exhibits state-of-the-art performance on various standard benchmarks and even outperforms its supervised counterpart.
updated: Sun Jul 24 2022 10:17:05 GMT+0000 (UTC)
published: Sun Jul 24 2022 10:17:05 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アソシエイト