A Visual Analytics System for Multi-Model Comparison on Clinical Data Predictions

Yiran Li, Takanori Fujiwara, Yong K. Choi, Katherine K. Kim, and Kwan-Liu Ma

IEEE PacificVis 2020 Workshop on Visualization Meets AI, 2020
image
A screenshot of our visual analytics system, which contains the Admission Overview (a), Feature Contributions View (b), Model Summary View (c), and Consistency Charts View (d).

PDF   ArXiv Version   Publisher's Version  

Abstract

There is a growing trend of applying machine learning methods to medical datasets in order to predict patients’ future status. Although some of these methods achieve high performance, challenges still exist in comparing and evaluating different models through their interpretable information. Such analytics can help clinicians improve evidence-based medical decision making. In this work, we develop a visual analytics system that compares multiple models’ prediction criteria and evaluates their consistency. With our system, users can generate knowledge on different models’ inner criteria and how confidently we can rely on each model’s prediction for a certain patient. Through a case study of a publicly available clinical dataset, we demonstrate the effectiveness of our visual analytics system to assist clinicians and researchers in comparing and quantitatively evaluating different machine learning methods.

BibTex

@article{li2020visual,
    title={A visual analytics system for multi-model comparison on clinical data predictions},
    author={Li, Yiran and Fujiwara, Takanori and Choi, Yong K and Kim, Katherine K and Ma, Kwan-Liu},
    journal={Visual Informatics},
    volume={4},
    number={2},
    pages={122--131},
    year={2020},
    publisher={Elsevier}
}