We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: SLISEMAP: Supervised dimensionality reduction through local explanations

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.
Comments: 26 pages, 10 figures, 3 tables. This revision replaces the $\lambda_z$ parameter with $z_{radius}$, which is more intuitive and stable. There are also many typographical and clarity improvements. The source code for our implementation of the algorithm (and the experiments) are available from GitHub at this https URL . Machine Learning (2022)
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Human-Computer Interaction (cs.HC)
Journal reference: Machine Learning 112, 1-43 (2023)
DOI: 10.1007/s10994-022-06261-1
Cite as: arXiv:2201.04455 [cs.LG]
  (or arXiv:2201.04455v2 [cs.LG] for this version)

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.