TY - GEN
T1 - Explaining GBDT by Probabilistic Finite-State Automata
AU - Chen, Yinkai
AU - Zhang, Rui
AU - Qiu, Xin
AU - Li, Xin
AU - Deng, Yuxin
N1 - Publisher Copyright:
© 2021 ACM.
PY - 2021/10/15
Y1 - 2021/10/15
N2 - Explainable artificial intelligence becomes vital for human users to understand and trust the decision-making process and results of machine learning methods. Unfortunately, most machine learning models are black-box and the algorithms running behind are opaque. In this work, we propose an approach to interpreting GBDT (Gradient Boosting Decision Tree Explanation) by extracting probabilistic finite-state automata from the trained model. Our method is inspired by and built upon a previous work that extracts probabilistic automata from RNN (Recurrent Neural Networks). To adapt the approach to our situation, we propose a series of techniques to ensure that the extracted probabilistic automaton approximates the GBDT model as accurately as possible. We conduct experiments on real-world datasets and our experimental results show that our method maintains a high level of fidelity of the extracted model as the size of the given GBDT model grows.
AB - Explainable artificial intelligence becomes vital for human users to understand and trust the decision-making process and results of machine learning methods. Unfortunately, most machine learning models are black-box and the algorithms running behind are opaque. In this work, we propose an approach to interpreting GBDT (Gradient Boosting Decision Tree Explanation) by extracting probabilistic finite-state automata from the trained model. Our method is inspired by and built upon a previous work that extracts probabilistic automata from RNN (Recurrent Neural Networks). To adapt the approach to our situation, we propose a series of techniques to ensure that the extracted probabilistic automaton approximates the GBDT model as accurately as possible. We conduct experiments on real-world datasets and our experimental results show that our method maintains a high level of fidelity of the extracted model as the size of the given GBDT model grows.
KW - Gradient Boosting Decision Tree
KW - Interpretable machine learning
KW - Probabilistic finite-state automata
UR - https://www.scopus.com/pages/publications/85124803794
U2 - 10.1145/3497623.3497676
DO - 10.1145/3497623.3497676
M3 - 会议稿件
AN - SCOPUS:85124803794
T3 - ACM International Conference Proceeding Series
SP - 328
EP - 333
BT - ICCPR 2021 - Proceedings of 2021 10th International Conference on Computing and Pattern Recognition
PB - Association for Computing Machinery
T2 - 10th International Conference on Computing and Pattern Recognition, ICCPR 2021
Y2 - 15 October 2021 through 17 October 2021
ER -