MonoForest framework for tree ensemble analysis
Kuralenok, Igor, Ershov, Vasilii, Labutin, Igor
–Neural Information Processing Systems
In this work, we introduce a new decision tree ensemble representation framework: instead of using a graph model we transform each tree into a well-known polynomial form. We apply the new representation to three tasks: theoretical analysis, model reduction, and interpretation. The polynomial form of a tree ensemble allows a straightforward interpretation of the original model. In our experiments, it shows comparable results with state-of-the-art interpretation techniques. Another application of the framework is the ensemble-wise pruning: we can drop monomials from the polynomial, based on train data statistics.
Neural Information Processing Systems
Mar-19-2020, 02:17:03 GMT