FitCF: A Framework for Automatic Feature Importance-guided Counterfactual Example Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Qianli, Feldhus, Nils, Ostermann, Simon, Villa-Arenas, Luis Felipe, Möller, Sebastian, Schmitt, Vera
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908382288412672
author Wang, Qianli
Feldhus, Nils
Ostermann, Simon
Villa-Arenas, Luis Felipe
Möller, Sebastian
Schmitt, Vera
author_facet Wang, Qianli
Feldhus, Nils
Ostermann, Simon
Villa-Arenas, Luis Felipe
Möller, Sebastian
Schmitt, Vera
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2501_00777
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle FitCF: A Framework for Automatic Feature Importance-guided Counterfactual Example Generation
Wang, Qianli
Feldhus, Nils
Ostermann, Simon
Villa-Arenas, Luis Felipe
Möller, Sebastian
Schmitt, Vera
Computation and Language
Machine Learning
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.
title FitCF: A Framework for Automatic Feature Importance-guided Counterfactual Example Generation
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2501.00777