Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Context-DPO: Aligning Language Models for Context-Faithfulness

About

Reliable responses from large language models (LLMs) require adherence to user instructions and retrieved information. While alignment techniques help LLMs align with human intentions and values, improving context-faithfulness through alignment remains underexplored. To address this, we propose $\textbf{Context-DPO}$, the first alignment method specifically designed to enhance LLMs' context-faithfulness. We introduce $\textbf{ConFiQA}$, a benchmark that simulates Retrieval-Augmented Generation (RAG) scenarios with knowledge conflicts to evaluate context-faithfulness. By leveraging faithful and stubborn responses to questions with provided context from ConFiQA, our Context-DPO aligns LLMs through direct preference optimization. Extensive experiments demonstrate that our Context-DPO significantly improves context-faithfulness, achieving 35% to 280% improvements on popular open-source models. Further analysis demonstrates that Context-DPO preserves LLMs' generative capabilities while providing interpretable insights into context utilization. Our code and data are released at https://github.com/byronBBL/Context-DPO

Baolong Bi, Shaohan Huang, Yiwei Wang, Tianchi Yang, Zihan Zhang, Haizhen Huang, Lingrui Mei, Junfeng Fang, Zehao Li, Furu Wei, Weiwei Deng, Feng Sun, Qi Zhang, Shenghua Liu• 2024

Related benchmarks

TaskDatasetResultRank
Open-book generation under knowledge conflictConFiQA 1,500 subset
Ps Score81.07
32
Instruction FollowingMQuAKE
Accuracy82.5
24
Context-faithful Question AnsweringConFiQA--
24
Retrieval FollowingConFiQA QA 1.0 (test)
Pc92.3
20
Retrieval FollowingConFiQA MR 1.0 (test)
Pc61.2
20
Retrieval FollowingConFiQA MC 1.0 (test)
Pc54.9
20
Retrieval FollowingNatural Questions (test)
Pc98.4
20
Open-book generation under knowledge conflictConFiQA MR 1,500
Ps Score59.8
16
Context-faithful Multi-hop ReasoningConFiQA MR--
8
Context-faithful ReasoningConFiQA MC--
8
Showing 10 of 10 rows

Other info

Code

Follow for update