FitCF: A Framework for Automatic Feature Importance-guided Counterfactual Example Generation
This work addresses the problem of generating reliable counterfactual examples for model improvement and explainability in NLP, representing an incremental advancement with specific gains in quality metrics.
The paper tackles the challenge of automated counterfactual example generation for NLP and XAI by introducing FitCF, a framework that uses feature importance to guide generation and verification, achieving improved performance over state-of-the-art baselines as measured by flip rate, perplexity, and similarity metrics.
Counterfactual examples are widely used in natural language processing (NLP) as valuable data to improve models, and in explainable artificial intelligence (XAI) to understand model behavior. The automated generation of counterfactual examples remains a challenging task even for large language models (LLMs), despite their impressive performance on many tasks. In this paper, we first introduce ZeroCF, a faithful approach for leveraging important words derived from feature attribution methods to generate counterfactual examples in a zero-shot setting. Second, we present a new framework, FitCF, which further verifies aforementioned counterfactuals by label flip verification and then inserts them as demonstrations for few-shot prompting, outperforming two state-of-the-art baselines. Through ablation studies, we identify the importance of each of FitCF's core components in improving the quality of counterfactuals, as assessed through flip rate, perplexity, and similarity measures. Furthermore, we show the effectiveness of LIME and Integrated Gradients as backbone attribution methods for FitCF and find that the number of demonstrations has the largest effect on performance. Finally, we reveal a strong correlation between the faithfulness of feature attribution scores and the quality of generated counterfactuals, which we hope will serve as an important finding for future research in this direction.