When to Trust LLMs: Aligning Confidence with Response Quality

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Tao, Shuchang, Yao, Liuyi, Ding, Hanxing, Xie, Yuexiang, Cao, Qi, Sun, Fei, Gao, Jinyang, Shen, Huawei, Ding, Bolin
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914959484518400
author Tao, Shuchang
Yao, Liuyi
Ding, Hanxing
Xie, Yuexiang
Cao, Qi
Sun, Fei
Gao, Jinyang
Shen, Huawei
Ding, Bolin
author_facet Tao, Shuchang
Yao, Liuyi
Ding, Hanxing
Xie, Yuexiang
Cao, Qi
Sun, Fei
Gao, Jinyang
Shen, Huawei
Ding, Bolin
contents Despite the success of large language models (LLMs) in natural language generation, much evidence shows that LLMs may produce incorrect or nonsensical text. This limitation highlights the importance of discerning when to trust LLMs, especially in safety-critical domains. Existing methods often express reliability by confidence level, however, their effectiveness is limited by the lack of objective guidance. To address this, we propose CONfidence-Quality-ORDer-preserving alignment approach (CONQORD), which leverages reinforcement learning guided by a tailored dual-component reward function. This function integrates quality reward and order-preserving alignment reward functions. Specifically, the order-preserving reward incentivizes the model to verbalize greater confidence for responses of higher quality to align the order of confidence and quality. Experiments demonstrate that CONQORD significantly improves the alignment performance between confidence and response accuracy, without causing over-cautious. Furthermore, the aligned confidence provided by CONQORD informs when to trust LLMs, and acts as a determinant for initiating the retrieval process of external knowledge. Aligning confidence with response quality ensures more transparent and reliable responses, providing better trustworthiness.
format Preprint
id arxiv_https___arxiv_org_abs_2404_17287
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle When to Trust LLMs: Aligning Confidence with Response Quality
Tao, Shuchang
Yao, Liuyi
Ding, Hanxing
Xie, Yuexiang
Cao, Qi
Sun, Fei
Gao, Jinyang
Shen, Huawei
Ding, Bolin
Computation and Language
Despite the success of large language models (LLMs) in natural language generation, much evidence shows that LLMs may produce incorrect or nonsensical text. This limitation highlights the importance of discerning when to trust LLMs, especially in safety-critical domains. Existing methods often express reliability by confidence level, however, their effectiveness is limited by the lack of objective guidance. To address this, we propose CONfidence-Quality-ORDer-preserving alignment approach (CONQORD), which leverages reinforcement learning guided by a tailored dual-component reward function. This function integrates quality reward and order-preserving alignment reward functions. Specifically, the order-preserving reward incentivizes the model to verbalize greater confidence for responses of higher quality to align the order of confidence and quality. Experiments demonstrate that CONQORD significantly improves the alignment performance between confidence and response accuracy, without causing over-cautious. Furthermore, the aligned confidence provided by CONQORD informs when to trust LLMs, and acts as a determinant for initiating the retrieval process of external knowledge. Aligning confidence with response quality ensures more transparent and reliable responses, providing better trustworthiness.
title When to Trust LLMs: Aligning Confidence with Response Quality
topic Computation and Language
url https://arxiv.org/abs/2404.17287