Decision Support System Improving the Interpretability of Generated Tree-Based Models

Diana Klimonová, Viera Anderková, František Babič, Ljiljana Trtica Majnaric

Decision Support System Improving the Interpretability of Generated Tree-Based Models

Číslo: 2/2022
Periodikum: Acta Electrotechnica et Informatica
DOI: 10.2478/aei-2022-0009

Klíčová slova: decision tree, interpretability, decision support

Pro získání musíte mít účet v Citace PRO.

Přečíst po přihlášení

Anotace: A decision tree represents one of the most used data analysis methods for classification tasks. The generated decision models can be visualized as a graph, but this visualization is quite complicated for a domain expert to understand in large or heterogeneous data. Our previous experience with medical data analytics related to the classification of patients with Metabolic Syndrome, Mild Cognitive Impairment, heart disease, or Frailty motivated us to evaluate the potential of new visualizations for this decision model in the medical domain. We managed a user study to design and implement a decision support system containing selected methods to improve the interpretability of the generated tree-based decision model. We hypothesized that this approach would result in more effective communication between data analysts and medical experts, reduce necessary time and energy and bring more comprehensive results. For this purpose, we selected two model-agnostic methods, LIME and SHAP, and one new interactive visualization called Sunburst. We used two data samples for design and evaluation: the publicly available heart disease dataset and the Metabolic Syndrome dataset the participating medical expert provided. We will use the collected feedback and experience for further improvements, like more evaluation metrics related to the usability of the decision models.