Counterfactual Selective Imitation for Long-Horizon Agents

Chunlin Yu, Ruiqi Xu, Huan Li, Chuliang Weng


Abstract
Long-horizon agents with structured actions execute tasks through sequences of dependent decisions, so a local decision mistake can change later observations and cause the task to fail. Stronger expert models offer a natural source of supervision, yet direct imitation or behavior cloning treats many differences between expert and base actions as equally useful targets. In this paper, we propose counterfactual selective imitation (CSI), a method that uses same-state counterfactual comparison to identify positive return differences. CSI assigns expert targets only when counterfactual rollouts show a high positive return difference, and routine states receive base-policy targets. The learned update is restricted to structured action spans, leaves the base model frozen, and does not require expert queries or replay at deployment. Experiments on AppWorld, ToolSandbox, and WebShop show that CSI raises average task success from 33.3% to 49.7% over the Qwen3-8B base and reduces average invalid/tool error from 9.3% to 4.9%. Compared with the strongest same-deployment adaptation baseline, CSI improves average task success by 1.3 points with only 5% additional deployed-inference latency.
Anthology ID:
2026.findings-emnlp.1532
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
32605–32618
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1532/
DOI:
Bibkey:
Cite (ACL):
Chunlin Yu, Ruiqi Xu, Huan Li, and Chuliang Weng. 2026. Counterfactual Selective Imitation for Long-Horizon Agents. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 32605–32618, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Counterfactual Selective Imitation for Long-Horizon Agents (Yu et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1532.pdf