Knowledge-Guided Explanations for Tree Ensemble Models via Levels-Structure Values

Abstract Tree ensemble models, such as gradient boosting machines and random forests, have achieved strong predictive performance on tabular data by capturing complex nonlinear relationships and feature interactions. However, their flexibility often comes at the cost of interpretability, especially in scientific applications where features are organized by domain knowledge into a priori multi-level structures. Existing game-theoretic explanation methods, including TreeSHAP and TreeOwen, provide principled attribution frameworks based on the Shapley and Owen values; yet, they address either individual features or a single grouping level. In this paper, I introduce Tree Levels-Structure Value (TLSV), a knowledge-guided, principled, and efficient framework for computing the levels-structure value—which generalizes the Owen value from a single a priori partition to a chain of nested partitions—for tree ensemble models. TLSV therefore accommodates multi-level coalition structures, such as features within groups and groups within higher-level supergroups, while preserving coherent attributions across all resolutions, with single-level Owen-value attribution recovered as a special case. Algorithmically, TLSV combines hierarchy-aware recursion with tree-aware dynamic programming to avoid exhaustive subset enumeration within terminal feature groups. Its exact algorithm returns levels-structure values for the tree-aware cooperative game induced by the fitted ensemble. For high-branching hierarchies, I also develop a recursive Monte Carlo estimator that provides scalable approximation with unbiasedness and consistency guarantees. Through simulation studies, I assess attribution accuracy, finite-budget Monte Carlo behavior, numerical coherence, and computational scalability. Finally, I illustrate TLSV using microbiome data from an immunotherapy study, showing how taxonomically nested microbial features contribute coherently to predictions of treatment response.

Authors

Institutions

Publication Details

Journal
Research Square
Published
2026-11-04
DOI
https://doi.org/10.21203/rs.3.rs-11600957/v1
Primary Topic
Explainable Artificial Intelligence (XAI)
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Knowledge-Guided Explanations for Tree Ensemble Models via Levels-Structure Values

Hyunwook Koh
Research Square
Explainable Artificial Intelligence (XAI)
preprint

Knowledge-Guided Explanations for Tree Ensemble Models via Levels-Structure Values

Hyunwook Koh
preprint en

Abstract

Abstract Tree ensemble models, such as gradient boosting machines and random forests, have achieved strong predictive performance on tabular data by capturing complex nonlinear relationships and feature interactions. However, their flexibility often comes at the cost of interpretability, especially in scientific applications where features are organized by domain knowledge into a priori multi-level structures. Existing game-theoretic explanation methods, including TreeSHAP and TreeOwen, provide principled attribution frameworks based on the Shapley and Owen values; yet, they address either individual features or a single grouping level. In this paper, I introduce Tree Levels-Structure Value (TLSV), a knowledge-guided, principled, and efficient framework for computing the levels-structure value—which generalizes the Owen value from a single a priori partition to a chain of nested partitions—for tree ensemble models. TLSV therefore accommodates multi-level coalition structures, such as features within groups and groups within higher-level supergroups, while preserving coherent attributions across all resolutions, with single-level Owen-value attribution recovered as a special case. Algorithmically, TLSV combines hierarchy-aware recursion with tree-aware dynamic programming to avoid exhaustive subset enumeration within terminal feature groups. Its exact algorithm returns levels-structure values for the tree-aware cooperative game induced by the fitted ensemble. For high-branching hierarchies, I also develop a recursive Monte Carlo estimator that provides scalable approximation with unbiasedness and consistency guarantees. Through simulation studies, I assess attribution accuracy, finite-budget Monte Carlo behavior, numerical coherence, and computational scalability. Finally, I illustrate TLSV using microbiome data from an immunotherapy study, showing how taxonomically nested microbial features contribute coherently to predictions of treatment response.

Research Square
SUNY Korea (KR)
Explainable Artificial Intelligence (XAI)
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon