Ideally, please use the following format for the reward evaluations, ```python def format(messages): format_text = "[INST] You must read the following conversation carefully and rate the assistant's response from score 0-100 in these aspects: helpfulness, correctness, coherence, honesty, complexity.\n" for message in messages: if message['role'] == "user": format_text = format_text + "\nUser: " + message['content'] elif message['role'] == 'assistant': format_text = format_text + "\nAssistant: " + message['content'] return format_text ``` The reward model can be used for iterative SFT/DPO ``` @article{dong2023raft, title={Raft: Reward ranked finetuning for generative foundation model alignment}, author={Dong, Hanze and Xiong, Wei and Goyal, Deepanshu and Pan, Rui and Diao, Shizhe and Zhang, Jipeng and Shum, Kashun and Zhang, Tong}, journal={arXiv preprint arXiv:2304.06767}, year={2023} } @article{xiong2023gibbs, title={Gibbs sampling from human feedback: A provable kl-constrained framework for rlhf}, author={Xiong, Wei and Dong, Hanze and Ye, Chenlu and Zhong, Han and Jiang, Nan and Zhang, Tong}, journal={arXiv preprint arXiv:2312.11456}, year={2023} } ```