Towards Reliable Latent Knowledge Estimation in LLMs: Zero-Prompt Many-Shot Based Factual Knowledge Extraction

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wu, Qinyuan, Khan, Mohammad Aflah, Das, Soumi, Nanda, Vedant, Ghosh, Bishwamittra, Kolling, Camila, Speicher, Till, Bindschaedler, Laurent, Gummadi, Krishna P., Terzi, Evimaria
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916527496757248
author Wu, Qinyuan
Khan, Mohammad Aflah
Das, Soumi
Nanda, Vedant
Ghosh, Bishwamittra
Kolling, Camila
Speicher, Till
Bindschaedler, Laurent
Gummadi, Krishna P.
Terzi, Evimaria
author_facet Wu, Qinyuan
Khan, Mohammad Aflah
Das, Soumi
Nanda, Vedant
Ghosh, Bishwamittra
Kolling, Camila
Speicher, Till
Bindschaedler, Laurent
Gummadi, Krishna P.
Terzi, Evimaria
contents In this paper, we focus on the challenging task of reliably estimating factual knowledge that is embedded inside large language models (LLMs). To avoid reliability concerns with prior approaches, we propose to eliminate prompt engineering when probing LLMs for factual knowledge. Our approach, called Zero-Prompt Latent Knowledge Estimator (ZP-LKE), leverages the in-context learning ability of LLMs to communicate both the factual knowledge question as well as the expected answer format. Our knowledge estimator is both conceptually simpler (i.e., doesn't depend on meta-linguistic judgments of LLMs) and easier to apply (i.e., is not LLM-specific), and we demonstrate that it can surface more of the latent knowledge embedded in LLMs. We also investigate how different design choices affect the performance of ZP-LKE. Using the proposed estimator, we perform a large-scale evaluation of the factual knowledge of a variety of open-source LLMs, like OPT, Pythia, Llama(2), Mistral, Gemma, etc. over a large set of relations and facts from the Wikidata knowledge base. We observe differences in the factual knowledge between different model families and models of different sizes, that some relations are consistently better known than others but that models differ in the precise facts they know, and differences in the knowledge of base models and their finetuned counterparts. Code available at: https://github.com/QinyuanWu0710/ZeroPrompt_LKE
format Preprint
id arxiv_https___arxiv_org_abs_2404_12957
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Reliable Latent Knowledge Estimation in LLMs: Zero-Prompt Many-Shot Based Factual Knowledge Extraction
Wu, Qinyuan
Khan, Mohammad Aflah
Das, Soumi
Nanda, Vedant
Ghosh, Bishwamittra
Kolling, Camila
Speicher, Till
Bindschaedler, Laurent
Gummadi, Krishna P.
Terzi, Evimaria
Computation and Language
Machine Learning
In this paper, we focus on the challenging task of reliably estimating factual knowledge that is embedded inside large language models (LLMs). To avoid reliability concerns with prior approaches, we propose to eliminate prompt engineering when probing LLMs for factual knowledge. Our approach, called Zero-Prompt Latent Knowledge Estimator (ZP-LKE), leverages the in-context learning ability of LLMs to communicate both the factual knowledge question as well as the expected answer format. Our knowledge estimator is both conceptually simpler (i.e., doesn't depend on meta-linguistic judgments of LLMs) and easier to apply (i.e., is not LLM-specific), and we demonstrate that it can surface more of the latent knowledge embedded in LLMs. We also investigate how different design choices affect the performance of ZP-LKE. Using the proposed estimator, we perform a large-scale evaluation of the factual knowledge of a variety of open-source LLMs, like OPT, Pythia, Llama(2), Mistral, Gemma, etc. over a large set of relations and facts from the Wikidata knowledge base. We observe differences in the factual knowledge between different model families and models of different sizes, that some relations are consistently better known than others but that models differ in the precise facts they know, and differences in the knowledge of base models and their finetuned counterparts. Code available at: https://github.com/QinyuanWu0710/ZeroPrompt_LKE
title Towards Reliable Latent Knowledge Estimation in LLMs: Zero-Prompt Many-Shot Based Factual Knowledge Extraction
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2404.12957