WizMap: Scalable Interactive Visualization for Exploring Large Machine Learning Embeddings

Zijie J. Wang, Fred Hohman, Duen Horng Chau


Abstract
Machine learning models often learn latent embedding representations that capture the domain semantics of their training data. These embedding representations are valuable for interpreting trained models, building new models, and analyzing new datasets. However, interpreting and using embeddings can be challenging due to their opaqueness, high dimensionality, and the large size of modern datasets. To tackle these challenges, we present WizMap, an interactive visualization tool to help researchers and practitioners easily explore large embeddings. With a novel multi-resolution embedding summarization method and a familiar map-like interaction design, WizMap enables users to navigate and interpret embedding spaces with ease. Leveraging modern web technologies such as WebGL and Web Workers, WizMap scales to millions of embedding points directly in users’ web browsers and computational notebooks without the need for dedicated backend servers. WizMap is open-source and available at the following public demo link: https://poloclub.github.io/wizmap.
Anthology ID:
2023.acl-demo.50
Volume:
Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 3: System Demonstrations)
Month:
July
Year:
2023
Address:
Toronto, Canada
Editors:
Danushka Bollegala, Ruihong Huang, Alan Ritter
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
516–523
Language:
URL:
https://aclanthology.org/2023.acl-demo.50
DOI:
10.18653/v1/2023.acl-demo.50
Bibkey:
Cite (ACL):
Zijie J. Wang, Fred Hohman, and Duen Horng Chau. 2023. WizMap: Scalable Interactive Visualization for Exploring Large Machine Learning Embeddings. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 3: System Demonstrations), pages 516–523, Toronto, Canada. Association for Computational Linguistics.
Cite (Informal):
WizMap: Scalable Interactive Visualization for Exploring Large Machine Learning Embeddings (Wang et al., ACL 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.acl-demo.50.pdf
Video:
 https://aclanthology.org/2023.acl-demo.50.mp4