Interpretation with baseline shapley value for feature groups on tree models.

Autor: Xu, Fan, Zhou, Zhi-Jian, Ni, Jie, Gao, Wei
Zdroj: Frontiers of Computer Science; May2025, Vol. 19 Issue 5, p1-14, 14p
Abstrakt: Tree models have made an impressive progress during the past years, while an important problem is to understand how these models predict, in particular for critical applications such as finance and medicine. For this issue, most previous works measured the importance of individual features. In this work, we consider the interpretation of feature groups, which is more effective to capture intrinsic structures and correlations of multiple features. We propose the Baseline Group Shapley value (short for BGShapvalue) to calculate the importance of a feature group for tree models. We further develop a polynomial algorithm, BGShapTree, to deal with the sum of exponential terms in the BGShapvalue. The basic idea is to decompose the BGShapvalue into leaves' weights and exploit the relationships between features and leaves. Based on this idea, we could greedily search salient feature groups with large BGShapvalues. Extensive experiments have validated the effectiveness of our approach, in comparison with state-of-the-art methods on the interpretation of tree models. [ABSTRACT FROM AUTHOR]
Databáze: Complementary Index
Externí odkaz: