CoBe: A Benchmark For Conversational Counterfactual Text Editing
Organizations: Causal Artificial Intelligence Lab Columbia University
Abstract
Counterfactual text editing is the ability to revise a given text in light of a hypothetical change while preserving facts not causally affected by that change. When asked what would have happened if a doctor had not recommended surgery, the model output should retain the patient's medical history and modify only the consequences of the decision. This requires causal understanding: a model must determine, without being explicitly instructed to, what events are affected by the hypothetical change and what should remain fixed. This skill is central to human reasoning, especially in safety-critical domains like healthcare or law. By contrast, the task of associational text generation merely generates text statistically associated with the altered condition. In the medical example, an associational response might make the patient's symptoms less severe because milder cases correlate with no surgery. Such responses produce a plausible scenario, but not the counterfactual version of the given one. Prior counterfactual-text datasets do not clearly separate these modes of reasoning. We introduce the Counterfactual Benchmark (CoBe), a testbed for models' zero-shot ability to counterfactually edit a given textual scenario when prompted to do so in a conversationally natural way. CoBe contains 2000 questions spanning healthcare, science and engineering, business, and everyday scenarios. We test a suite of frontier and smaller models, with frontier systems achieving an average score of . Our findings indicate that models frequently default to associational generation rather than preserving the causal structure of the original scenario. Moreover, performance does not scale consistently with either model size or output token length, suggesting that the failure may reflect model training objectives rather than insufficient model capacity.
Figures & tables
| Prior work | Ctf-editing 3 steps | Narrative scenarios | No causal structure | Open-ended generation | Expert/formal verification |
| TimeTravel | ✓ | ✓ | ✗ | ✓ | ✗ |
| CRASS | ✗ | ✓ | ✓ | ✗ | ✗ |
| CLadder | ✓ | ✗ | ✗ | ✗ | ✓ |
| CounterBench | ✓ | ✗ | ✗ | ✗ | ✓ |
| CoBe (our work) | ✓ | ✓ | ✓ | ✓ | ✓ |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Stages (1-4) | Output | Steps taken | Quality assurance and filtration |
|---|---|---|---|
| Counterfactual example collection | 425 candidate counterfactual examples | A cohort of student researchers constructed examples following the dataset schema in Sec. 3. Each example contained an original scenario, a counterfactual edit request, and explicit evaluation criteria for LLM answers. | All contributors had completed a graduate causal inference course. Each example was required to specify a coherent counterfactual intervention, the resulting edit, and criteria identifying which facts should remain fixed and which consequences should change. |
| Core-set selection | 106 diverse core examples crafted | The authors manually reviewed the 425 candidates and used them as a basis to craft a core set spanning diverse domains, event interactions, and causal structures. | We rejected candidates with ill-defined counterfactuals, ambiguous or unsound causal relationships, incoherent scenarios, or evaluation criteria that did not clearly distinguish valid from invalid edits. |
| Augmentation of core set | 25 candidate variations per core example | Each core example was augmented using LLMs to produce variations across diverse themes and domains while preserving the original causal structure and logical relationships among events. | Generation was repeated until 25 satisfactory variations were obtained for each core example. We rejected and regenerated variations that were incoherent or contained unsound evaluation criteria. |
| Final verification and filtering | 2,650 examples; 2,000 final examples retained after quality checks | The authors manually reviewed every set of 25 augmented examples and its associated evaluation criteria. | Every augmented set of 25 variations was checked for story coherence and for the correctness and clarity of its evaluation criteria. Only examples passing this review were included in the final dataset. External expert feedback led to additional examples being removed. |