Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CausalMoE: A Billion-Scale Multimodal Foundation Model for Granger Causal Discovery with Pattern-Routed Heterogeneous Experts

About

Granger Causal Discovery (GCD) is fundamental for analyzing temporal dependencies in complex systems. However, existing neural GCD methods predominantly rely on a "one-size-fits-all" paradigm, struggling to capture distribution shifts and dynamic regime changes inherent in real-world time series. This often leads to entangled representations and spurious causal graphs. In this paper, we propose CausalMoE, a billion-scale multimodal Granger causal foundation model that explicitly models patch-level heterogeneity. CausalMoE introduces a Pattern-Routed Mixture of Heterogeneous Experts, which dynamically identifies latent temporal patterns and routes patches to specialized domain experts, effectively decoupling regime-specific mechanisms from shared dynamics. To ensure interpretable graph recovery, we design a Causality-Aware Self-Attention mechanism operating across variables, yielding sparse Granger causal graphs via proximal optimization. Furthermore, CausalMoE is the first to integrate LLMs and VLMs to align numerical signals with textual and visual priors, regularizing causal estimation in complex scenarios. Extensive experiments demonstrate that CausalMoE establishes a new state-of-the-art on fully supervised benchmarks, while effectively generalizing to few-shot settings where traditional methods fail.

Bo Liu, Di Dai, Jingwei Liu, Jiarui Jin, Xiaocheng Fang, Guangkun Nie, Hongyan Li, Shenda Hong• 2026

Related benchmarks

TaskDatasetResultRank
Causal DiscoveryDREAM3
Score (E.coli-1)84.5
17
Granger Causal DiscoveryVAR (20,500,20)
AUROC0.976
16
Granger Causal DiscoveryVAR(20,1000,5)
AUROC98.9
8
Granger Causal DiscoveryVAR (40,1000,20)
AUROC95.3
8
Granger Causal DiscoveryLorenz(20,1000,10)
AUROC98.6
8
Granger Causal DiscoveryLorenz (40, 1000, 20)
AUROC93.9
8
Granger Causal DiscoveryDREAM-4
Gene 1 Score0.829
8
Showing 7 of 7 rows

Other info

Follow for update