cs.LGMar 11, 2026

Causal Concept Graphs in LLM Latent Space for Stepwise Reasoning

Authors: Md Muntaqim Meherab, Noor Islam S. Mohammad, Faiza Feroz

Organizations: Daffodil International University, Dhaka, Bangladesh · New York University, Brooklyn, NY, USA

Abstract

Sparse autoencoders can localize where concepts live in language models, but not how they interact during multi-step reasoning. We propose Causal Concept Graphs (CCG): a directed acyclic graph over sparse, interpretable latent features, where edges capture learned causal dependencies between concepts. We combine task-conditioned sparse autoencoders for concept discovery with DAGMA-style differentiable structure learning for graph recovery and introduce the Causal Fidelity Score (CFS) to evaluate whether graph-guided interventions induce larger downstream effects than random ones. On ARC-Challenge, StrategyQA, and LogiQA with GPT-2 Medium, across five seeds (n=15n{=}15 paired runs), CCG achieves \CFS=5.654±0.625\CFS=5.654\pm0.625, outperforming ROME-style tracing (3.382±0.2333.382\pm0.233), SAE-only ranking (2.479±0.1962.479\pm0.196), and a random baseline (1.032±0.0341.032\pm0.034), with p<0.0001p<0.0001 after Bonferroni correction. Learned graphs are sparse (5-6% edge density), domain-specific, and stable across seeds.

Explore similar work

CardsList
  1. CausalRAG2: Hierarchical Causal Knowledge Graph Design for RAG

    Feb 4, 2026Nengbo Wang, Tuo Liang, Vikash Singh +6KG ConstructionRetrieval-Augmented Generation

  2. LLM Explainability with Counterfactual Chains and Causal Graphs

    Jun 4, 2026Nirit Nussbaum-Hoffer, Nitay Calderon, Liat Ein-Dor +1LLM Reasoning with GraphsLLM Interpretability

  3. Stepwise Reasoning Enhancement for LLMs via External Subgraph Generation

    Jun 3, 2026Xin Zhang, Yang Cao, Baoxing Wu +2LLM GroundingLLM Reasoning with Graphs