arXiv reaDer
微分可能な関数による動的なテスト時間の増強
Dynamic Test-Time Augmentation via Differentiable Functions
実世界でよく発生する分布シフトは、深層学習システムの精度を低下させるため、実用化にはロバスト性の向上が不可欠です。堅牢性を向上させるために、認識モデルを再トレーニングせずに認識しやすい画像を生成する画像強調方法を研究します。微分可能なデータ拡張技術に基づいており、多くの拡張画像からブレンド画像を生成して、分布シフト下での認識精度を向上させる新しい画像強調方法、DynTTA を提案します。標準的なデータ拡張に加えて、DynTTA にはディープ ニューラル ネットワーク ベースの画像変換も組み込まれており、堅牢性がさらに向上しています。 DynTTA は微分可能な関数で構成されているため、認識モデルの分類損失で直接トレーニングされます。 Vision Transformer や MLP-Mixer などのさまざまな分類モデルを使用して、広く使用されている画像認識データセットを実験します。 DynTTA は、きれいな画像の分類精度をほとんど低下させることなく堅牢性を向上させます。これは、既存の方法よりも優れた結果です。さらに、DynTTA を使用して分布シフトされたデータセットのトレーニング時間の増加を推定し、推定された増加で認識モデルを再トレーニングすると、ロバスト性が大幅に向上することを示します。
Distribution shifts, which often occur in the real world, degrade the accuracy of deep learning systems, and thus improving robustness is essential for practical applications. To improve robustness, we study an image enhancement method that generates recognition-friendly images without retraining the recognition model. We propose a novel image enhancement method, DynTTA, which is based on differentiable data augmentation techniques and generates a blended image from many augmented images to improve the recognition accuracy under distribution shifts. In addition to standard data augmentations, DynTTA also incorporates deep neural network-based image transformation, which further improves the robustness. Because DynTTA is composed of differentiable functions, it is directly trained with the classification loss of the recognition model. We experiment with widely used image recognition datasets using various classification models, including Vision Transformer and MLP-Mixer. DynTTA improves the robustness with almost no reduction in classification accuracy for clean images, which is a better result than the existing methods. Furthermore, we show that estimating the training time augmentation for distribution-shifted datasets using DynTTA and retraining the recognition model with the estimated augmentation significantly improves robustness.
updated: Wed Mar 15 2023 04:06:46 GMT+0000 (UTC)
published: Fri Dec 09 2022 06:06:47 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アソシエイト