Uniform Information Density-Based Preference Optimization
Organizations: Department of Computer Science and Engineering Southern University of Science and Technology Shenzhen, Guangdong, China
Abstract
To enable large language models to generate outputs that align with human preferences, optimization algorithms based on the comparison between chosen and rejected output pairs have been developed and been proven effective. Some recent endeavors have utilized certain linguistically-derived assumptions about the preference pairs to guide the design of more precise optimization directions. However, human preferences are complex and hard to measure by simplistic criteria, and it is important to incorporate broader and more sophisticated insights about preferred vs. not-preferred language. In this study, we adopt a well-established psycholinguistic theory, Uniform Information Density, to develop a novel preference optimization algorithm, UIDPO. The basic assumption is that humans prefer to produce language in which information is uniformly distributed, because uniformity induces less cognitive processing effort. To implement this idea into the algorithm, we derive a compound metric for information uniformity by combining the global standard deviation and the local second-order difference, and integrate it to the objective of preference optimization. We compare UIDPO with SimPO, a previous state-of-the-art method, obtaining comparable and sometimes better performance on AlpacaEval2, ArenaHard, and MT-Bench. Additional linguistic analysis on complement clauses also shows that UIDPO can better reflect the tendencies of keeping information uniform in natural language than baselines. Our results show that the subtle human preference over the information rate in language is a helpful source for achieving better human alignment for LLMs. The code repository is published here: https://github.com/CLCS-SUSTech/UIDPO.
Figures & tables
| Benchmark | Baseline Model | Judge Model | Metric |
|---|---|---|---|
| AlpacaEval 2 | GPT-4 1106 Preview | GPT-4 1106 Preview | LC & raw win rate |
| AlpacaEval 2* | GPT-4 1106 Preview | Llama-3.3 70B IT | LC & raw win rate |
| Arena-Hard v0.1 | GPT-4 0314 | Llama-3.3 70B IT | Win rate |
| MT-Bench | – | Llama-3.3 70B IT | 0-10 rating |
| Method | AlpacaEval 2 | AlpacaEval 2* | Arena-Hard v0.1 | MT-Bench | ||
|---|---|---|---|---|---|---|
| LC(%) | WR(%) | LC(%) | WR(%) | WR(%) | Score(0-10) | |
| Instruct | 26.0 | 25.3 | 45.80 | 43.66 | 40.8 | 7.2 |
| SimPO v0.2 | 53.7 | 47.5 | – | – | – | – |
| SimPO reprod. | 51.9 | 47.4 | 60.8 | 56.1 | 66.6 | 7.3 |
| UIDPO | 49.0 | 44.8 | 63.2 | 56.2 | 62.7 | 7.5 |
| Llama-3-8B-Instruct | Mistral-7B-Instruct | |||||||
| LC(%) | WR(%) | SE | avg. length | LC(%) | WR(%) | SE | avg. length | |
| 0.25 | 63.2 | 56.2 | 1.75 | 1799 | 34.3 | 23.2 | 1.49 | 1550 |
| 0.25 orig. | 21.0 | 25.0 | 11.4 | 1893 | 31.29 | 33.29 | 1.66 | 2895 |
| 1.00 | 32.0 | 27.1 | 1.57 | 1743 | 9.91 | 4.66 | 0.74 | 812 |
| 1.00 orig. | 40.8 | 35.8 | 1.69 | 1829 | 5.4 | 7.6 | 0.93 | 7221 |
| Inst. | 45.8 | 43.7 | 1.75 | 1946 | 19.24 | 13.42 | 1.20 | 1500 |
| Llama-3-8B-Instruct | Mistral-7B-Instruct | |||||||
| / | LC(%) | WR(%) | SE | avg. length | LC(%) | WR(%) | SE | avg. length |
| 0.000 | 59.9 | 52.7 | 1.92 | 1839 | 34.0 | 32.4 | 1.65 | 1970 |
| 0.125 | 60.7 | 54.2 | 1.76 | 1847 | 34.1 | 30.7 | 1.63 | 1836 |
| 0.250 | 63.2 | 56.2 | 1.75 | 1799 | 34.3 | 23.2 | 1.49 | 1550 |
| 0.375 | 62.3 | 55.3 | 1.75 | 1816 | 28.0 | 21.4 | 1.45 | 1616 |
| 0.500 | 41.0 | 33.6 | 1.66 | 1745 | 25.7 | 20.0 | 1.41 | 1621 |
| Metric | SFT | SimPO | UIDPO |
|---|---|---|---|
| # of necessary that : | |||
| present/correct ( ) | 495 (0%) | 617 (24.65%) | 676 (36.57%) |
| absent/mistaken ( ) | 186 (0%) | 181 (-2.69%) | 191 (2.69%) |
| # of redundant that : | |||
| absent/correct ( ) | 348 (0%) | 343 (-1.44%) | 345 (-0.86%) |
| present/mistaken ( ) | 3719 (0%) | 3607 (-3.01%) | 3536 (-4.92%) |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | LC(%) | WR(%) | Length |
| RRHF ( Yuan et al., 2023 ) | 37.9 | 31.6 | 1700 |
| SLiC-HF ( Zhao et al., 2023 ) | 33.9 | 32.5 | 1938 |
| DPO ( Rafailov et al., 2024 ) | 48.2 | 47.5 | 2000 |
| IPO ( Azar et al., 2023 ) | 46.8 | 42.4 | 1830 |
| CPO ( Xu et al., 2024 ) | 34.1 | 36.4 | 2086 |
| KTO ( Ethayarajh et al., 2024 ) | 34.1 | 32.1 | 1878 |
| answer extraction | Llama-3-8B-Instruct | SimPO run 1 | SimPO run 2 | UIDPO |
|---|---|---|---|---|
| Level 1 | 0.4098 | 0.3935 | 0.3910 | 0.4122 |
| Level 2 | 0.3983 | 0.3965 | 0.4122 |
| Method | AlpacaEval 2 | AlpacaEval 2* | Arena-Hard v0.1 | ||
|---|---|---|---|---|---|
| SE | avg. length | SE | avg. length | CI(%) | |
| SimPO reprod. | 1.54 | 1861 | 1.75 | 1861 | -2.0/+1.8 |
| UIDPO | 1.50 | 1825 | 1.75 | 1799 | -1.9/+1.7 |
Preference Optimization
Preference optimization aligns AI systems with human or task-specific judgments by training them to favor better-ranked outputs.
See all 150 papers