DimVis: Interpreting Visual Clusters in Dimensionality Reduction With Explainable Boosting Machine

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Salmanian, Parisa, Chatzimparmpas, Angelos, Karaca, Ali Can, Martins, Rafael M.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916211242041344
author Salmanian, Parisa
Chatzimparmpas, Angelos
Karaca, Ali Can
Martins, Rafael M.
author_facet Salmanian, Parisa
Chatzimparmpas, Angelos
Karaca, Ali Can
Martins, Rafael M.
contents Dimensionality Reduction (DR) techniques such as t-SNE and UMAP are popular for transforming complex datasets into simpler visual representations. However, while effective in uncovering general dataset patterns, these methods may introduce artifacts and suffer from interpretability issues. This paper presents DimVis, a visualization tool that employs supervised Explainable Boosting Machine (EBM) models (trained on user-selected data of interest) as an interpretation assistant for DR projections. Our tool facilitates high-dimensional data analysis by providing an interpretation of feature relevance in visual clusters through interactive exploration of UMAP projections. Specifically, DimVis uses a contrastive EBM model that is trained in real time to differentiate between the data inside and outside a cluster of interest. Taking advantage of the inherent explainable nature of the EBM, we then use this model to interpret the cluster itself via single and pairwise feature comparisons in a ranking based on the EBM model's feature importance. The applicability and effectiveness of DimVis are demonstrated via a use case and a usage scenario with real-world data. We also discuss the limitations and potential directions for future research.
format Preprint
id arxiv_https___arxiv_org_abs_2402_06885
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DimVis: Interpreting Visual Clusters in Dimensionality Reduction With Explainable Boosting Machine
Salmanian, Parisa
Chatzimparmpas, Angelos
Karaca, Ali Can
Martins, Rafael M.
Human-Computer Interaction
Machine Learning
Computation
Dimensionality Reduction (DR) techniques such as t-SNE and UMAP are popular for transforming complex datasets into simpler visual representations. However, while effective in uncovering general dataset patterns, these methods may introduce artifacts and suffer from interpretability issues. This paper presents DimVis, a visualization tool that employs supervised Explainable Boosting Machine (EBM) models (trained on user-selected data of interest) as an interpretation assistant for DR projections. Our tool facilitates high-dimensional data analysis by providing an interpretation of feature relevance in visual clusters through interactive exploration of UMAP projections. Specifically, DimVis uses a contrastive EBM model that is trained in real time to differentiate between the data inside and outside a cluster of interest. Taking advantage of the inherent explainable nature of the EBM, we then use this model to interpret the cluster itself via single and pairwise feature comparisons in a ranking based on the EBM model's feature importance. The applicability and effectiveness of DimVis are demonstrated via a use case and a usage scenario with real-world data. We also discuss the limitations and potential directions for future research.
title DimVis: Interpreting Visual Clusters in Dimensionality Reduction With Explainable Boosting Machine
topic Human-Computer Interaction
Machine Learning
Computation
url https://arxiv.org/abs/2402.06885