Skip to yearly menu bar Skip to main content

Workshop: Causal Representation Learning

Causal Markov Blanket Representations for Domain Generalization Prediction

Naiyu Yin · Hanjing Wang · Tian Gao · Amit Dhurandhar · Qiang Ji

Keywords: [ Causal Markov Blanket representation learning; OOD Generalization ]


The pursuit of generalizable representations in the realm of machine learning and computer vision is a dynamic field of research. Typically, current methods aim to secure invariant representations by either harnessing domain expertise or leveraging data from multiple domains. In this paper, we introduce a novel approach that involves acquiring Causal Markov Blanket (CMB) representations to improve prediction performance in the face of distribution shifts. Causal Markov Blanket representations comprise the direct causes and effects of the target variable, rendering them invariant across diverse domains. To elaborate, our approach commences with the introduction of a novel structural causal model (SCM) equipped with latent representations, designed to capture the underlying causal mechanisms governing the data generation process. Subsequently, we propose a CMB representation learning framework that derives representations conforming to the proposed SCM. In comparison to state-of-the-art domain generalization methods, our approach exhibits robustness and adaptability under distribution shifts.

Chat is not available.