arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.06337cs.CLcs.AIcs.LG

训练后能否将大语言模型转变为因果推理者?

Can Post-Training Transform LLMs into Causal Reasoners?

Shanghai Artificial Intelligence Library(上海人工智能图书馆);Fudan University(复旦大学);Tongji University(同济大学)

Junqi Chen, Sirui Chen, Chaochao Lu

AI总结:

本文通过CauGym数据集评估了训练后方法对LLM因果推理能力的提升,发现适当训练可使小型模型在因果推理任务中超越大模型。

摘要:

因果推断对于决策至关重要,但对非专家而言仍然具有挑战性。尽管大型语言模型(LLMs)在这一领域显示出潜力,但它们的精确因果估计能力仍然有限,训练后对这些能力的影响尚不明确。本文探讨了训练后能否增强LLM的因果推理能力。我们引入了CauGym,一个包含七个核心因果任务用于训练和五个多样测试集的综合数据集。使用此数据集,我们系统评估了五种训练后方法:SFT、DPO、KTO、PPO和GRPO。在五个领域内和四个现有基准测试中,我们的实验表明,适当训练后的小型LLM能够进行竞争性的因果推理,通常超越了更大的模型。我们的14B参数模型在CaLM基准测试中达到93.5%的准确率,相比OpenAI o3的55.4%。此外,训练后的LLM在现实世界条件下,如分布偏移和噪声数据中表现出强泛化能力和鲁棒性。这些发现提供了首个系统证据,表明有针对性的训练后处理可以产生可靠且稳健的基于LLM的因果推理者。我们的数据和GRPO模型可在https://github.com/OpenCausaLab/CauGym获取。

英文摘要:

Causal inference is essential for decision-making but remains challenging for non-experts. While large language models (LLMs) show promise in this domain, their precise causal estimation capabilities are still limited, and the impact of post-training on these abilities is insufficiently explored. This paper examines the extent to which post-training can enhance LLMs' capacity for causal inference. We introduce CauGym, a comprehensive dataset comprising seven core causal tasks for training and five diverse test sets. Using this dataset, we systematically evaluate five post-training approaches: SFT, DPO, KTO, PPO, and GRPO. Across five in-domain and four existing benchmarks, our experiments demonstrate that appropriate post-training enables smaller LLMs to perform causal inference competitively, often surpassing much larger models. Our 14B parameter model achieves 93.5% accuracy on the CaLM benchmark, compared to 55.4% by OpenAI o3. Furthermore, the post-trained LLMs exhibit strong generalization and robustness under real-world conditions such as distribution shifts and noisy data. Collectively, these findings provide the first systematic evidence that targeted post-training can produce reliable and robust LLM-based causal reasoners. Our data and GRPO-model are available at https://github.com/OpenCausaLab/CauGym.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑