English

Causal-Driven Feature Evaluation for Cross-Domain Image Classification

Machine Learning 2026-01-30 v2 Artificial Intelligence

Abstract

Out-of-distribution (OOD) generalization remains a fundamental challenge in real-world classification, where test distributions often differ substantially from training data. Most existing approaches pursue domain-invariant representations, implicitly assuming that invariance implies reliability. However, features that are invariant across domains are not necessarily causally effective for prediction. In this work, we revisit OOD classification from a causal perspective and propose to evaluate learned representations based on their necessity and sufficiency under distribution shift. We introduce an explicit segment-level framework that directly measures causal effectiveness across domains, providing a more faithful criterion than invariance alone. Experiments on multi-domain benchmarks demonstrate consistent improvements in OOD performance, particularly under challenging domain shifts, highlighting the value of causal evaluation for robust generalization.

Keywords

Cite

@article{arxiv.2601.20176,
  title  = {Causal-Driven Feature Evaluation for Cross-Domain Image Classification},
  author = {Chen Cheng and Ang Li},
  journal= {arXiv preprint arXiv:2601.20176},
  year   = {2026}
}

Comments

Preprint

R2 v1 2026-07-01T09:23:08.144Z