iTAG: Inverse Design for Natural Text Generation with Accurate Causal Graph Annotations
AI 摘要
iTAG通过逆向设计,生成带精确因果图标注的自然文本,提升文本因果发现算法的benchmark。
主要贡献
- 提出iTAG方法,提高因果图标注精度和文本自然度
- 使用Chain-of-Thought推理迭代优化概念选择
- 验证iTAG生成的数据可作为文本因果发现算法的实用benchmark
方法论
将文本生成视为逆问题,通过CoT推理迭代改进概念选择,保证生成文本的因果关系与目标图一致。
原文摘要
A fundamental obstacle to causal discovery from text is the lack of causally annotated text data for use as ground truth, due to high annotation costs. This motivates an important task of generating text with causal graph annotations. Early template-based generation methods sacrifice text naturalness in exchange for high causal graph annotation accuracy. Recent Large Language Model (LLM)-dependent methods directly generate natural text from target graphs through LLMs, but do not guarantee causal graph annotation accuracy. Therefore, we propose iTAG, which performs real-world concept assignment to nodes before converting causal graphs into text in existing LLM-dependent methods. iTAG frames this process as an inverse problem with the causal graph as the target, iteratively examining and refining concept selection through Chain-of-Thought (CoT) reasoning so that the induced relations between concepts are as consistent as possible with the target causal relationships described by the causal graph. iTAG demonstrates both extremely high annotation accuracy and naturalness across extensive tests, and the results of testing text-based causal discovery algorithms with the generated data show high statistical correlation with real-world data. This suggests that iTAG-generated data can serve as a practical surrogate for scalable benchmarking of text-based causal discovery algorithms.