Mechanism

Samples KK candidate responses {y1,…,yK}\{y_1, \dots, y_K\} with external scores {r1,…,rK}\{r_1, \dots, r_K\} and optimizes a pairwise ranking margin hinge loss over length-normalized conditional log-probabilities pi=1∣yi∣∑tlog⁡πθ(yi,t∣x,yi,<t)p_i = \frac{1}{|y_i|}\sum_t \log \pi_\theta(y_{i,t}\mid x, y_{i,<t}) plus an SFT loss on the highest-ranked candidate yi⋆y_{i^\star}: LRRHF(θ)=∑ri<rjmax⁡(0,pi−pj)−λ∑tlog⁡πθ(yi⋆,t∣x,yi⋆,<t)\mathcal{L}_{\text{RRHF}}(\theta) = \sum_{r_i < r_j} \max(0, p_i - p_j) - \lambda \sum_{t} \log \pi_\theta(y_{i^\star, t}\mid x, y_{i^\star, <t})

References

  1. RRHF: Rank Responses to Align Language Models with Human Feedback without tears ↗

    Zheng Yuan, Hongyi Yuan, Chuanqi Tan, Wei Wang, Songfang Huang, Fei Huang · 2023