Why it matters

Reward model quality determines RLHF quality. Understanding shapes RM training.

Advertisement

The architecture

Bradley-Terry loss on preference pairs.

Model outputs reward for chosen > rejected.

Reward model trainingPreference pairchosen, rejectedModel scoresr(chosen), r(rejected)BT losslog-sigmoidReward model shape mirrors policy; usually LM + scalar head
RM training.
Advertisement

How it works end to end

Architecture: LM base + scalar head.

Loss: -log sigmoid(r(chosen) - r(rejected)).

Regularization: prevent reward hacking.