arXiv reaDer
バックボーンに注意: バックボーンの歪みを最小限に抑えて確実な物体検出を実現
Mind the Backbone: Minimizing Backbone Distortion for Robust Object Detection
ドメインのシフトに対して堅牢なオブジェクト検出器を構築することは、現実世界のアプリケーションにとって非常に重要です。従来のアプローチでは、事前トレーニングされたバックボーンを微調整するため、ディストリビューション内 (ID) データにオーバーフィットし、ディストリビューション外 (OOD) の一般化に役立つ特徴が歪曲される危険性があります。私たちは、歪みに対するバックボーンの脆弱性を測定する方法として相対勾配ノルム (RGN) を使用することを提案し、高い RGN が実際に低い OOD パフォーマンスと相関していることを示します。 RGN の分析では、興味深い発見が得られました。一部のバックボーンは微調整中に OOD の堅牢性を失いますが、他のバックボーンは、アーキテクチャーがパラメーターが初期モデルから大きく変更されないため、堅牢性を獲得します。これらの調査結果を踏まえて、両方のタイプのバックボーンの OOD 堅牢性を高めるレシピを紹介します。具体的には、調整されたバックボーンが一般化可能な機能を失わないように、勾配更新を最小限に抑えるための正則化とアーキテクチャの選択を調査します。私たちが提案した手法は相互に補完し合い、さまざまなアーキテクチャやデータセットのベースラインに比べて大幅な改善が見られます。コードは https://github.com/VisionLearningGroup/mind_back で入手できます。
Building object detectors that are robust to domain shifts is critical for real-world applications. Prior approaches fine-tune a pre-trained backbone and risk overfitting it to in-distribution (ID) data and distorting features useful for out-of-distribution (OOD) generalization. We propose to use Relative Gradient Norm (RGN) as a way to measure the vulnerability of a backbone to feature distortion, and show that high RGN is indeed correlated with lower OOD performance. Our analysis of RGN yields interesting findings: some backbones lose OOD robustness during fine-tuning, but others gain robustness because their architecture prevents the parameters from changing too much from the initial model. Given these findings, we present recipes to boost OOD robustness for both types of backbones. Specifically, we investigate regularization and architectural choices for minimizing gradient updates so as to prevent the tuned backbone from losing generalizable features. Our proposed techniques complement each other and show substantial improvements over baselines on diverse architectures and datasets. Code is available at https://github.com/VisionLearningGroup/mind_back.
updated: Mon May 15 2023 18:52:42 GMT+0000 (UTC)
published: Sun Mar 26 2023 14:50:43 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アソシエイト