.png){kind=icon}
返回
NormAUG: Normalization-Guided Augmentation for Domain Generalization
.png){kind=icon}
OA
AI
DOI：10.1109/TIP.2024.3364516.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Deep learning has made significant advancements in supervised learning. However, models trained in this setting often face challenges due to domain shift between training and test sets, resulting in a significant drop in performance during testing. To address this issue, several domain generalization methods have been developed to learn robust and domain-invariant features from multiple training domains that can generalize well to unseen test domains. Data augmentation plays a crucial role in achieving this goal by enhancing the diversity of the training data. In this paper, inspired by the observation that normalizing an image with different statistics generated by different batches with various domains can perturb its feature, we propose a simple yet effective method called NormAUG (Normalization-guided Augmentation). Our method includes two paths: the main path and the auxiliary (augmented) path. During training, the auxiliary path includes multiple sub-paths, each corresponding to batch normalization for a single domain or a random combination of multiple domains. This introduces diverse information at the feature level and improves the generalization of the main path. Moreover, our NormAUG method effectively reduces the existing upper boundary for generalization based on theoretical perspectives. During the test stage, we leverage an ensemble strategy to combine the predictions from the auxiliary path of our model, further boosting performance. Extensive experiments are conducted on multiple benchmark datasets to validate the effectiveness of our proposed method.
Keyword:
Training
Training data
Data augmentation
Data models
Feature extraction
Ensemble learning
Task analysis
Normalization-guided augmentation
domain generalization
domain-shift
期刊
IF:
13.7
论文数:
1.0W
被引数:
8.4W
机构
引用论文
A cancer pharmacogenomic screen powering crowd-sourced advancement of drug combination prediction.png)
OAAI
IF0
OAAI分享收藏Contrastive-ACE: Domain Generalization Through Alignment of Causal Mechanisms对比-ACE: 通过因果机制的对齐进行域泛化OAAI
OAAI分享收藏分享收藏
查看更多内容