PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
March 6, 2026Journal of Classification1 citationsOpen Access

Class-Focused Variable Importance in Random Forests for Multi-Class Outcomes

RHRoman HornungAHAlexander Hapfelmeier

Key Points

  • This research introduces new variable importance measures to better identify covariates that differentiate between outcome classes in multi-class prediction tasks.
  • Proposes a class-focused variable importance measure that evaluates covariates at each node based on class-specific distinctions.
  • Introduces a discriminatory variable importance measure for assessing general covariate influence from actual node splits.
  • Conducts simulations to compare the performance of the novel measures against conventional methods.
  • The class-focused variable importance measure ranks covariates specifically related to class distinctions significantly higher than traditional measures.
  • Real data examples confirm that both novel measures can provide clearer interpretations of covariate roles in multi-class outcomes.

Abstract

Abstract In multi-class prediction tasks with interpretative goals, covariates that help distinguish individual classes, termed “class-related covariates,” can be of particular interest. Conventional variable importance measures (VIMs) from random forests, such as permutation and Gini importance, rank covariates by overall predictive contribution and thus also assign high importance to covariates that differentiate between groups of classes. We propose a novel VIM, the class-focused VIM, which ranks covariates by their ability to distinguish individual outcome classes. It evaluates covariates using hypothetical multi-way class-based partitions at each node, without altering tree construction. As a complement, we introduce the discriminatory VIM, which measures general covariate influence based on the actual node splits. Simulations show that, unlike conventional VIMs, the class-focused VIM specifically ranks class-related covariates high. Real data examples illustrate how both suggested VIMs behave on real datasets and how their results can be interpreted.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Hornung et al. (2026) studied this question.

synapsesocial.com/papers/69aa7037531e4c4a9ff59c91https://doi.org/10.1007/s00357-026-09545-6
Ask AI
Helpful
Bookmark
Share
View Full Paper