In this paper, we propose a taxonomy of five knowledge statuses and introduce KScope, a hierarchical testing framework to characterize LLM knowledge status.
We apply KScope to nine LLMs across four datasets and investigate three questions:
- How does context update LLMs' knowledge status?
- What context features drive the desired knowledge update?
- What context augmentations work best across knowledge statuses?
Please note that we are unable to upload the datasets and model outputs due to size constraints.
However, all results can be easily reproduced using the provided code.
- Dataset Preparation:
- Run
1-Preparation.ipynbto prepare the four evaluation datasets.
- Run
- Hyperparameter Search:
- Run
2-Hyperparameter.ipynbto determine the number of question paraphrases and sample responses required for consistent characterization of LLM knowledge status.
- Run
- Model Response Collection:
- Run
response.shto collect responses from the nine LLMs on the four datasets in the multi-choice setting with gold context. - To test with noisy context, replace "evidence" with "retrieved" in
response.py. - To switch to the open-ended setting, use the provided commented instruction prompts in
response.py.
- Run
- Knowledge Status Characterization (Q1):
- Apply KScope to the collected responses to characterize LLM knowledge status:
- Run
status.shfor the multi-choice setting (with either gold or noisy context). - Run
status_open.shfor the open-ended setting (with gold context). - Use
3-Status.ipynband4-Settings.ipynbto further explore and visualize context-induced shifts in knowledge status.
- Feature Importance Analysis (Q2):
- Run
uncertainty.shto compute perplexity and entropy. - Run
5-Feature.ipynbto compute additional features. - Run
6-Analysis.ipynbto analyze feature importance.
- Run
- Context Augmentation Strategies (Q3):
- Run
7-GPT.ipynbto collect GPT-4o responses. - Run
8-Augmentation.ipynbto evaluate the effectiveness of different augmentation strategies.
- Run
@inproceedings{xiao2025kscope,
title={KScope: A Framework for Characterizing the Knowledge Status of Language Models},
author={Xiao, Yuxin and Chen, Shan and Gallifant, Jack and Bitterman, Danielle and Hartvigsen, Thomas and Ghassemi, Marzyeh},
booktitle={NeurIPS},
year={2025}
}