Latent Space Chain-of-Embedding Enables Output-free LLM Self-Evaluation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Yiming, Zhang, Pei, Yang, Baosong, Wong, Derek F., Wang, Rui
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929758082695168
author Wang, Yiming
Zhang, Pei
Yang, Baosong
Wong, Derek F.
Wang, Rui
author_facet Wang, Yiming
Zhang, Pei
Yang, Baosong
Wong, Derek F.
Wang, Rui
contents LLM self-evaluation relies on the LLM's own ability to estimate response correctness, which can greatly improve its deployment reliability. In this research track, we propose the Chain-of-Embedding (CoE) in the latent space to enable LLMs to perform output-free self-evaluation. CoE consists of all progressive hidden states produced during the inference time, which can be treated as the latent thinking path of LLMs. We find that when LLMs respond correctly and incorrectly, their CoE features differ, these discrepancies assist us in estimating LLM response correctness. Experiments in four diverse domains and seven LLMs fully demonstrate the effectiveness of our method. Meanwhile, its label-free design intent without any training and millisecond-level computational cost ensures real-time feedback in large-scale scenarios. More importantly, we provide interesting insights into LLM response correctness from the perspective of hidden state changes inside LLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2410_13640
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Latent Space Chain-of-Embedding Enables Output-free LLM Self-Evaluation
Wang, Yiming
Zhang, Pei
Yang, Baosong
Wong, Derek F.
Wang, Rui
Computation and Language
Artificial Intelligence
Machine Learning
LLM self-evaluation relies on the LLM's own ability to estimate response correctness, which can greatly improve its deployment reliability. In this research track, we propose the Chain-of-Embedding (CoE) in the latent space to enable LLMs to perform output-free self-evaluation. CoE consists of all progressive hidden states produced during the inference time, which can be treated as the latent thinking path of LLMs. We find that when LLMs respond correctly and incorrectly, their CoE features differ, these discrepancies assist us in estimating LLM response correctness. Experiments in four diverse domains and seven LLMs fully demonstrate the effectiveness of our method. Meanwhile, its label-free design intent without any training and millisecond-level computational cost ensures real-time feedback in large-scale scenarios. More importantly, we provide interesting insights into LLM response correctness from the perspective of hidden state changes inside LLMs.
title Latent Space Chain-of-Embedding Enables Output-free LLM Self-Evaluation
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.13640