Robust Random Forest-Based All-Relevant Feature Ranks for Trustworthy AI

Stud Health Technol Inform. 2022 May 25:294:137-138. doi: 10.3233/SHTI220418.

Abstract

Feature selection is a fundamental challenge in machine learning. For instance in bioinformatics, it is essential when one wishes to detect biomarkers. Tree-based methods are predominantly used for this purpose. In this paper, we study the stability of the feature selection methods BORUTA, VITA, and RRF (regularized random forest). In particular, we investigate the feature ranking instability of the associated stochastic algorithms. For stabilization of the feature ranks, we propose to compute consensus values from multiple feature selection runs, applying rank aggregation techniques. Our results show that these consolidated features are more accurate and robust, which helps to make practical machine learning applications more trustworthy.

Keywords: Feature Selection; Random Forest; Rank Aggregation; Trustworthy AI.

MeSH terms

  • Algorithms*
  • Biomarkers
  • Computational Biology / methods
  • Machine Learning*

Substances

  • Biomarkers