Knowledgeable In-Context Tuning: Exploring and Exploiting Factual Knowledge for In-Context Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Jianing, Wang, Chengyu, Tan, Chuanqi, Huang, Jun, Gao, Ming
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929296933650432
author Wang, Jianing
Wang, Chengyu
Tan, Chuanqi
Huang, Jun
Gao, Ming
author_facet Wang, Jianing
Wang, Chengyu
Tan, Chuanqi
Huang, Jun
Gao, Ming
contents Large language models (LLMs) enable in-context learning (ICL) by conditioning on a few labeled training examples as a text-based prompt, eliminating the need for parameter updates and achieving competitive performance. In this paper, we demonstrate that factual knowledge is imperative for the performance of ICL in three core facets: the inherent knowledge learned in LLMs, the factual knowledge derived from the selected in-context examples, and the knowledge biases in LLMs for output generation. To unleash the power of LLMs in few-shot learning scenarios, we introduce a novel Knowledgeable In-Context Tuning (KICT) framework to further improve the performance of ICL: 1) injecting knowledge into LLMs during continual self-supervised pre-training, 2) judiciously selecting the examples for ICL with high knowledge relevance, and 3) calibrating the prediction results based on prior knowledge. We evaluate the proposed approaches on autoregressive models (e.g., GPT-style LLMs) over multiple text classification and question-answering tasks. Experimental results demonstrate that KICT substantially outperforms strong baselines and improves by more than 13% and 7% on text classification and question-answering tasks, respectively.
format Preprint
id arxiv_https___arxiv_org_abs_2309_14771
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Knowledgeable In-Context Tuning: Exploring and Exploiting Factual Knowledge for In-Context Learning
Wang, Jianing
Wang, Chengyu
Tan, Chuanqi
Huang, Jun
Gao, Ming
Computation and Language
Artificial Intelligence
Large language models (LLMs) enable in-context learning (ICL) by conditioning on a few labeled training examples as a text-based prompt, eliminating the need for parameter updates and achieving competitive performance. In this paper, we demonstrate that factual knowledge is imperative for the performance of ICL in three core facets: the inherent knowledge learned in LLMs, the factual knowledge derived from the selected in-context examples, and the knowledge biases in LLMs for output generation. To unleash the power of LLMs in few-shot learning scenarios, we introduce a novel Knowledgeable In-Context Tuning (KICT) framework to further improve the performance of ICL: 1) injecting knowledge into LLMs during continual self-supervised pre-training, 2) judiciously selecting the examples for ICL with high knowledge relevance, and 3) calibrating the prediction results based on prior knowledge. We evaluate the proposed approaches on autoregressive models (e.g., GPT-style LLMs) over multiple text classification and question-answering tasks. Experimental results demonstrate that KICT substantially outperforms strong baselines and improves by more than 13% and 7% on text classification and question-answering tasks, respectively.
title Knowledgeable In-Context Tuning: Exploring and Exploiting Factual Knowledge for In-Context Learning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2309.14771