Efficient Exploration for LLMs
We present evidence of substantial benefit from efficient exploration in
gathering human feedback to improve large language models. In our experiments,
an agent sequentially generates queries while fitting a reward model to the
feedback received. Our best-performing agent generates queries using double
Thompson sampling, with uncertainty represented by an epistemic neural network.
Our results demonstrate that efficient exploration enables high levels of
performance with far fewer queries. Further, both uncertainty estimation and
the choice of exploration scheme play critical roles.
65bc65ce259bc6caeba86ac1
Vikranth Dwaracherla
65bc65ce259bc6caeba86ac2
Seyed Mohammad Asghari
65bc65ce259bc6caeba86ac3
Botao Hao
65bc65ce259bc6caeba86ac4
Benjamin Van Roy
Efficient Exploration for LLMs
We present evidence of substantial benefit from efficient exploration in
gathering human feedback to improve large language models. In our experiments,
an agent sequentially generates queries while fitting a reward model to the
feedback received. Our best-performing agent generates queries using double
Thompson sampling, with uncertainty represented by an epistemic neural network.
Our results demonstrate that efficient exploration enables high levels of
performance with far fewer queries. Further, both uncertainty estimation and
the choice of exploration scheme play critical roles.
Efficient exploration using double Thompson sampling and epistemic neural network uncertainty estimation significantly improves large language models with reduced query numbers.
double Thompson sampling epistemic neural network reward model