Mechanism

Adapts LambdaRank from classic Information Retrieval to LLM alignment: given a list of KK responses with real-valued reward labels ψ=(s1,…,sK)\psi = (s_1, \dots, s_K), weights each pairwise DPO log-ratio gradient by the change in Normalized Discounted Cumulative Gain ∣ΔNDCG(i,j)∣|\Delta \text{NDCG}(i, j)|: LLiPO-λ(θ)=−Ex,{yi}[∑si>sj∣ΔNDCG(i,j)∣log⁡σ(βlog⁡πθ(yi∣x)πref(yi∣x)−βlog⁡πθ(yj∣x)πref(yj∣x))]\mathcal{L}_{\text{LiPO-}\lambda}(\theta) = -\mathbb{E}_{x, \{y_i\}}\left[\sum_{s_i > s_j} |\Delta \text{NDCG}(i, j)| \log \sigma\left(\beta \log \frac{\pi_\theta(y_i\mid x)}{\pi_{\text{ref}}(y_i\mid x)} - \beta \log \frac{\pi_\theta(y_j\mid x)}{\pi_{\text{ref}}(y_j\mid x)}\right)\right]

References

  1. LiPO: Listwise Preference Optimization through Learning-to-Rank ↗

    Tianqi Liu, Zhen Qin, Junru Wu, Jiaming Shen, Misha Khalman, Rishabh Joshi et al. · 2024

    All 12 authors

    Tianqi Liu, Zhen Qin, Junru Wu, Jiaming Shen, Misha Khalman, Rishabh Joshi, Yao Zhao, Mohammad Saleh, Simon Baumgartner, Jialu Liu, Peter J. Liu, Xuanhui Wang