Domain Generalization Guided by Gradient Signal to Noise Ratio of Parameters

Publication Date: 10/2/2023

Event: ICCV 2023

Reference: pp. 6177-6188, 2023

Authors: Mateusz Michalkiewicz, University of Queensland, NEC Laboratories America, Inc.; Masoud Faraki, NEC Laboratories America, Inc.; Xiang Yu, NEC Laboratories America, Inc.; Manmohan Chandraker, NEC Laboratories America, Inc.; Mahsa Baktashmotlagh, University of Queensland

Abstract: Overfitting to the source domain is a common issue in gradient-based training of deep neural networks. To compensate for the over-parameterized models, numerous regularization techniques have been introduced such as those based on dropout. While these methods achieve significant improvements on classical benchmarks such as ImageNet, their performance diminishes with the introduction of domain shift in the test set i.e. when the unseen data comes from a significantly different distribution. In this paper, we move away from the classical approach of Bernoulli sampled dropout mask construction and propose to base the selection on gradient-signal-to-noise ratio (GSNR) of network’s parameters. Specifically, at each training step, parameters with high GSNR will be discarded. Furthermore, we alleviate the burden of manually searching for the optimal dropout ratio by leveraging a meta-learning approach. We evaluate our method on standard domain generalization benchmarks and achieve competitive results on classification and face anti-spoofing problems.

Publication Link: https://openaccess.thecvf.com/content/ICCV2023/html/Michalkiewicz_Domain_Generalization_Guided_by_Gradient_Signal_to_Noise_Ratio_of_ICCV_2023_paper.html

arXiv Link: https://arxiv.org/pdf/2310.07361.pdf