Researchers have developed a new training framework to enhance the robustness of convolutional neural networks against context shift, a phenomenon where diverse test-time environments degrade model performance. The proposed method uses spatial feature attribution to guide models towards object-relevant features rather than non-causal contextual cues. This involves a novel attribution variant called ContrastiveCAMs and a modified loss function, Context-Regularized Cross-Entropy (CR-CE), which suppresses the influence of contextual regions. The approach demonstrated consistent improvements in context-shift robustness across several datasets, including Waterbirds, Spawrious, and ImageNet variants. AI
IMPACT Enhances the reliability of image classification models in real-world, unpredictable environments.
RANK_REASON The cluster contains an academic paper detailing a new method for improving machine learning models. [lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- Context-Regularized Cross-Entropy
- ContrastiveCAMs
- Hard-ImageNet
- ImageNet
- ImageNet-BG
- Spawrious
- Waterbirds
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →