Random forest similarity maps: A scalable visual representation for global and local interpretation

Dipankar Mazumdar, Mário Popolin Neto, Fernando V. Paulovich (Corresponding author)

Research output: Contribution to journalArticleAcademicpeer-review

3 Citations (Scopus)

Abstract

Machine Learning prediction algorithms have made significant contributions in today’s world, leading to increased usage in various domains. However, as ML algorithms surge, the need for transparent and interpretable models becomes essential. Visual representations have shown to be instrumental in addressing such an issue, allowing users to grasp models’ inner workings. Despite their popularity, visualization techniques still present visual scalability limitations, mainly when applied to analyze popular and complex models, such as Random Forests (RF). In this work, we propose Random Forest Similarity Map (RFMap), a scalable interactive visual analytics tool designed to analyze RF ensemble models. RFMap focuses on explaining the inner working mechanism of models through different views describing individual data instance predictions, providing an overview of the entire forest of trees, and highlighting instance input feature values. The interactive nature of RFMap allows users to visually interpret model errors and decisions, establishing the necessary confidence and user trust in RF models and improving performance.

Original languageEnglish
Article number2862
Number of pages20
JournalElectronics
Volume10
Issue number22
DOIs
Publication statusPublished - 1 Nov 2021
Externally publishedYes

Bibliographical note

Funding Information:
Acknowledgments: We acknowledge the support of the Natural Sciences and Engineering Research Council of Canada (NSERC).

Keywords

  • Classification model visualization
  • Dimensionality reduction
  • Explainable artificial intelligence (XAI)
  • Random Forest

Fingerprint

Dive into the research topics of 'Random forest similarity maps: A scalable visual representation for global and local interpretation'. Together they form a unique fingerprint.

Cite this