arXiv reaDer
DAFormer:ドメイン適応型セマンティックセグメンテーションのためのネットワークアーキテクチャとトレーニング戦略の改善
DAFormer: Improving Network Architectures and Training Strategies for Domain-Adaptive Semantic Segmentation
セマンティックセグメンテーションのために実世界の画像のピクセル単位の注釈を取得することはコストのかかるプロセスであるため、代わりにモデルをよりアクセスしやすい合成データでトレーニングし、注釈を必要とせずに実際の画像に適合させることができます。このプロセスは、教師なしドメイン適応(UDA)で研究されています。多くの方法が新しい適応戦略を提案していますが、それらはほとんどが時代遅れのネットワークアーキテクチャに基づいています。最近のネットワークアーキテクチャの影響は体系的に研究されていないため、最初にUDAのさまざまなネットワークアーキテクチャのベンチマークを行い、UDAセマンティックセグメンテーションに対するトランスフォーマーの可能性を新たに明らかにします。調査結果に基づいて、新しいUDAメソッドDAFormerを提案します。 DAFormerのネットワークアーキテクチャは、Transformerエンコーダとマルチレベルのコンテキストアウェア機能融合デコーダで構成されています。これは、トレーニングを安定させ、ソースドメインへの過剰適合を回避するための、3つのシンプルで重要なトレーニング戦略によって可能になります。(1)ソースドメインでのレアクラスサンプリングは、自己の確証バイアスを軽減することにより、疑似ラベルの品質を向上させます。一般的なクラスに向けたトレーニング、(2)Thing-Class ImageNet機能距離、および(3)学習率のウォームアップにより、ImageNet事前トレーニングからの機能転送が促進されます。 DAFormerは、UDAの大きな進歩を表しています。 GTA-to-Cityscapesで10.8mIoU、Synthia-to-Cityscapesで5.4 mIoUの最先端技術を向上させ、電車、バス、トラックなどの難しいクラスでも上手に学習できるようにします。実装はhttps://github.com/lhoyer/DAFormerで入手できます。
As acquiring pixel-wise annotations of real-world images for semantic segmentation is a costly process, a model can instead be trained with more accessible synthetic data and adapted to real images without requiring their annotations. This process is studied in unsupervised domain adaptation (UDA). Even though a large number of methods propose new adaptation strategies, they are mostly based on outdated network architectures. As the influence of recent network architectures has not been systematically studied, we first benchmark different network architectures for UDA and newly reveal the potential of Transformers for UDA semantic segmentation. Based on the findings, we propose a novel UDA method, DAFormer. The network architecture of DAFormer consists of a Transformer encoder and a multi-level context-aware feature fusion decoder. It is enabled by three simple but crucial training strategies to stabilize the training and to avoid overfitting to the source domain: While (1) Rare Class Sampling on the source domain improves the quality of the pseudo-labels by mitigating the confirmation bias of self-training toward common classes, (2) a Thing-Class ImageNet Feature Distance and (3) a learning rate warmup promote feature transfer from ImageNet pretraining. DAFormer represents a major advance in UDA. It improves the state of the art by 10.8 mIoU for GTA-to-Cityscapes and 5.4 mIoU for Synthia-to-Cityscapes and enables learning even difficult classes such as train, bus, and truck well. The implementation is available at https://github.com/lhoyer/DAFormer.
updated: Tue Mar 29 2022 10:16:35 GMT+0000 (UTC)
published: Mon Nov 29 2021 19:00:46 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アソシエイト