English

iFlip: Iterative Feedback-driven Counterfactual Example Refinement

Computation and Language 2026-01-06 v1 Machine Learning

Abstract

Counterfactual examples are minimal edits to an input that alter a model's prediction. They are widely employed in explainable AI to probe model behavior and in natural language processing (NLP) to augment training data. However, generating valid counterfactuals with large language models (LLMs) remains challenging, as existing single-pass methods often fail to induce reliable label changes, neglecting LLMs' self-correction capabilities. To explore this untapped potential, we propose iFlip, an iterative refinement approach that leverages three types of feedback, including model confidence, feature attribution, and natural language. Our results show that iFlip achieves an average 57.8% higher validity than the five state-of-the-art baselines, as measured by the label flipping rate. The user study further corroborates that iFlip outperforms baselines in completeness, overall satisfaction, and feasibility. In addition, ablation studies demonstrate that three components are paramount for iFlip to generate valid counterfactuals: leveraging an appropriate number of iterations, pointing to highly attributed words, and early stopping. Finally, counterfactuals generated by iFlip enable effective counterfactual data augmentation, substantially improving model performance and robustness.

Keywords

Cite

@article{arxiv.2601.01446,
  title  = {iFlip: Iterative Feedback-driven Counterfactual Example Refinement},
  author = {Yilong Wang and Qianli Wang and Nils Feldhus},
  journal= {arXiv preprint arXiv:2601.01446},
  year   = {2026}
}

Comments

In submission

R2 v1 2026-07-01T08:49:47.303Z