DiCoFlex: Model-Agnostic Diverse Counterfactuals with Flexible Control

Patryk Marszałek (Jagiellonian University in Krakow) · Marek Śmieja (Jagiellonian University, Golebia 24, 31-007 Krkaów, NIP: 675-000-22-36) · Oleksii Furman (Wroclaw University of Science and Technology) · Ulvi Movsum-zada (Jagiellonian University in Krakow) · Maciej Zieba (Wroclaw University of Science and Technology, Tooploox)
actionabilitycomputationally intensive optimizationconditional generative frameworkconditional normalizing flowsconstraint adherencecounterfactual explanationsdiverse counterfactualsdiversityexplainable artificial intelligencemachine learning model decisionsmodel-agnostic frameworkreal-time customizationsparsityuser-defined constraintsvalidity

Counterfactual explanations play a pivotal role in explainable artificial intelligence (XAI) by offering intuitive, human-understandable alternatives that elucidate machine learning model decisions. Despite their significance, existing methods for generating counterfactuals often require constant access to the predictive model, involve computationally intensive optimization for each instance, and lack the flexibility to adapt to new user-defined constraints without retraining. In this paper, we propose DiCoFlex, a novel model-agnostic, conditional generative framework that produces multiple diverse counterfactuals in a single forward pass. Leveraging conditional normalizing flows trained solely on labeled data, DiCoFlex addresses key limitations by enabling real-time, user-driven customization of constraints such as sparsity and actionability at inference time. Extensive experiments on standard benchmark datasets show that DiCoFlex outperforms existing methods in terms of validity, diversity, proximity, and constraint adherence, making it a practical and scalable solution for counterfactual generation in sensitive decision-making domains.