← Search

Tomasz Korbak

10 accepted papers

2024

Compositional Preference Models for Aligning LMs

ICLR 2024poster

As language models (LMs) become more capable, it is increasingly important to align them with human preferences. However, the dominant paradigm for training Preference Models (PMs) for that purpose suffers from fundamental limitations, such as lack of transparency and scalability, along with suscept…

2024

The Reversal Curse: LLMs trained on “A is B” fail to learn “B is A”

ICLR 2024poster

We expose a surprising failure of generalization in auto-regressive large language models (LLMs). If a model is trained on a sentence of the form ''_A_ is _B_'', it will not automatically generalize to the reverse direction ''_B_ is _A_''. This is the **Reversal Curse**. For instance, if a model is…

2024

Towards Understanding Sycophancy in Language Models

ICLR 2024poster

Reinforcement learning from human feedback (RLHF) is a popular technique for training high-quality AI assistants. However, RLHF may also encourage model responses that match user beliefs over truthful responses, a behavior known as sycophancy. We investigate the prevalence of sycophancy in RLHF-trai…

2023

Aligning Language Models with Preferences through $f$-divergence Minimization

ICML 2023poster

Aligning language models with preferences can be posed as approximating a target distribution representing some desired behavior. Existing approaches differ both in the functional form of the target distribution and the algorithm used to approximate it. For instance, Reinforcement Learning from Huma…

2023

Pretraining Language Models with Human Preferences

ICML 2023oral

Language models (LMs) are pretrained to imitate text from large and diverse datasets that contain content that would violate human preferences if generated by an LM: falsehoods, offensive comments, personally identifiable information, low-quality or buggy code, among others. Here, we explore alterna…

2022

Controlling Conditional Language Models without Catastrophic Forgetting

ICML 2022spotlight

Machine learning is shifting towards general-purpose pretrained generative models, trained in a self-supervised manner on large amounts of data, which can then be applied to solve a large number of tasks. However, due to their generic training methodology, these models often fail to meet some of the…

2022

On Reinforcement Learning and Distribution Matching for Fine-Tuning Language Models with no Catastrophic Forgetting

NeurIPS 2022accept

The availability of large pre-trained models is changing the landscape of Machine Learning research and practice, moving from a "training from scratch" to a "fine-tuning'' paradigm. While in some applications the goal is to "nudge'' the pre-trained distribution towards preferred outputs, in others i…

2021

Catalytic Role Of Noise And Necessity Of Inductive Biases In The Emergence Of Compositional Communication

NeurIPS 2021poster

Communication is compositional if complex signals can be represented as a combination of simpler subparts. In this paper, we theoretically show that inductive biases on both the training framework and the data are needed to develop a compositional communication. Moreover, we prove that compositiona…

Cited by 23SourcePDFScholar