$\mathbf{(N,K)}$-Puzzle: A Cost-Efficient Testbed for Benchmarking Reinforcement Learning Algorithms in Generative Language Model

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Yufeng, Chen, Liyu, Liu, Boyi, Yang, Yingxiang, Cui, Qiwen, Tao, Yunzhe, Yang, Hongxia
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914710810525696
author Zhang, Yufeng
Chen, Liyu
Liu, Boyi
Yang, Yingxiang
Cui, Qiwen
Tao, Yunzhe
Yang, Hongxia
author_facet Zhang, Yufeng
Chen, Liyu
Liu, Boyi
Yang, Yingxiang
Cui, Qiwen
Tao, Yunzhe
Yang, Hongxia
contents Recent advances in reinforcement learning (RL) algorithms aim to enhance the performance of language models at scale. Yet, there is a noticeable absence of a cost-effective and standardized testbed tailored to evaluating and comparing these algorithms. To bridge this gap, we present a generalized version of the 24-Puzzle: the $(N,K)$-Puzzle, which challenges language models to reach a target value $K$ with $N$ integers. We evaluate the effectiveness of established RL algorithms such as Proximal Policy Optimization (PPO), alongside novel approaches like Identity Policy Optimization (IPO) and Direct Policy Optimization (DPO).
format Preprint
id arxiv_https___arxiv_org_abs_2403_07191
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle $\mathbf{(N,K)}$-Puzzle: A Cost-Efficient Testbed for Benchmarking Reinforcement Learning Algorithms in Generative Language Model
Zhang, Yufeng
Chen, Liyu
Liu, Boyi
Yang, Yingxiang
Cui, Qiwen
Tao, Yunzhe
Yang, Hongxia
Machine Learning
Artificial Intelligence
Computation and Language
Recent advances in reinforcement learning (RL) algorithms aim to enhance the performance of language models at scale. Yet, there is a noticeable absence of a cost-effective and standardized testbed tailored to evaluating and comparing these algorithms. To bridge this gap, we present a generalized version of the 24-Puzzle: the $(N,K)$-Puzzle, which challenges language models to reach a target value $K$ with $N$ integers. We evaluate the effectiveness of established RL algorithms such as Proximal Policy Optimization (PPO), alongside novel approaches like Identity Policy Optimization (IPO) and Direct Policy Optimization (DPO).
title $\mathbf{(N,K)}$-Puzzle: A Cost-Efficient Testbed for Benchmarking Reinforcement Learning Algorithms in Generative Language Model
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2403.07191