Compressing and interpreting word embeddings with latent space regularization and interactive semantics probing

被引:34
作者
Li, Haoyu [1 ,2 ]
Wang, Junpeng [2 ]
Zheng, Yan [2 ]
Wang, Liang [2 ]
Zhang, Wei [2 ]
Shen, Han-Wei [1 ]
机构
[1] Ohio State Univ, Dept Comp Sci & Engn, GRAV Res Grp, 786 Dreese Lab,2015 Neil Ave, Columbus, OH 43210 USA
[2] Visa Res, Palo Alto, CA USA
基金
美国国家科学基金会;
关键词
High-dimensional data visualization; visual analytics; neural networks; word embedding; VISUAL ANALYTICS; EXPLORATION;
D O I
10.1177/14738716221130338
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
Word embedding, a high-dimensional (HD) numerical representation of words generated by machine learning models, has been used for different natural language processing tasks, for example, translation between two languages. Recently, there has been an increasing trend of transforming the HD embeddings into a latent space (e.g. via autoencoders) for further tasks, exploiting various merits the latent representations could bring. To preserve the embeddings' quality, these works often map the embeddings into an even higher-dimensional latent space, making the already complicated embeddings even less interpretable and consuming more storage space. In this work, we borrow the idea of beta VAE to regularize the HD latent space. Our regularization implicitly condenses information from the HD latent space into a much lower-dimensional space, thus compressing the embeddings. We also show that each dimension of our regularized latent space is more semantically salient, and validate our assertion by interactively probing the encoding-level of user-proposed semantics in the dimensions. To the end, we design a visual analytics system to monitor the regularization process, explore the HD latent space, and interpret latent dimensions' semantics. We validate the effectiveness of our embedding regularization and interpretation approach through both quantitative and qualitative evaluations.
引用
收藏
页码:52 / 68
页数:17
相关论文
共 33 条
  • [31] DeepVID: Deep Visual Interpretation and Diagnosis for Image Classifiers via Knowledge Distillation
    Wang, Junpeng
    Gou, Liang
    Zhang, Wei
    Yang, Hao
    Shen, Han-Wei
    [J]. IEEE TRANSACTIONS ON VISUALIZATION AND COMPUTER GRAPHICS, 2019, 25 (06) : 2168 - 2180
  • [32] Diagnosing Concept Drift with Visual Analytics
    Yang, Weikai
    Li, Zhen
    Liu, Mengchen
    Lu, Yafeng
    Cao, Kelei
    Maciejewski, Ross
    Liu, Shixia
    [J]. 2020 IEEE CONFERENCE ON VISUAL ANALYTICS SCIENCE AND TECHNOLOGY (VAST 2020), 2020, : 12 - 23
  • [33] A survey of visual analytics techniques for machine learning
    Yuan, Jun
    Chen, Changjian
    Yang, Weikai
    Liu, Mengchen
    Xia, Jiazhi
    Liu, Shixia
    [J]. COMPUTATIONAL VISUAL MEDIA, 2021, 7 (01) : 3 - 36