Auto Causal Inference

🚀 银行业自动因果推断

本项目借助大语言模型（LLM）的强大语义理解和推理能力，让用户只需指定“处理”和“结果”变量，即可自动推断变量角色并生成建议的因果图，解决了传统因果推断中变量角色识别依赖专业知识和手动构建的难题，实现了银行业因果推断的自动化，提高了实验效率和结果的透明度与可重复性。

🚀 快速开始

本项目提供了两种不同的代理架构来运行因果推断工作流，你可以根据需求选择运行方式。

运行 LangGraph

cd agent
python app.py

若要使用 LangGraph Studio 进行测试：

langgraph dev

UI 地址为：https://smith.langchain.com/studio/?baseUrl=http://127.0.0.1:2024

结合 Claude Desktop 运行 MCP

cd mcp_agent
python client.py

✨ 主要特性

• 自动化变量角色识别：利用大语言模型（LLM），用户只需指定“处理”和“结果”变量，即可自动推断变量角色和建议的因果图。
• 多工具集成：集成了 CausalNex、CausalTune、DoWhy 等工具，实现了从因果关系发现、估计到调优的全流程自动化。
• 可扩展性架构：提供了 LangGraph Agent 和 MCP Agent 两种不同的代理架构，支持同步或异步任务执行，便于扩展和模块化服务部署。
• 结果验证与修正：运行反驳测试来检查因果结构，并在测试不通过时提出修正解决方案。

📦 安装指南

环境要求

• Python 3.10
• Claude Desktop（用于运行 MCP）

安装依赖

pip install requirements.txt

💻 使用示例

基础用法

以下是一个简单的示例，展示了如何使用本项目进行因果推断：

# 示例输入
User asks: "Does offering a promotion increase digital product activation ?"

# 示例输出
# 因果关系
age -> promotion_offer;
age -> activated_ib;
income -> promotion_offer;
income -> activated_ib;
education -> promotion_offer;
education -> activated_ib;

region_code -> promotion_offer;

promotion_offer -> branch_visits;
branch_visits -> activated_ib;

promotion_offer -> customer_engagement;
activated_ib -> customer_engagement;

channel_preference -> activated_ib;
promotion_offer -> activated_ib

# 因果变量
{
  "confounders": ["age", "income", "education"],
  "mediators": ["branch_visits"],
  "effect_modifiers": ["channel_preference"],
  "colliders": ["customer_engagement"],
  "instruments": ["region_code"],
  "causal_graph": "...DOT format...",
  "dowhy_code": "...Python code..."
}

# 计算平均处理效应（ATE）
import dowhy
from dowhy import CausalModel

model = CausalModel(
    data=df,
    treatment='promotion_offer',
    outcome='activated_ib',
    common_causes=['age', 'income', 'education'],
    instruments=['region_code'],
    mediators=['branch_visits']
)

identified_model = model.identify_effect()
estimate = model.estimate_effect(identified_model, method_name='backdoor.propensity_score_matching')
print(estimate)

高级用法

以下是一些高级用法示例，包括模型调优和反驳测试：

# 模型调优
estimators = ["S-learner", "T-learner", "X-learner"]
# base_learners = ["random_forest", "neural_network"]

cd = CausalityDataset(data=df, treatment=state['treatment'], outcomes=[state["outcome"]],
                    common_causes=state['confounders'])
cd.preprocess_dataset()

estimators = ["SLearner", "TLearner"]
# base_learners = ["random_forest", "neural_network"]

ct = CausalTune(
    estimator_list=estimators,
    metric="energy_distance",
    verbose=1,
    components_time_budget=10, # in seconds trial for each model
    outcome_model="auto",
)

# run causaltune
ct.fit(data=cd, outcome=cd.outcomes[0])

print(f"Best estimator: {ct.best_estimator}")
print(f"Best score: {ct.best_score}")

# 反驳测试
refute_results = []
refute_methods = [
    "placebo_treatment_refuter",
    "random_common_cause",
    "data_subset_refuter"
]
for method in refute_methods:
    refute = model.refute_estimate(identified_estimand, estimate, method_name=method)
    refute_results.append({"method": method, "result": str(refute)})

pass_test = all("fail" not in r["result"].lower() for r in refute_results)

📚 详细文档

示例用例

分析变量列表

项目结构

auto_causal_inference/
├── agent/                 # LangGraph agent source code
│   ├── data/              # Sample data (bank.db)
│   ├── app.py             # Main entry point for LangGraph causal agent
│   ├── generate_data.py   # Data generation script for causal inference
│   ├── requirements.txt   # Dependencies for LangGraph agent
│   └── ...                # Other helper modules and notebooks
│
├── mcp_agent/             # MCP agent implementation
│   ├── data/              # Sample data (bank.db)
│   ├── server.py          # MCP causal inference server
│   ├── client.py          # MCP client to call the causal inference server
│   ├── requirements.txt   # Dependencies for MCP agent
│   └── ...                # Additional files
│
└── README.md              # This documentation file

结果分析

最佳估计器：backdoor.econml.metalearners.TLearner，得分：483.1930697900207

反驳测试通过：True。

与其他工具/方法的比较

🔧 技术细节

版本变更说明

• v1.1（当前版本）：集成了 CausalNex、CausalTune、反驳测试 等，使自动因果推断更加健壮。
• v1.0 (链接)：依赖 LLM 的强大语义理解和推理能力，实时识别整个因果结构（因果关系、因果变量等）。

工作原理

本项目展示了一个用于银行业用例的自动化因果推断管道，用户只需指定：

• 一个 处理 变量
• 一个 结果 变量

应用程序将自动执行以下步骤：

• 在数据库中搜索相关变量
• 使用 CausalNex 查找因果关系
• 识别因果变量
• 使用 DoWhy 执行因果模型
• 使用 CausalTune 寻找最佳估计器和基础学习器
• 运行反驳测试以检查因果结构
• 如果反驳测试未通过，提出修复解决方案（并进行重新运行循环）

📄 许可证

文档中未提及许可证相关信息。