The loss function of reward model.

Author: huzechuanCreated Jan 31, 2023Updated Feb 12, 2023

Hi, I am confused that the loss function of ChatGPT's reward model takes as input the difference of two responses and then passes a sigmoid function. However, the loss function in this repo only takes one response as input and uses the ranking score as a label to calculate the CE loss. Is there an advantage to this?

Source: lucidrains/PaLM-rlhf-pytorch