K-IPO: Preserving Feature Importance in Imbalanced Data Oversampling.

Marios Tyrovolas, Argiris Sofotasios, Dimitris Metaxakis, Georgios Mermigkis, George Georgoulas, Panagiotis Hadjidoukas, Chrysostomos Stylios· July 21, 2026 View original

Summary

This paper introduces Kendall-constrained Importance-Preserving Oversampling (K-IPO), a novel framework that augments imbalanced tabular data while actively preserving the original feature importance ranking. K-IPO improves predictive performance and explanation consistency.

Oversampling is a common technique to address class imbalance in tabular classification datasets. However, existing oversampling methods often inadvertently distort the feature importance ranking, which is crucial for model interpretability and explanations. This distortion can lead to misleading insights about which features are truly driving a model's predictions. To tackle this, researchers have developed Kendall-constrained Importance-Preserving Oversampling (K-IPO). This generator-agnostic framework operates by iteratively generating minority-class candidates and accepting them only if their inclusion maintains a user-defined minimum Kendall's tau correlation with the original data's feature importance ranking. Optional stricter constraints can be applied to the most highly ranked features. Evaluated across 20 imbalanced binary classification datasets with various classifiers and explanation methods, K-IPO consistently demonstrated superior or competitive performance. It excelled in preserving feature importance, maintaining explanation consistency, and improving class separability. Furthermore, K-IPO generally enhanced predictive performance while keeping computational overhead manageable, offering a significant advancement for interpretable AI on imbalanced data.

Why it matters

For data scientists and ML engineers working with imbalanced datasets, K-IPO offers a method to improve model performance without sacrificing the interpretability of feature importance, which is vital for trustworthy AI and regulatory compliance.

How to implement this in your domain

  1. 1Integrate K-IPO into your data preprocessing pipeline when dealing with imbalanced tabular datasets.
  2. 2Experiment with the Kendall's tau correlation constraint to find the optimal balance between data augmentation and feature importance preservation.
  3. 3Compare K-IPO's performance against traditional oversampling methods using your specific datasets and models.
  4. 4Utilize K-IPO to ensure that model explanations remain consistent and reliable after data augmentation.

Who benefits

FinanceHealthcareMarketingFraud DetectionCustomer Analytics

Key takeaways

  • Existing oversampling methods can distort feature importance, hindering model interpretability.
  • K-IPO preserves original feature importance rankings during data augmentation using Kendall's tau correlation.
  • The method improves predictive performance, explanation consistency, and class separability on imbalanced data.
  • K-IPO is a generator-agnostic framework with competitive computational overhead.

Original post by Marios Tyrovolas, Argiris Sofotasios, Dimitris Metaxakis, Georgios Mermigkis, George Georgoulas, Panagiotis Hadjidoukas, Chrysostomos Stylios

"arXiv:2607.16478v1 Announce Type: new Abstract: Oversampling is widely used to address class imbalance in tabular classification, but existing methods can distort the feature importance ranking underlying model explanations. Although recent studies have quantified this distortion…"

View on X

Originally posted by Marios Tyrovolas, Argiris Sofotasios, Dimitris Metaxakis, Georgios Mermigkis, George Georgoulas, Panagiotis Hadjidoukas, Chrysostomos Stylios on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses