arXiv reaDer
生涯にわたるテスト時間適応のための確率的フレームワーク
A Probabilistic Framework for Lifelong Test-Time Adaptation
テスト時間適応 (TTA) は、異なるターゲット ドメインからのテスト入力が与えられたときに、事前トレーニング済みのソース モデルを推論時に更新する問題です。ほとんどの既存の TTA アプローチは、ターゲット ドメインが静止している設定を想定しています。つまり、すべてのテスト入力が単一のターゲット ドメインから取得されます。ただし、多くの実際の設定では、テスト入力分布は、時間の経過とともに生涯にわたる/継続的な変化を示す可能性があります。さらに、既存の TTA アプローチには、信頼できる不確実性の見積もりを提供する機能もありません。これは、ソース ドメインとターゲット ドメインの間で分布のシフトが発生した場合に重要です。これらの問題に対処するために、確率論的アプローチを使用して生涯にわたる TTA を解決する PETAL (Probabilistic lifelong Test-time Adaptation with seLf-training prior) を提示し、(1) 教師モデルが生徒モデルの指数移動平均、および (2) ソース モデルを正則化子として使用して、推論時にモデルの更新を正則化します。生涯/継続的なTTA設定でのモデルのドリフトを防ぐために、無関係なパラメーターのみを復元することにより、エラーの蓄積を減らし、最近のドメインの知識を維持するのに役立つデータ駆動型パラメーター復元手法も提案します。予測エラー率だけでなく、Brier スコアや負の対数尤度などの不確実性ベースのメトリックに関しても、私たちの方法は、次のようなさまざまなベンチマークにわたるオンライン生涯テスト時間適応の現在の最先端技術よりも優れた結果を達成します。 CIFAR-10C、CIFAR-100C、ImageNetC、および ImageNet3DCC データセット。私たちのアプローチのソース コードは、https://github.com/dhanajitb/petal でアクセスできます。
Test-time adaptation (TTA) is the problem of updating a pre-trained source model at inference time given test input(s) from a different target domain. Most existing TTA approaches assume the setting in which the target domain is stationary, i.e., all the test inputs come from a single target domain. However, in many practical settings, the test input distribution might exhibit a lifelong/continual shift over time. Moreover, existing TTA approaches also lack the ability to provide reliable uncertainty estimates, which is crucial when distribution shifts occur between the source and target domain. To address these issues, we present PETAL (Probabilistic lifElong Test-time Adaptation with seLf-training prior), which solves lifelong TTA using a probabilistic approach, and naturally results in (1) a student-teacher framework, where the teacher model is an exponential moving average of the student model, and (2) regularizing the model updates at inference time using the source model as a regularizer. To prevent model drift in the lifelong/continual TTA setting, we also propose a data-driven parameter restoration technique which contributes to reducing the error accumulation and maintaining the knowledge of recent domains by restoring only the irrelevant parameters. In terms of predictive error rate as well as uncertainty based metrics such as Brier score and negative log-likelihood, our method achieves better results than the current state-of-the-art for online lifelong test-time adaptation across various benchmarks, such as CIFAR-10C, CIFAR-100C, ImageNetC, and ImageNet3DCC datasets. The source code for our approach is accessible at https://github.com/dhanajitb/petal.
updated: Tue Apr 04 2023 07:52:40 GMT+0000 (UTC)
published: Mon Dec 19 2022 18:42:19 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アソシエイト