跳到主要导航 跳到搜索 跳到主要内容

Extracting optimal explanations for ensemble trees via automated reasoning

  • Gelin Zhang
  • , Zhé Hóu
  • , Yanhong Huang*
  • , Jianqi Shi
  • , Hadrien Bride
  • , Jin Song Dong
  • , Yongsheng Gao
  • *此作品的通讯作者
  • East China Normal University
  • Griffith University Queensland
  • National University of Singapore

科研成果: 期刊稿件文章同行评审

摘要

Ensemble trees are a popular machine learning model which often yields high prediction performance when analysing structured data. Although individual small decision trees are deemed explainable by nature, an ensemble of large trees is often difficult to understand. In this work, we propose an approach called optimised explanation (OptExplain) that faithfully extracts global explanations of ensemble trees using a combination of logical reasoning, sampling, and nature-inspired optimisation. OptExplain is an interpretable surrogate model that is as close as possible to the prediction ability of the original model. Building on top of this, we propose a method called the profile of equivalent classes (ProClass), which simplify the explanation even further by solving the maximum satisfiability problem (MAX-SAT). ProClass gives the profile of the classes and features from the perspective of the model. Experiment on several datasets shows that our approach can provide high-quality explanations to large ensemble tree models, and it betters recent top-performers.

源语言英语
页(从-至)14371-14382
页数12
期刊Applied Intelligence
53
11
DOI
出版状态已出版 - 6月 2023

指纹

探究 'Extracting optimal explanations for ensemble trees via automated reasoning' 的科研主题。它们共同构成独一无二的指纹。

引用此