Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

UNA: A Unified Supervised Framework for Efficient LLM Alignment Across Feedback Types

About

RL alignment methods, including RLHF and DPO, are primarily based on pairwise preference data. Although scalar or score-based feedback has been collected in some settings, it is rarely used directly, and preference magnitude information is typically ignored. Furthermore, current alignment frameworks offer limited capability for unifying heterogeneous supervision signals, making it difficult to jointly leverage diverse data types within a single training paradigm. This limitation constrains the richness and scalability of the alignment process. To address this gap, we propose a \textbf{UN}ified \textbf{A}lignment (UNA) framework capable of training across different types of feedback, including binary, pairwise, and score-based, through a generalized implicit reward function. The reward function is theoretically proved to be the optimal policy by the log sum inequality. Extensive experiments on classical benchmarks consistently demonstrate the advantage of the proposed unified framework with typical LLM base models.

Zhichao Wang, Bin Bi, Can Huang, Shiva Kumar Pentyala, Zixu James Zhu, Sitaram Asur, Na Claire Cheng, Cheng Wan, Dong Nie, Lingzi Hong• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande
Accuracy74.03
1442
Instruction FollowingIFEval
IFEval Accuracy39.17
836
ReasoningARC
Accuracy68.69
245
Mathematical ReasoningMATH Hard
Accuracy42.6
198
Science Question AnsweringARC
ARC Accuracy55.2
76
Commonsense ReasoningHellaSwag
HellaSwag Score86.86
62
Multitask Language UnderstandingMMLU-Pro
pass@148.94
38
Factuality Question AnsweringTruthfulQA
Score66.92
28
LLM AlignmentAlpacaEval--
24
LLM AlignmentAlpacaEval Length-Controlled (test)
LC Win Rate8.78
12
Showing 10 of 11 rows

Other info

Follow for update