Bayesian Example Selection Improves In-Context Learning for Speech, Text, and Visual Modalities

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Siyin, Yang, Chao-Han Huck, Wu, Ji, Zhang, Chao
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910488556732416
author Wang, Siyin
Yang, Chao-Han Huck
Wu, Ji
Zhang, Chao
author_facet Wang, Siyin
Yang, Chao-Han Huck
Wu, Ji
Zhang, Chao
contents Large language models (LLMs) can adapt to new tasks through in-context learning (ICL) based on a few examples presented in dialogue history without any model parameter update. Despite such convenience, the performance of ICL heavily depends on the quality of the in-context examples presented, which makes the in-context example selection approach a critical choice. This paper proposes a novel Bayesian in-Context example Selection method (ByCS) for ICL. Extending the inference probability conditioned on in-context examples based on Bayes' theorem, ByCS focuses on the inverse inference conditioned on test input. Following the assumption that accurate inverse inference probability (likelihood) will result in accurate inference probability (posterior), in-context examples are selected based on their inverse inference results. Diverse and extensive cross-tasking and cross-modality experiments are performed with speech, text, and image examples. Experimental results show the efficacy and robustness of our ByCS method on various models, tasks and modalities.
format Preprint
id arxiv_https___arxiv_org_abs_2404_14716
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bayesian Example Selection Improves In-Context Learning for Speech, Text, and Visual Modalities
Wang, Siyin
Yang, Chao-Han Huck
Wu, Ji
Zhang, Chao
Computation and Language
Artificial Intelligence
Computer Vision and Pattern Recognition
Sound
Audio and Speech Processing
Large language models (LLMs) can adapt to new tasks through in-context learning (ICL) based on a few examples presented in dialogue history without any model parameter update. Despite such convenience, the performance of ICL heavily depends on the quality of the in-context examples presented, which makes the in-context example selection approach a critical choice. This paper proposes a novel Bayesian in-Context example Selection method (ByCS) for ICL. Extending the inference probability conditioned on in-context examples based on Bayes' theorem, ByCS focuses on the inverse inference conditioned on test input. Following the assumption that accurate inverse inference probability (likelihood) will result in accurate inference probability (posterior), in-context examples are selected based on their inverse inference results. Diverse and extensive cross-tasking and cross-modality experiments are performed with speech, text, and image examples. Experimental results show the efficacy and robustness of our ByCS method on various models, tasks and modalities.
title Bayesian Example Selection Improves In-Context Learning for Speech, Text, and Visual Modalities
topic Computation and Language
Artificial Intelligence
Computer Vision and Pattern Recognition
Sound
Audio and Speech Processing
url https://arxiv.org/abs/2404.14716