English

Forest-Guided Clustering -- Shedding Light into the Random Forest Black Box

Machine Learning 2025-07-28 v1

Abstract

As machine learning models are increasingly deployed in sensitive application areas, the demand for interpretable and trustworthy decision-making has increased. Random Forests (RF), despite their widespread use and strong performance on tabular data, remain difficult to interpret due to their ensemble nature. We present Forest-Guided Clustering (FGC), a model-specific explainability method that reveals both local and global structure in RFs by grouping instances according to shared decision paths. FGC produces human-interpretable clusters aligned with the model's internal logic and computes cluster-specific and global feature importance scores to derive decision rules underlying RF predictions. FGC accurately recovered latent subclass structure on a benchmark dataset and outperformed classical clustering and post-hoc explanation methods. Applied to an AML transcriptomic dataset, FGC uncovered biologically coherent subpopulations, disentangled disease-relevant signals from confounders, and recovered known and novel gene expression patterns. FGC bridges the gap between performance and interpretability by providing structure-aware insights that go beyond feature-level attribution.

Keywords

Cite

@article{arxiv.2507.19455,
  title  = {Forest-Guided Clustering -- Shedding Light into the Random Forest Black Box},
  author = {Lisa Barros de Andrade e Sousa and Gregor Miller and Ronan Le Gleut and Dominik Thalmeier and Helena Pelin and Marie Piraud},
  journal= {arXiv preprint arXiv:2507.19455},
  year   = {2025}
}
R2 v1 2026-07-01T04:19:12.383Z