Mechanism

Formulates LLM alignment from scalar reward annotations {r(x,yk)}k=1K\{r(x, y_k)\}_{k=1}^K as a Noise Contrastive Estimation (NCE) problem, training the policy log-ratio βlog⁡πθ(yk∣x)πref(yk∣x)\beta \log \frac{\pi_\theta(y_k\mid x)}{\pi_{\text{ref}}(y_k\mid x)} to match the softmax distribution of scalar rewards exp⁡(r(x,yk)/α)∑jexp⁡(r(x,yj)/α)\frac{\exp(r(x,y_k)/\alpha)}{\sum_j \exp(r(x,y_j)/\alpha)} without throwing away absolute scalar reward magnitudes.

References

  1. Noise Contrastive Alignment of Language Models with Explicit Rewards ↗

    Huayu Chen, Guande He, Lifan Yuan, Ganqu Cui, Hang Su, Jun Zhu · 2024