mlsanityData to 7 Oct 2026

Validity of Feature Importance in Low-Performing Machine Learning for Tabular Biomedical Data

arXiv 2409.13342 ↗First posted 20 Sep 2024stat.ML cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

In tabular biomedical data analysis, tuning models to high accuracy is considered a prerequisite for discussing feature importance, as medical practitioners expect the validity of feature importance to correlate with performance. In this work, we challenge the prevailing belief, showing that low-performing models may also be used for feature importance. We propose experiments to observe changes in feature rank as performance degrades sequentially. Using three synthetic datasets and six real biomedical datasets, we compare the rank of features from full datasets to those with reduced sample sizes (data cutting) or fewer features (feature cutting). In synthetic datasets, feature cutting does not change feature rank, while data cutting shows higher discrepancies with lower performance. In real datasets, feature cutting shows similar or smaller changes than data cutting, though some datasets exhibit the opposite. When feature interactions are controlled by removing correlations, feature cutting consistently shows better stability. By analyzing the distribution of feature importance values and theoretically examining the probability that the model cannot distinguish feature importance between features, we reveal that models can still distinguish feature importance despite performance degradation through feature cutting, but not through data cutting. We conclude that the validity of feature importance can be maintained even at low performance levels if the data size is adequate, which is a significant factor contributing to suboptimal performance in tabular medical data analysis. This paper demonstrates the potential for utilizing feature importance analysis alongside statistical analysis to compare features relatively, even when classifier performance is not satisfactory.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 57 references that are in the map. First posted on arXiv 20 Sep 2024.

Builds on

11 of 57 references.

References · 11

  1. 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
  2. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  3. 2021Revisiting Deep Learning Models for Tabular Datacited by 725
  4. 2020Transformer Interpretability Beyond Attention Visualizationcited by 620
  5. 2021Tabular Data: Deep Learning is Not All You Needcited by 530
  6. 2018A Benchmark for Interpretability Methods in Deep Neural Networkscited by 467
  7. 2021SAINT: Improved Neural Networks for Tabular Data via Row Attention and Contrastive Pre-Trainingcited by 310
  8. 2019Unrestricted Permutation forces Extrapolation: Variable Importance Requires at least One More Model, or There Is No Free Variable Importancecited by 85
  9. 2018Feature-Wise Bias Amplificationcited by 31
  10. 2021Addressing Fairness, Bias and Class Imbalance in Machine Learning: the FBI-losscited by 6
  11. 2023Rethinking Class Imbalance in Machine Learningcited by 4

Built on by

Not cited yet.