When Weak LLMs Speak with Confidence, Preference Alignment Gets Stronger
Organizations: EPFL · Sharif University of Technology
Abstract
Preference alignment is an essential step in adapting large language models (LLMs) to human values, but existing approaches typically depend on costly human annotations or large-scale API-based models. We explore whether a weak LLM can instead act as an effective annotator. We surprisingly find that selecting only a subset of a weak LLM's highly confident samples leads to substantially better performance than using full human annotations. Building on this insight, we propose Confidence-Weighted Preference Optimization (CW-PO), a general framework that re-weights training samples by a weak LLM's confidence and can be applied across different preference optimization objectives. Notably, the model aligned by CW-PO with just 20% of human annotations outperforms the model trained with 100% of annotations under standard DPO. These results suggest that weak LLMs, when paired with confidence weighting, can dramatically reduce the cost of preference alignment while even outperforming methods trained on fully human-labeled data.
Figures & tables
| OPT-125M OPT-13B | |||||||||
| DPO | IPO | rDPO | |||||||
| Dataset | Human | WS-DPO | CW-DPO | Human | WS-DPO | CW-IPO | Human | WS-DPO | CW-rDPO |
| HH-RLHF | 56.9 | 56.7 | 61.3 | 58.2 | 62.8 | 63.5 | 55.9 | 57.6 | 63.0 |
| TL;DR | 57.0 | 53.5 | 56.6 | 53.3 | 49.7 | 54.6 | 54.2 | 47.7 | 61.4 |
| UFB | 61.3 | 63.4 | 63.1 | 63.4 | 61.3 | 66.4 | 58.9 | 61.2 | 63.7 |
| Avg. | 58.4 | 57.9 | 60.3 | 58.3 | 57.9 | 61.5 | 56.3 | 55.5 | 62.7 |
| Dataset | Strong | OPT | Strong | Qwen | ||||
| Human | WS-DPO | CW-DPO | Human | WS-DPO | CW-DPO | |||
| HH-RLHF | 1.3B | 71.5 | 66.7 | 69.9 | 1.5B | 53.4 | 55.8 | 63.3 |
| 2.7B | 55.1 | 58.5 | 60.3 | 3B | 66.0 | 63.3 | 73.3 | |
| 6.7B | 56.1 | 62.8 | 67.6 | 7B | 71.1 | 72.0 | 75.2 | |
| 13B | 56.9 | 56.7 | 61.3 | 14B | 78.8 | 81.4 | 80.6 | |
| Avg. | 59.9 | 61.2 | 64.8 | Avg. | 67.3 | 68.1 | 73.1 | |
| Dataset | OPT-125M OPT-1.3B | Qwen2.5-0.5B Qwen2.5-7B | ||
| Human (100%) | CW-DPO | Human (100%) | CW-DPO | |
| Harmless | 69.2 | 72.9 (+3.7) | 65.7 | 72.0 (+6.3) |
| Helpful | 70.2 | 72.7 (+2.5) | 58.5 | 70.8 (+12.3) |
| HH-RLHF | 71.9 | 69.9 ( 2.0) | 72.7 | 75.2 (+2.5) |
| TL;DR | 54.2 | 59.5 (+5.3) | 63.4 | 64.4 (+1.0) |
| Avg. | 66.4 | 68.8 (+2.4) | 65.1 | 70.6 (+5.5) |
| Dataset | ||||
| Harmless | 72.0 | 70.3 | 68.6 | 69.1 |
| Helpful | 70.8 | 67.8 | 67.4 | 68.7 |
| HH-RLHF | 75.2 | 70.1 | 69.2 | 72.5 |
| Avg. | 72.7 | 69.4 | 68.4 | 70.1 |
| Model | Dataset | DPO | SFT+DPO | BT |
| OPT-125M | Harmless | 55.2 | 56.3 | 69.1 |
| Helpful | 54.1 | 55.4 | 64.2 | |
| HH-RLHF | 50.8 | 52.1 | 63.8 | |
| Qwen-0.5B | Harmless | 56.1 | 57.1 | 65.3 |
| Helpful | 55.2 | 56.0 | 63.1 | |
| HH-RLHF | 51.4 | 52.6 | 63.2 |
Appendix figures & tables23 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Value |
| Model(s) | OPT-125M; Qwen2.5-0.5B |
| Training epochs | 5 |
| Optimizer | Adam |
| Learning rate | |
| Per-device train batch size | 32 |
| Gradient accumulation steps | 1 |
| Parameter | Value |
| Training epochs | 5 |
| Learning rate | |
| LR scheduler | cosine |
| Warmup steps | 100 |
| Weight decay | 0.05 |
| Optimizer | paged_adamw_32bit |
| Parameter | Value |
| Training epochs | 3 |
| Learning rate | |
| LR scheduler | cosine |
| Warmup steps | 100 |
| Weight decay | 0.05 |
| Optimizer | paged_adamw_32bit |
| Setting | 30% | 50% | 75% | 100% | Human |
| Harmless | 72.3 | 67.9 | 67.7 | 67.4 | 66.2 |
| Helpful | 70.1 | 68.0 | 67.2 | 66.8 | 66.4 |
| HH-RLHF | 67.7 | 67.5 | 66.9 | 65.9 | 64.4 |
| Avg. | 70.03 | 67.80 | 67.27 | 66.70 | 65.67 |
| Setting | 30% | 50% | 75% | 100% | Human |
| Harmless | 70.6 | 68.4 | 66.7 | 66.9 | 65.1 |
| Helpful | 58.7 | 55.2 | 51.5 | 52.3 | 54.4 |
| HH-RLHF | 71.3 | 68.4 | 64.9 | 63.2 | 61.8 |
| Avg. | 66.87 | 64.00 | 61.03 | 60.80 | 60.43 |
| Weak Model Size | 0.5B | 1.5B | 3B | 7B |
| Harmlessness | 63.5 | 65.9 | 66.6 | 67.1 |
| Helpfulness | 63.2 | 64.7 | 65.3 | 67.2 |
| TL;DR | 60.7 | 61.0 | 61.6 | 62.9 |
| Avg. | 62.5 | 63.9 | 64.5 | 65.7 |
| Dataset % | 10% | 20% | 30% | 40% | 50% |
| Harmlessness | 62.6 | 65.5 | 67.2 | 67.8 | 67.1 |
| Helpfulness | 61.9 | 63.4 | 65.3 | 66.4 | 65.3 |
| Both | 56.7 | 60.2 | 61.9 | 62.6 | 62.2 |
| Avg. | 60.4 | 63.1 | 64.8 | 65.6 | 64.9 |
| Human | WS-DPO | CW-DPO | |
| Dataset | OPT-125M Qwen-2.5-14B | ||
| Harmless | 67.3 | 66.4 | 72.2 |
| Helpful | 54.1 | 55.9 | 57.2 |
| HH-RLHF | 78.8 | 77.7 | 79.7 |
| Qwen-2.5-0.5B OPT-13B | |||
| Harmless | 74.3 | 72.4 | 75.4 |
| OPT-125M OPT-13B | |||||||||
| DPO | IPO | rDPO | |||||||
| Dataset | Human | WS-DPO | CW-DPO | Human | WS-DPO | CW-IPO | Human | WS-DPO | CW-rDPO |
| HH-RLHF | 43.9 | 44.6 | 51.9 | 43.7 | 33.5 | 45.8 | 46.4 | 42.1 | 55.1 |
| TL;DR | 55.7 | 52.8 | 52.8 | 59.5 | 53.4 | 63.2 | 56.6 | 54.5 | 61.0 |
| UFB | 59.8 | 64.7 | 64.3 | 62.1 | 60.2 | 63.9 | 55.3 | 56.8 | 59.4 |
| Avg. | 53.13 | 54.03 | 56.33 | 55.10 | 49.03 | 57.63 | 52.77 | 51.13 | 58.50 |
| OPT-125M OPT-13B | |||||||||
| DPO | IPO | rDPO | |||||||
| Dataset | Human | WS-DPO | CW-DPO | Human | WS-DPO | CW-IPO | Human | WS-DPO | CW-rDPO |
| HH-RLHF | 56.88 ± 4.43 | 55.01 ± 3.58 | 60.79 ± 2.98 | 58.87 ± 3.18 | 61.79 ± 1.75 | 64.30 ± 2.24 | 55.92 ± 4.90 | 57.63 ± 3.59 | 62.95 ± 1.96 |
| TL;DR | 57.79 ± 3.47 | 53.45 ± 2.39 | 53.85 ± 5.33 | 51.54 ± 1.64 | 49.30 ± 2.14 | 54.80 ± 2.33 | 55.04 ± 0.28 | 48.14 ± 3.97 | 62.12 ± 4.80 |
| UFB | 62.12 ± 2.18 | 63.72 ± 2.93 | 63.61 ± 3.17 | 62.63 ± 1.07 | 60.96 ± 1.15 | 65.17 ± 2.32 | 57.11 ± 2.18 | 63.41 ± 2.35 | 63.95 ± 2.33 |
| Avg. | 58.9 ± 2.29 | 57.4 ± 4.52 | 59.4 ± 4.10 | 57.7 ± 4.61 | 57.4 ± 5.70 | 61.4 ± 4.70 | 56.0 ± 0.85 | 56.4 ± 6.29 | 63.0 ± 0.75 |
| Dataset | Online DPO | Online CW-DPO | Online WS-DPO |
| HH-RLHF | 85.3 | 55.1 | 56.3 |
| TL;DR | 66.9 | 46.2 | 44.1 |
| Dataset | CW-DPO (online) | CW-DPO (offline) |
| HH-RLHF | 55.1 | 61.3 |
| TL;DR | 46.2 | 56.6 |
| Domain | Strong | Human | WS-DPO | CW-DPO |
| askacademia | 7B | 53.6 | 55.1 | 56.4 |
| 14B | 50.6 | 53.3 | 55.1 | |
| askbaking | 7B | 49.1 | 50.9 | 51.4 |
| 14B | 56.6 | 54.5 | 58.4 | |
| askengineers | 7B | 56.9 | 58.0 | 59.1 |
| 14B | 54.9 | 57.2 | 55.4 |
| mixture | Harmless | Helpful | ||
| CW-DPO | WS-DPO | CW-DPO | WS-DPO | |
| 50% + 50% | 60.5 | 57.1 | 63.6 | 58.9 |
| 80% + 20% | 61.6 (+1.1) | 59.3 (+2.2) | 60.5 (-3.1) | 54.8 (-4.1) |
| 20% + 80% | 56.1 (-4.4) | 52.2 (-4.9) | 67.1 (+3.5) | 62.4 (+3.5) |
| Human | WS-DPO | PairRM | CW-DPO | |
| HH-RLHF | 71.1 | 72.0 | 71.6 | 75.2 |
| TL;DR | 61.2 | 60.1 | 62.2 | 64.4 |
| Avg. | 66.15 | 66.05 | 66.90 | 69.80 |
Preference Alignment
Preference alignment studies how AI systems are trained or steered to produce outputs that match human or user priorities.
See all 109 papers