arXiv reaDer
単一ソース ドメインの一般化のための CNN 特徴マップ拡張
CNN Feature Map Augmentation for Single-Source Domain Generalization
堅牢で一般化可能な機械学習モデルを求めて、ドメイン汎化 (DG) がここ数年で大きな注目を集めています。 DG の目標は、トレーニング中に利用可能なデータ分布とは異なるデータ分布が与えられた場合でも、引き続き良好なパフォーマンスを発揮するモデルを作成することです。ディープ畳み込みニューラル ネットワーク (CNN) は、下流のコンピューター ビジョン タスクで優れたパフォーマンスを達成することができましたが、これまでに見たことのないデータ ドメインでは一般化できないことがよくあります。したがって、この研究では、データ分散シフトの下でも堅牢性を維持できるモデルの作成に焦点を当て、単一ソース DG 画像分類設定における畳み込みニューラル ネットワーク アーキテクチャの代替正則化手法を提案します。ソース データとターゲット データの間のドメイン シフトによって引き起こされる問題を軽減するために、CNN の中間特徴マップを強化することを提案します。具体的には、モデルを新しい拡張レイヤーに通過させて、モデルがトレーニング セットに過剰適合するのを防ぎ、クロスドメインの一般化を改善します。私たちの知る限り、これは DG 画像分類設定のこのような設定を提案した最初の論文です。 PACS、VLCS、Office-Home、TerraIncognita の DG ベンチマーク データセットでの実験により、ほとんどの場合においてモデルが最先端のアルゴリズムを上回る手法の有効性が検証されました。
In search of robust and generalizable machine learning models, Domain Generalization (DG) has gained significant traction during the past few years. The goal in DG is to produce models which continue to perform well when presented with data distributions different from the ones available during training. While deep convolutional neural networks (CNN) have been able to achieve outstanding performance on downstream computer vision tasks, they still often fail to generalize on previously unseen data Domains. Therefore, in this work we focus on producing a model which is able to remain robust under data distribution shift and propose an alternative regularization technique for convolutional neural network architectures in the single-source DG image classification setting. To mitigate the problem caused by domain shift between source and target data, we propose augmenting intermediate feature maps of CNNs. Specifically, we pass them through a novel Augmentation Layer} to prevent models from overfitting on the training set and improve their cross-domain generalization. To the best of our knowledge, this is the first paper proposing such a setup for the DG image classification setting. Experiments on the DG benchmark datasets of PACS, VLCS, Office-Home and TerraIncognita validate the effectiveness of our method, in which our model surpasses state-of-the-art algorithms in most cases.
updated: Mon Dec 04 2023 09:52:25 GMT+0000 (UTC)
published: Fri May 26 2023 08:48:17 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アソシエイト