Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: SLISEMAP: Supervised dimensionality reduction through local explanations
(Submitted on 12 Jan 2022 (v1), last revised 18 May 2022 (this version, v2))
Abstract: Existing methods for explaining black box learning models often focus on building local explanations of model behaviour for a particular data item. It is possible to create global explanations for all data items, but these explanations generally have low fidelity for complex black box models. We propose a new supervised manifold visualisation method, SLISEMAP, that simultaneously finds local explanations for all data items and builds a (typically) two-dimensional global visualisation of the black box model such that data items with similar local explanations are projected nearby. We provide a mathematical derivation of our problem and an open source implementation implemented using the GPU-optimised PyTorch library. We compare SLISEMAP to multiple popular dimensionality reduction methods and find that SLISEMAP is able to utilise labelled data to create embeddings with consistent local white box models. We also compare SLISEMAP to other model-agnostic local explanation methods and show that SLISEMAP provides comparable explanations and that the visualisations can give a broader understanding of black box regression and classification models.
Submission history
From: Anton Björklund [view email][v1] Wed, 12 Jan 2022 13:06:21 GMT (423kb,D)
[v2] Wed, 18 May 2022 10:52:29 GMT (1475kb,D)
Link back to: arXiv, form interface, contact.