about
Thinking LLMs: General Instruction Following with Thought Generation (arxiv.org)
2 points by mnk47 on Oct 15, 2024 | hide | past | pdf | 1 comment on HN

In plain words: The model is taught to think before answering: it tries many thoughts, a judge scores only the final answers, and learns which work best—without human examples. It beat the usual answer-now training on instruction tests, helping marketing, health and general knowledge, not just reasoning.

Abstract

LLMs are typically trained to answer user questions or follow instructions similarly to how human experts respond. However, in the standard alignment framework they lack the basic ability of explicit thinking before answering. Thinking is important for complex questions that require reasoning and planning -- but can be applied to any task. We propose a training method for equipping existing LLMs with such thinking abilities for general instruction following without use of additional human data. We achieve this by an iterative search and optimization procedure that explores the space of possible thought generations, allowing the model to learn how to think without direct supervision. For each instruction, the thought candidates are scored using a judge model to evaluate their responses only, and then optimized via preference optimization. We show that this procedure leads to superior performance on AlpacaEval and Arena-Hard, and shows gains from thinking on non-reasoning categories such as marketing, health and general knowledge, in addition to more traditional reasoning & problem-solving tasks.

Tianhao Wu, Janice Lan, Weizhe Yuan, Jiantao Jiao, Jason Weston, Sainbayar Sukhbaatar
arXiv:2410.10630 · cs.CL, cs.AI · submitted Oct 14, 2024
abstract · pdf · html

add comment on HN
Also discussed: Nov 2024 (1 point, 0 comments) · Oct 2024 (3 points, 1 comment) · Oct 2024 (2 points, 0 comments)

By Meta FAIR, UC Berkeley and NYU

X thread by one of the authors: https://x.com/jaseweston/status/1846011492245672043