Selective Cross-Correlation Consistency Loss for Out-of-Distribution Generalization

Zining Chen,Weiqiu Wang,Zhicheng Zhao,Fei Su,Aidong Men
DOI: https://doi.org/10.1109/icme57554.2024.10688222
2024-01-01
Abstract:Deep learning methods usually succeed in independent and identically distributed (IID) data distribution, but suffer from sharp performance decay in real-world out-of-distribution (OOD) data. OOD generalization emerges to alleviate the large distribution shift between source and target domains. Recently, domain-invariant learning has boosted the research on OOD generalization, but most methods indulge complex architectures and training strategies. Hence, we propose a simple yet effective Selective Cross-Correlation Consistency (SC 3 ) loss to align the cross-correlation matrix of features from identical categories. Specifically, we design the Semantic-Oriented Selection (SOS) algorithm in SC 3 loss to eliminate negative effects on spurious channels. Extensive experiments demonstrate that the SC 3 loss achieves superior performance on multiple OOD scenarios, including domain generalization (DG) and single domain gener-alization (SDG) tasks. Also, our loss consumes negligible computational resource which conforms to real-world applications. Source code is available at https://github.com/znchen666/SC3.
What problem does this paper attempt to address?