DPO step

DPO step

Loss

×

tensor

Its gradient updates the policy, at learning rate 10−510^{-5}.

shape
L\mathcal L [1]
Sources

Contextual evidence from Add, 0.2 on the NLL

§4.1.4, Direct Preference Optimization: DPO on the SFT models with learning rate 10⁻⁵ and β = 0.1; special formatting tokens, including header and termination tokens, are masked out of the loss in chosen and rejected responses; an NLL term with scaling coefficient 0.2 is added on the chosen sequences.

meta · arXiv 2407.21783 / Llama3-Herd-2407.21783v3.pdf
Pinned revision v3 ↗ · p. 16