用语言模型生成可控因果数据,测试治疗效果估计能力
Language Models as Causal Effect Generators
- 用语言模型定义机制,构建可定制的因果模型
- 生成数千个反事实数据集,验证各类因果推断方法
- 适合评估因果算法、检测模型偏见,尤其关注个体化效应
本文提出序列驱动的结构因果模型(SD-SCM),通过用户定义的结构与语言模型定义的机制构建因果模型。该框架支持根据预设因果结构采样观测、干预和反事实分布。我们基于此设计新型基准,生成个体层面的反事实数据,用于测试平均、条件平均及个体治疗效应估计方法。创建包含数千个数据集的基准,测试多种主流估计方法。结果表明:(1) 因果方法优于非因果方法;(2) 即使最先进方法在个体化效应估计上仍表现不佳,说明该基准捕捉到因果推断的固有挑战。此外,该技术还可用于审计语言模型是否存在误导或歧视性因果效应。我们认为SD-SCM可广泛应用于需要可控因果结构的序列数据场景。
原文摘要 · Abstract (English)
In this work, we present sequence-driven structural causal models (SD-SCMs), a framework for specifying causal models with user-defined structure and language-model-defined mechanisms. We characterize how an SD-SCM enables sampling from observational, interventional, and counterfactual distributions according to the desired causal structure. We then leverage this procedure to propose a new type of benchmark for causal inference methods, generating individual-level counterfactual data to test treatment effect estimation. We create an example benchmark consisting of thousands of datasets, and test a suite of popular estimation methods for average, conditional average, and individual treatment effect estimation. We find under this benchmark that (1) causal methods outperform non-causal methods and that (2) even state-of-the-art methods struggle with individualized effect estimation, suggesting this benchmark captures some inherent difficulties in causal estimation. Apart from generating data, this same technique can underpin the auditing of language models for (un)desirable causal effects, such as misinformation or discrimination. We believe SD-SCMs can serve as a useful tool in any application that would benefit from sequential data with controllable causal structure.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。