VAPER: A deep learning model for explainable probabilistic regression

Seungwon Jung, Yoona Noh, Jaeuk Moon, Eenjun Hwang

Research output: Contribution to journalArticlepeer-review

Abstract

A probabilistic regression model provides decision-makers with the regression output along with its quantitative uncertainty for given input variables. Even though this uncertainty may help to avoid serious consequences such as misdiagnosis or blackout due to overconfidence in the output, it only provides a measure of the uncertainty of the output, and has a limitation in that it cannot explain the reasons for the output and its uncertainty. If they can be presented along with their reasons, more suitable alternatives to the output can be found. However, despite the development of artificial intelligence methods to explain machine learning models and their outputs, few probabilistic regression models with this functionality have been proposed. Therefore, in this paper, we propose a variational autoencoder-based model for explainable probabilistic regression, called VAPER. VAPER provides a parametric probability distribution of an output variable over input variables and interprets it using layer-wise relevance propagation to investigate the effect of each input variable. To evaluate the effectiveness of the proposed model, we performed extensive experiments using several datasets. The experimental results demonstrated that VAPER has competitive regression performance compared to existing models, even with effective explainability.

Original languageEnglish
Article number101824
JournalJournal of Computational Science
Volume63
DOIs
Publication statusPublished - 2022 Sep

Keywords

  • Explainable artificial intelligence
  • Layer-wise relevance propagation
  • Probabilistic regression
  • Variational autoencoder

ASJC Scopus subject areas

  • Theoretical Computer Science
  • Computer Science(all)
  • Modelling and Simulation

Fingerprint

Dive into the research topics of 'VAPER: A deep learning model for explainable probabilistic regression'. Together they form a unique fingerprint.

Cite this