Diffusion-Based Causal Representation LearningVisa övriga samt affilieringar
2024 (Engelska)Ingår i: Entropy, E-ISSN 1099-4300, Vol. 26, nr 7, artikel-id 556
Artikel i tidskrift (Refereegranskat) 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.
Ort, förlag, år, upplaga, sidor
MDPI AG , 2024. Vol. 26, nr 7, artikel-id 556
Nyckelord [en]
causal representation learning, diffusion models, diffusion-based representations, weak supervision
Nationell ämneskategori
Data- och informationsvetenskap
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
Anmärkning
QC 20250923
2024-08-132024-08-132025-09-23Bibliografiskt granskad