USEVis: Visual analytics of attention-based neural embedding in information retrieval

Xiaonan Ji, Yamei Tu, Wenbin He, Junpeng Wang, Han Wei Shen, Po Yin Yen

Research output: Contribution to journalArticlepeer-review

5 Scopus citations

Abstract

Neural attention-based encoders, which effectively attend sentence tokens to their associated context without being restricted by long-term distance or dependency, have demonstrated outstanding performance in embedding sentences into meaningful representations (embeddings). The Universal Sentence Encoder (USE) is one of the most well-recognized deep neural network (DNN) based solutions, which is facilitated with an attention-driven transformer architecture and has been pre-trained on a large number of sentences from the Internet. Besides the fact that USE has been widely used in many downstream applications, including information retrieval (IR), interpreting its complicated internal working mechanism remains challenging. In this work, we present a visual analytics solution towards addressing this challenge. Specifically, focused on semantics and syntactics (concepts and relations) that are critical to domain clinical IR, we designed and developed a visual analytics system, i.e., USEVis. The system investigates the power of USE in effectively extracting sentences’ semantics and syntactics through exploring and interpreting how linguistic properties are captured by attentions. Furthermore, by thoroughly examining and comparing the inherent patterns of these attentions, we are able to exploit attentions to retrieve sentences/documents that have similar semantics or are closely related to a given clinical problem in IR. By collaborating with domain experts, we demonstrate use cases with inspiring findings to validate the contribution of our work and the effectiveness of our system.

Original languageEnglish
Pages (from-to)1-12
Number of pages12
JournalVisual Informatics
Volume5
Issue number2
DOIs
StatePublished - Jun 2021

Keywords

  • Attention mechanism
  • Clinical decision-making
  • Document understanding
  • Information retrieval
  • Interactive visual system
  • Neural embedding

Fingerprint

Dive into the research topics of 'USEVis: Visual analytics of attention-based neural embedding in information retrieval'. Together they form a unique fingerprint.

Cite this