about
KnowledgeVIS: Interpreting LLMs by Comparing Fill-in-the-Blank Prompts (arxiv.org)
1 point by Jimmc414 on Mar 11, 2024 | hide | past | pdf | discuss on HN

In plain words: A tool that fills in blanks in sentences and compares the words a language model predicts, grouping similar guesses and showing them visually to reveal what the model has learned. Six language experts used it to find biomedical facts, harmful stereotypes, and concept relationships.

Abstract · KnowledgeVIS: Interpreting Language Models by Comparing Fill-in-the-Blank Prompts

Recent growth in the popularity of large language models has led to their increased usage for summarizing, predicting, and generating text, making it vital to help researchers and engineers understand how and why they work. We present KnowledgeVis, a human-in-the-loop visual analytics system for interpreting language models using fill-in-the-blank sentences as prompts. By comparing predictions between sentences, KnowledgeVis reveals learned associations that intuitively connect what language models learn during training to natural language tasks downstream, helping users create and test multiple prompt variations, analyze predicted words using a novel semantic clustering technique, and discover insights using interactive visualizations. Collectively, these visualizations help users identify the likelihood and uniqueness of individual predictions, compare sets of predictions between prompts, and summarize patterns and relationships between predictions across all prompts. We demonstrate the capabilities of KnowledgeVis with feedback from six NLP experts as well as three different use cases: (1) probing biomedical knowledge in two domain-adapted models; and (2) evaluating harmful identity stereotypes and (3) discovering facts and relationships between three general-purpose models.

Adam Coscia, Alex Endert
arXiv:2403.04758 · cs.HC, cs.AI, cs.CY, cs.LG · submitted Mar 7, 2024
abstract · pdf · html · Accepted to IEEE TVCG. 20 pages, 10 figures, 1 table. For a demo video, see https://youtu.be/hBX4rSUMr_I . For a live demo, visit https://adamcoscia.com/papers/knowledgevis/demo/ . The source code is available at https://github.com/AdamCoscia/KnowledgeVIS

add comment on HN