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

Aligner: Efficient Alignment by Learning to Correct

About

With the rapid development of large language models (LLMs) and ever-evolving practical requirements, finding an efficient and effective alignment method has never been more critical. However, the tension between the complexity of current alignment methods and the need for rapid iteration in deployment scenarios necessitates the development of a model-agnostic alignment approach that can operate under these constraints. In this paper, we introduce Aligner, a novel and simple alignment paradigm that learns the correctional residuals between preferred and dispreferred answers using a small model. Designed as a model-agnostic, plug-and-play module, Aligner can be directly applied to various open-source and API-based models with only one-off training, making it suitable for rapid iteration. Notably, Aligner can be applied to any powerful, large-scale upstream models. Moreover, it can even iteratively bootstrap the upstream models using corrected responses as synthetic human preference data, breaking through the model's performance ceiling. Our experiments demonstrate performance improvements by deploying the same Aligner model across 11 different LLMs, evaluated on the 3H dimensions (helpfulness, harmlessness, and honesty). Specifically, Aligner-7B has achieved an average improvement of 68.9% in helpfulness and 23.8% in harmlessness across the tested LLMs while also effectively reducing hallucination. In the Alpaca-Eval leaderboard, stacking Aligner-2B on GPT-4 Turbo improved its LC Win Rate from 55.0% to 58.3%, surpassing GPT-4 Omni's 57.5% Win Rate (community report).

Jiaming Ji, Boyuan Chen, Hantao Lou, Donghai Hong, Borong Zhang, Xuehai Pan, Juntao Dai, Tianyi Qiu, Yaodong Yang• 2024

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Accuracy42.74
983
Mathematical ReasoningMATH
Accuracy15.62
643
Code GenerationMBPP
Accuracy30.88
120
Truthful QATruthful QA
Accuracy46.61
83
Code GenerationMBPP+
Accuracy30.54
75
Question AnsweringMMLU
Accuracy52.56
62
Question AnsweringGPQA Diamond
Accuracy24.15
62
Question AnsweringMMLU-Pro
Accuracy32.39
56
Question AnsweringMMLU-Redux
Accuracy48.59
42
Empathy evaluationE-Dialogue
Empathy Score99.4
33
Showing 10 of 26 rows

Other info

Code

Follow for update