Enhancing Language Model with Both Human and Artificial Intelligence Feedback Data
The proliferation of language models has marked a significant advancement in technology and industry in recent years. The training of these models largely involves human feedback, a procedure that faces challenges including intensive resource demands and subjective human preferences. In this work, we incorporate feedback provided by artificial intelligence (AI) models instead of relying entirely on human feedback. We propose a simulation optimization framework to train the language model. The objective function for training is approximated using feedback from both human and AI models. We employ the method of control variate to reduce the variance of the approximated objective function. Additionally, we provide a procedure for deciding the sample size to acquire preferences from both human and AI models. Numerical experiments demonstrate that our proposed procedure enhances the performance of the language model.
Authors
- Zeyu Zheng (ORCID: https://orcid.org/0000-0001-5653-152X)
- Jingxu Xu (ORCID: https://orcid.org/0000-0002-1327-6490)
- Jinghai He (ORCID: https://orcid.org/0000-0002-5458-2048)
- Jingshen Wang
- Haoting Zhang (ORCID: https://orcid.org/0000-0001-9416-2024)
Institutions
- University of California, Berkeley (US)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/wsc63780.2024.10838929
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 0.00