Trustworthy Alignment of Retrieval-Augmented Large Language Models via Reinforcement Learning

Zongmeng Zhang, Yufeng Shi, Jinhua Zhu, Wen-gang Zhou, Xiang Qi +2 more
10/1/2024

No official code found. Best implementation: zmzhang2000/trustworthy-alignment (12 stars, last commit 2024-09-02).

Abstract

Trustworthiness is an essential prerequisite for the real-world application of large language models. In this paper, we focus on the trustworthiness of language models with respect to retrieval augmentation. Despite being supported with external evidence, retrieval-augmented generation still suffers from hallucinations, one primary cause of which is the conflict between contextual and parametric knowledge. We deem that retrieval-augmented language models have the inherent capabilities of supplying response according to both contextual and parametric knowledge. Inspired by aligning language models with human preference, we take the first step towards aligning retrieval-augmented language models to a status where it responds relying merely on the external evidence and disregards the interference of parametric knowledge. Specifically, we propose a reinforcement learning based algorithm Trustworthy-Alignment, theoretically and experimentally demonstrating large language models' capability of reaching a trustworthy status without explicit supervision on how to respond. Our work highlights the potential of large language models on exploring its intrinsic abilities by its own and expands the application scenarios of alignment from fulfilling human preference to creating trustworthy agents.

View on arXivView PDF

Code Implementations(1)

Official repository for Trustworthy Alignment of Retrieval-Augmented Large Language Models via Reinforcement Learning

122Shell, PythonDec 19, 20232 years ago

Cite this paper

@article{zhang2024trustworthy,
  title  = {Trustworthy Alignment of Retrieval-Augmented Large Language Models via Reinforcement Learning},
  author = {Zongmeng Zhang and Yufeng Shi and Jinhua Zhu and Wen-gang Zhou and Xiang Qi and Peng Zhang and Houqiang Li},
  year   = {2024},
  eprint = {2410.16843},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2410.16843},
  journal = {ICML 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.16843.svg)](https://codeofpaper.com/paper/2410.16843)

Discussion