Uncertainty Quantification for In-Context Learning of Large Language Models
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2024
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866914732786581504 |
|---|---|
| author | Ling, Chen Zhao, Xujiang Zhang, Xuchao Cheng, Wei Liu, Yanchi Sun, Yiyou Oishi, Mika Osaki, Takao Matsuda, Katsushi Ji, Jie Bai, Guangji Zhao, Liang Chen, Haifeng |
| author_facet | Ling, Chen Zhao, Xujiang Zhang, Xuchao Cheng, Wei Liu, Yanchi Sun, Yiyou Oishi, Mika Osaki, Takao Matsuda, Katsushi Ji, Jie Bai, Guangji Zhao, Liang Chen, Haifeng |
| contents | In-context learning has emerged as a groundbreaking ability of Large Language Models (LLMs) and revolutionized various fields by providing a few task-relevant demonstrations in the prompt. However, trustworthy issues with LLM's response, such as hallucination, have also been actively discussed. Existing works have been devoted to quantifying the uncertainty in LLM's response, but they often overlook the complex nature of LLMs and the uniqueness of in-context learning. In this work, we delve into the predictive uncertainty of LLMs associated with in-context learning, highlighting that such uncertainties may stem from both the provided demonstrations (aleatoric uncertainty) and ambiguities tied to the model's configurations (epistemic uncertainty). We propose a novel formulation and corresponding estimation method to quantify both types of uncertainties. The proposed method offers an unsupervised way to understand the prediction of in-context learning in a plug-and-play fashion. Extensive experiments are conducted to demonstrate the effectiveness of the decomposition. The code and data are available at: https://github.com/lingchen0331/UQ_ICL. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_10189 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Uncertainty Quantification for In-Context Learning of Large Language Models Ling, Chen Zhao, Xujiang Zhang, Xuchao Cheng, Wei Liu, Yanchi Sun, Yiyou Oishi, Mika Osaki, Takao Matsuda, Katsushi Ji, Jie Bai, Guangji Zhao, Liang Chen, Haifeng Computation and Language Machine Learning In-context learning has emerged as a groundbreaking ability of Large Language Models (LLMs) and revolutionized various fields by providing a few task-relevant demonstrations in the prompt. However, trustworthy issues with LLM's response, such as hallucination, have also been actively discussed. Existing works have been devoted to quantifying the uncertainty in LLM's response, but they often overlook the complex nature of LLMs and the uniqueness of in-context learning. In this work, we delve into the predictive uncertainty of LLMs associated with in-context learning, highlighting that such uncertainties may stem from both the provided demonstrations (aleatoric uncertainty) and ambiguities tied to the model's configurations (epistemic uncertainty). We propose a novel formulation and corresponding estimation method to quantify both types of uncertainties. The proposed method offers an unsupervised way to understand the prediction of in-context learning in a plug-and-play fashion. Extensive experiments are conducted to demonstrate the effectiveness of the decomposition. The code and data are available at: https://github.com/lingchen0331/UQ_ICL. |
| title | Uncertainty Quantification for In-Context Learning of Large Language Models |
| topic | Computation and Language Machine Learning |
| url | https://arxiv.org/abs/2402.10189 |