CodeIt: Self-Improving Language Models with Prioritized Hindsight Replay
Large language models are increasingly solving tasks that are commonly
believed to require human-level reasoning ability. However, these models still
perform very poorly on benchmarks of general intelligence such as the
Abstraction and Reasoning Corpus (ARC). In this paper, we approach ARC as a
programming-by-examples problem, and introduce a novel and scalable method for
language model self-improvement called Code Iteration (CodeIt). Our method
iterates between 1) program sampling and hindsight relabeling, and 2) learning
from prioritized experience replay. By relabeling the goal of an episode (i.e.,
the target program output given input) to the realized output produced by the
sampled program, our method effectively deals with the extreme sparsity of
rewards in program synthesis. Applying CodeIt to the ARC dataset, we
demonstrate that prioritized hindsight replay, along with pre-training and
data-augmentation, leads to successful inter-task generalization. CodeIt is the
first neuro-symbolic approach that scales to the full ARC evaluation dataset.
Our method solves 15% of ARC evaluation tasks, achieving state-of-the-art
performance and outperforming existing neural and symbolic baselines.
65c44ced9f6c4689fbac4ecf
Natasha Butt
claimed_verified
2024-10-31T09:55:49.570Z
67235358177d9eae488ff01b
/avatars/b94e228ceae5155898af4a6d1bb06ed3.svg
Natasha Butt
user
NatashaEB
65c44ced9f6c4689fbac4ed0
Blazej Manczak
claimed_verified
2024-02-08T13:01:42.918Z
6054ba9a3efc404ddc7c0c20
https://cdn-avatars.huggingface.co/v1/production/uploads/1677092793838-6054ba9a3efc404ddc7c0c20.jpeg
Blazej
user
Blazej
65c44ced9f6c4689fbac4ed1
Auke Wiggers
65c44ced9f6c4689fbac4ed2
Corrado Rainone
65c44ced9f6c4689fbac4ed3
David Zhang
65c44ced9f6c4689fbac4ed4
Michaël Defferrard
claimed_verified
2024-06-25T20:04:53.550Z
666aefbce7c15edd93433846
https://cdn-avatars.huggingface.co/v1/production/uploads/666aefbce7c15edd93433846/ET_a5cdmDizqJ_EEEzYOY.jpeg
Michaël Defferrard
user
mdeff
65c44ced9f6c4689fbac4ed5
Taco Cohen
CodeIt: Self-Improving Language Models with Prioritized Hindsight Replay
Large language models are increasingly solving tasks that are commonly
believed to require human-level reasoning ability. However, these models still
perform very poorly on benchmarks of general intelligence such as the
Abstraction and Reasoning Corpus (ARC). In this paper, we approach ARC as a
programming-by-examples problem, and introduce a novel and scalable method for
language model self-improvement called Code Iteration (CodeIt). Our method
iterates between 1) program sampling and hindsight relabeling, and 2) learning
from prioritized experience replay. By relabeling the goal of an episode (i.e.,
the target program output given input) to the realized output produced by the
sampled program, our method effectively deals with the extreme sparsity of
rewards in program synthesis. Applying CodeIt to the ARC dataset, we
demonstrate that prioritized hindsight replay, along with pre-training and
data-augmentation, leads to successful inter-task generalization. CodeIt is the
first neuro-symbolic approach that scales to the full ARC evaluation dataset.
Our method solves 15% of ARC evaluation tasks, achieving state-of-the-art
performance and outperforming existing neural and symbolic baselines.
A Code Iteration method for language models achieves state-of-the-art performance on the Abstraction and Reasoning Corpus by combining program sampling, hindsight relabeling, and prioritized experience replay.
programming-by-examples Code Iteration program sampling hindsight relabeling learning from prioritized experience replay extreme sparsity of rewards program synthesis prioritized hindsight replay pre-training data-augmentation inter-task generalization neuro-symbolic approach Abstraction and Reasoning Corpus