DAG-SHAP Improves Feature Attribution in Causal Models

Qiheng Sun, Junxu Liu, Xiaokai Mao, Haocheng Xia, Jinfei Liu, Kui Ren, Haibo Hu· June 16, 2026 View original

Summary

DAG-SHAP is a new feature attribution method designed for models with directed acyclic graph (DAG) causal structures, addressing limitations of existing node-centric approaches. By attributing importance to individual feature edges, it effectively captures both the externality and exogenous influence of features, leading to more reasonable interpretations.

Traditional Shapley value-based feature attribution methods often struggle with complex feature interactions and causal relationships, even when a causal structure is explicitly provided. These methods typically adopt a node-centric perspective, assigning importance solely to individual features. This can lead to incomplete or misleading interpretations, as they often fail to simultaneously account for both the external impact and the independent, exogenous influence of features within a causal graph. To address these shortcomings, researchers have developed DAG-SHAP, a novel feature attribution method grounded in edge intervention. Instead of focusing on individual features (nodes), DAG-SHAP treats each feature edge within a Directed Acyclic Graph (DAG) as a distinct attribution object. This allows the method to appropriately capture both the externality and exogenous contributions of features. The paper also introduces an efficient approximation method for computing DAG-SHAP. Extensive experiments on both synthetic and real-world datasets confirm the effectiveness of DAG-SHAP in providing more accurate and insightful feature attributions compared to existing techniques.

Why it matters

For professionals building and interpreting complex AI models, especially in domains where causal understanding is critical, DAG-SHAP offers a more nuanced and accurate way to understand feature importance, leading to more trustworthy and explainable AI systems.

How to implement this in your domain

  1. 1Investigate DAG-SHAP for interpreting models where causal relationships between features are known or hypothesized.
  2. 2Apply the edge-intervention approach to gain a deeper understanding of feature interactions beyond individual feature importance.
  3. 3Utilize the provided approximation method to efficiently compute feature attributions in large-scale causal models.
  4. 4Compare DAG-SHAP's interpretations with existing node-centric attribution methods to identify more robust insights.

Who benefits

HealthcareFinanceManufacturingMarketingScientific Research

Key takeaways

  • DAG-SHAP is a new feature attribution method for models with causal structures.
  • It attributes importance to feature edges, capturing externality and exogenous influence.
  • This approach provides more reasonable and comprehensive interpretations than node-centric methods.
  • An efficient approximation method is available for practical application.

Original post by Qiheng Sun, Junxu Liu, Xiaokai Mao, Haocheng Xia, Jinfei Liu, Kui Ren, Haibo Hu

"arXiv:2606.15273v1 Announce Type: new Abstract: Shapley value-based feature attribution methods face challenges in scenarios involving complex feature interactions and causal relationships, even when a causal structure is provided. Existing methods typically adopt a node-centric…"

View on X

Originally posted by Qiheng Sun, Junxu Liu, Xiaokai Mao, Haocheng Xia, Jinfei Liu, Kui Ren, Haibo Hu on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses