atlas

Research map / Large language models / Retrieval and knowledge

Knowledge Card: Filling LLMs' Knowledge Gaps with Plug-in Specialized Language Models

ICLR 2024 · Large language models › Retrieval and knowledge · approach: Knowledge-augmented question answering · 4 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Knowledge Card, a modular framework where many small domain-specialized LMs (OPT-1.3B based) generate background knowledge that is filtered by relevance, pruning, and factuality selectors and plugged into black-box LLMs via bottom-up or top-down integration.

Problem

Large LLMs are static and expensive to update, and fail on factual, long-tail, and up-to-date knowledge-intensive tasks; existing retrieval and generated-knowledge methods rely on a single fixed knowledge source and do not support modular, community-contributed knowledge for black-box LLMs.

Approach

Train many small specialized LMs (knowledge cards) on domain corpora, use them as parametric knowledge generators, filter their outputs with relevance, pruning (summarization), and factuality selectors (summarization factuality plus retrieval-augmented fact-checking with top-k sampling), and integrate them into a black-box LLM either by activating all cards at once (bottom-up) or by letting the LL

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use top-down explicit selection with Knowledge CardCodex · MMLUaccuracy: 68.3 -> 72.8Helped
Use bottom-up integration of Knowledge CardsCodex · LUN misinformation detection two-waybalanced accuracy: 65.6 -> 89.8Helped
Use bottom-up integration of Knowledge CardsCodex · LUN misinformation detection four-waybalanced accuracy: 52.8 -> 70.6Helped
Add a midterm-election News knowledge card with top-down automatic selectionCodex · MidtermQA open-bookexact match: 55.1 -> 87.5Helped
Add PubMed, IMDB, and BookCorpus knowledge cards (ablation)ChatGPT · LUN misinformation detection two-waybalanced accuracy: 80.1 -> 80.7; 80.6; 82.3Helped
Add an in-domain News knowledge card (ablation)ChatGPT · LUN misinformation detection two-waybalanced accuracy: 82.3 -> 85.7Helped
Add a Wikipedia knowledge card after News (ablation)ChatGPT · LUN misinformation detection two-waybalanced accuracy: 85.7 -> 76.5Hurt
Remove the relevance selector (ablation)Bottom-Up Knowledge Card · LUN misinformation detectionmodel performanceHurt
Remove the pruning selector (ablation)Bottom-Up Knowledge Card · LUN misinformation detectionmodel performanceHurt
Remove the factuality selector (ablation)Bottom-Up Knowledge Card · LUN misinformation detectionmodel performanceHurt
Increase the number of generated documents per knowledge card (n1) (ablation)Bottom-Up Knowledge Card · Knowledge stream analysismodel performanceNo clear effect
Retain more documents after the knowledge selectors (n2) (ablation)Bottom-Up Knowledge Card · Knowledge stream analysismodel performanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: MMLU LUN MidtermQA Wikipedia PubMed IMDB BookCorpus ATOMIC ConceptNet UMLS Wikidata YAGO

Benchmarks: MMLU LUN misinformation dataset (two-way and four-way) MidtermQA (open-book, two-way, four-way)

Metrics: accuracy balanced accuracy macro F1 exact match F1

Baselines: Codex PaLM Flan-PaLM Atlas RePlug RePlug LSR GKP Recitation GRTR Si et al.

Builds on: generated knowledge prompting RePlug recitation-augmented language models OPT Codex

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy