arXiv reaDer
SITA: 単一イメージのテスト時間適応
SITA: Single Image Test-time Adaptation
Test-time Adaptation (TTA) では、ソース モデルが与えられた場合、それを適応させて、ソースとは異なるディストリビューションからのテスト インスタンスをより適切に予測することが目標です。重要なことに、TTA は、ソース モデルを微調整するために、ソース データやターゲット ディストリビューションからの追加のラベル付き/ラベルなしサンプルへのアクセスさえも想定していません。この作業では、SITA (Single Image Test-time Adaptation) と呼ばれるより実用的な設定で TTA を検討します。ここで、予測を行うとき、モデルは、文献で通常考えられているように、インスタンスのバッチではなく、特定の単一のテスト インスタンスにのみアクセスできます。これは、着信バッチを遅らせるのではなく、オンデマンドで推論が必要な現実的なシナリオや、バッチ処理の範囲がないエッジ デバイス (携帯電話など) で推論が行われているという現実的なシナリオによって動機付けられています。 SITA での適応プロセス全体は、推論時に発生するため、非常に高速である必要があります。これに対処するために、単一のフォワードパスのみを必要とする新しいアプローチ AugBN を提案します。市販のトレーニング済みモデルで使用して、分類タスクとセグメンテーション タスクの両方の単一インスタンスをテストできます。 AugBN は、ラベル保存変換を伴う 1 つのフォワード パスのみを使用して、指定されたテスト イメージから見えないテスト分布の正規化統計を推定します。 AugBN はバックプロパゲーションを含まないため、最近のテスト時間の適応方法と比較して大幅に高速です。 AugBN をさらに拡張して、アルゴリズムをハイパーパラメータなしにします。厳密な実験により、私たちのシンプルなアルゴリズムが、さまざまなデータセット、タスク、およびネットワーク アーキテクチャに対して大幅なパフォーマンスの向上を達成できることが示されています。
In Test-time Adaptation (TTA), given a source model, the goal is to adapt it to make better predictions for test instances from a different distribution than the source. Crucially, TTA assumes no access to the source data or even any additional labeled/unlabeled samples from the target distribution to finetune the source model. In this work, we consider TTA in a more pragmatic setting which we refer to as SITA (Single Image Test-time Adaptation). Here, when making a prediction, the model has access only to the given single test instance, rather than a batch of instances, as typically been considered in the literature. This is motivated by the realistic scenarios where inference is needed on-demand instead of delaying for an incoming batch or the inference is happening on an edge device (like mobile phone) where there is no scope for batching. The entire adaptation process in SITA should be extremely fast as it happens at inference time. To address this, we propose a novel approach AugBN that requires only a single forward pass. It can be used on any off-the-shelf trained model to test single instances for both classification and segmentation tasks. AugBN estimates normalization statistics of the unseen test distribution from the given test image using only one forward pass with label-preserving transformations. Since AugBN does not involve any back-propagation, it is significantly faster compared to recent test time adaptation methods. We further extend AugBN to make the algorithm hyperparameter-free. Rigorous experimentation show that our simple algorithm is able to achieve significant performance gains for a variety of datasets, tasks, and network architectures.
updated: Wed Sep 07 2022 17:24:26 GMT+0000 (UTC)
published: Sat Dec 04 2021 15:01:35 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アソシエイト