About cookies on this site Our websites require some cookies to function properly (required). In addition, other cookies may be used with your consent to analyze site usage, improve the user experience and for advertising. For more information, please review your options. By visiting our website, you agree to our processing of information as described in IBM’sprivacy statement. To provide a smooth navigation, your cookie preferences will be shared across the IBM web domains listed here.
Publication
ECCV 2024
Conference paper
Causal Markov Blanket Representation Learning for Domain Generalization Prediction
Abstract
The pursuit of generalizable representations remains to be a dynamic field in the realm of machine learning and computer vision. Existing methods aim to secure invariant representations by either harnessing domain expertise or leveraging data from multiple domains. In this paper, we propose a novel approach that identifies the Causal Markov Blanket (CMB) representations and improves the Out-of-distribution prediction performance. To do so, we first introduce a structural causal model (SCM) to describe the underlying causal mechanisms governing the data generation process and distribution shifts. We theoretically prove that under our proposed SCM, the predictive mechanism using CMB representations remains invariant and predictive across domains. Subsequently, we propose a three-phase CMB representation learning procedure conforming to the proposed SCM assumptions. In comparison to state-of-the-art domain generalization methods, our approach exhibits robustness and adaptability under distribution shifts.