arXiv reaDer
流通シフトの下での一般化のためのCrossNormとSelfNorm
CrossNorm and SelfNorm for Generalization under Distribution Shifts
従来の正規化手法(バッチ正規化やインスタンス正規化など)は、一般的かつ単純に、トレーニングデータとテストデータが同じ分布に従うことを前提としています。実際のアプリケーションでは分散のシフトが避けられないため、以前の正規化方法を使用して十分にトレーニングされたモデルは、新しい環境ではパフォーマンスが低下する可能性があります。分布シフトの下で一般化のロバスト性を改善するための新しい正規化方法を開発できますか?この論文では、CrossNormとSelfNormを提案することで質問に答えます。 CrossNormは、特徴マップ間でチャネルごとの平均と分散を交換してトレーニング分布を拡大し、SelfNormは注意を使用して統計を再調整し、トレーニング分布とテスト分布の間のギャップを埋めます。 CrossNormとSelfNormは互いに補完し合うことができますが、統計の使用法のさまざまな方向性を探っています。さまざまな分野(視覚と言語)、タスク(分類とセグメンテーション)、設定(教師ありおよび半教師あり)、および分布シフトタイプ(合成および自然)に関する広範な実験が有効性を示しています。コードはhttps://github.com/amazon-research/crossnorm-selfnormで入手できます。
Traditional normalization techniques (e.g., Batch Normalization and Instance Normalization) generally and simplistically assume that training and test data follow the same distribution. As distribution shifts are inevitable in real-world applications, well-trained models with previous normalization methods can perform badly in new environments. Can we develop new normalization methods to improve generalization robustness under distribution shifts? In this paper, we answer the question by proposing CrossNorm and SelfNorm. CrossNorm exchanges channel-wise mean and variance between feature maps to enlarge training distribution, while SelfNorm uses attention to recalibrate the statistics to bridge gaps between training and test distributions. CrossNorm and SelfNorm can complement each other, though exploring different directions in statistics usage. Extensive experiments on different fields (vision and language), tasks (classification and segmentation), settings (supervised and semi-supervised), and distribution shift types (synthetic and natural) show the effectiveness. Code is available at https://github.com/amazon-research/crossnorm-selfnorm
updated: Mon Aug 16 2021 02:12:31 GMT+0000 (UTC)
published: Thu Feb 04 2021 18:59:20 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アソシエイト