Diffusion-Based Causal Representation LearningVise andre og tillknytning
2024 (engelsk)Inngår i: Entropy, E-ISSN 1099-4300, Vol. 26, nr 7, artikkel-id 556
Artikkel i tidsskrift (Fagfellevurdert) Published
Abstract [en]
Causal reasoning can be considered a cornerstone of intelligent systems. Having access to an underlying causal graph comes with the promise of cause–effect estimation and the identification of efficient and safe interventions. However, learning causal representations remains a major challenge, due to the complexity of many real-world systems. Previous works on causal representation learning have mostly focused on Variational Auto-Encoders (VAEs). These methods only provide representations from a point estimate, and they are less effective at handling high dimensions. To overcome these problems, we propose a Diffusion-based Causal Representation Learning (DCRL) framework which uses diffusion-based representations for causal discovery in the latent space. DCRL provides access to both single-dimensional and infinite-dimensional latent codes, which encode different levels of information. In a first proof of principle, we investigate the use of DCRL for causal representation learning in a weakly supervised setting. We further demonstrate experimentally that this approach performs comparably well in identifying the latent causal structure and causal variables.
sted, utgiver, år, opplag, sider
MDPI AG , 2024. Vol. 26, nr 7, artikkel-id 556
Emneord [en]
causal representation learning, diffusion models, diffusion-based representations, weak supervision
HSV kategori
Identifikatorer
URN: urn:nbn:se:kth:diva-351698DOI: 10.3390/e26070556ISI: 001277545500001PubMedID: 39056918Scopus ID: 2-s2.0-85199923243OAI: oai:DiVA.org:kth-351698DiVA, id: diva2:1888661
Merknad
QC 20250923
2024-08-132024-08-132025-09-23bibliografisk kontrollert