Speaker Notes

Learning to Reason with LLMs

Scaling test-time compute through search and reinforcement learning dramatically improves LLM reasoning capabilities, offering a viable alternative to traditional pretraining scaling walls.

Simons Institute for the Theory of Computing · 52:03 · YouTube ↗

Talk · solo speaker  Advanced · concepts and strategy

Highlights

7 clips · 14 min of 52 min
    • Adding 30 seconds of search reduced distance from Nash equilibrium significantly.
    • Search provided the equivalent of scaling parameters by 100,000x in poker.
    • Cultural factors in game theory favored precomputed Nash equilibria over search.
    • Academic competition constraints penalized test-time compute usage.
    • Researchers drastically underestimated the performance gains of search.
    • Raw neural nets alone in Go were not superhuman without search.
    • Monte Carlo tree search provided massive elo gains equivalent to huge scaling.
    • o1 scales test-time compute to dramatically improve math competition performance.
    • Combining reinforcement learning on chain of thought with test-time compute yields compounding gains.
    • Test-time compute scaling on challenging math exams drives accuracy from 20% to over 80%.
    • Applying reinforcement learning to generate a chain of thought creates a compounding performance effect as training compute increases.
    • Reasoning problems like math and Sudoku have a property where verifying a solution is much easier than generating one.
    • When models develop implicit verification capabilities, they can recognize poor paths and actively search for correct solutions.
    • Research in board games shows that increasing test-time compute can directly substitute for training compute scaling.
    • Inference scaling provides an escape hatch against approaching pretraining data and cost walls.
    • Classical logic and SAT solvers are efficient for specific domains but often lack broad generality.
    • The true value of o1 is its ability to generalize reasoning across completely disparate real-world tasks unlike prior domain-specific search models.

Questions this video answers

6 · every answer cites the moment it was said
01 Why did early poker AI struggle against top human professionals despite massive parameter scaling? 3 sources · 3:16

Early poker bots relied on massive pretraining through self-play and acted instantly, but lacked the ability to think carefully during difficult decisions. Human professionals, conversely, would pause and deliberate on complex problems, allowing them to outperform the instant-acting AI. Adding a search mechanism to let the bot think for 30 seconds provided the equivalent performance boost of scaling parameters by 100,000x.

02 Why was search largely ignored by the poker AI research community for years? 3 sources · 10:24

Cultural factors in the game theory community prioritized precomputing Nash equilibria rather than using search. Additionally, scaling inference compute made experiments painfully expensive and slow, especially given poker's high variance. Researchers were also heavily incentivized by competitions that enforced strict time limits on test-time compute.

03 How does simple consensus fall short when scaling inference compute in large language models? 3 sources · 18:48

Consensus involves querying an LLM multiple times and taking the most common response, which yields initial accuracy gains on math benchmarks. However, the performance quickly flatlines after querying the model multiple times, failing to provide additional returns. Thus, simple consensus is limited as a generalized method for scaling inference compute.

04 Why does inference compute work so well for certain reasoning tasks? 2 sources · 29:08

Inference compute succeeds on tasks where verifying a good solution is much easier than generating one, such as math proofs, coding, and puzzles. Large language models can implicitly perform verification during their chain of thought to recognize mistakes and try alternative strategies. This generator-verifier gap mirrors techniques successfully used in games like chess and Go.

05 Can scaling inference compute help bypass traditional pretraining scaling walls? 2 sources · 33:19

Increasing inference compute offers a viable alternative to escalating pretraining costs and data requirements. Research shows that increasing test-time compute by 15x is equivalent to a 10x increase in training time compute. While inference costs and latencies will rise, users are often willing to pay that price for breakthroughs on difficult problems.

06 What is the primary advantage of o1 over domain-specific planning techniques? 2 sources · 47:05

While domain-specific planning algorithms can achieve superhuman performance in restricted environments like poker or Hanabi, they often fail to generalize outside those domains. In contrast, o1 provides a general-purpose reasoning technique that successfully applies across diverse complex tasks. This broad applicability makes it a powerful tool that can save users days of work.

Section-by-section summary

11 parts

Scaling parameters alone failed in poker

Early poker bots relied on massive pretraining and self-play scaling without test-time search. Despite increasing parameters year over year, these bots still lost heavily to top human professionals.

Test-time search outperforms massive parameter scaling

Adding a search algorithm to let the poker bot think for thirty seconds drastically improved its strategy. This test-time compute yielded performance gains equivalent to a 100,000-fold increase in model parameters.

Cultural norms and competition rules hindered search

Game theory traditions and rigid competition constraints discouraged researchers from exploring test-time search. Furthermore, the community severely underestimated the massive performance multiplier that search could provide.

Monte Carlo tree search powers AlphaGo Zero

AlphaGo Zero achieved superhuman performance not through raw neural network scale alone, but by combining it with Monte Carlo tree search. Rules of thumb suggest bypassing search would require astronomical scaling to match those results.

Scaling inference compute in large language models

While consensus sampling offers limited improvements, OpenAI o1 introduces a more general approach by using reinforcement learning for chain of thought. This enables effective scaling of test-time compute, resulting in massive accuracy gains on complex reasoning benchmarks.

o1 shows massive jumps across diverse complex domains

The o1 model achieves breakthrough scores on expert-level benchmarks spanning competition mathematics, coding, and PhD-level science questions. These performance boosts extend surprisingly beyond STEM into subjects like the LSAT and literature.

Chain of thought enables systematic error correction

Unlike prompting models to output answers immediately, o1 uses large-scale RL to produce a prolonged, high-quality chain of thought. This allows the model to break down complex problems, test multiple strategies, and actively correct its own mistakes.

Verification gaps make inference compute effective

Reasoning tasks like math and coding share a property where verifying a solution is much easier than generating one. When models develop implicit verification capabilities, spending more inference compute allows them to systematically search for correct solutions.

Inference scaling bypasses traditional pretraining walls

Research in board games shows that increasing test-time compute can substitute for training compute scaling. This effectively bypasses concerns about hitting pretraining data walls, justifying higher latency and cost for high-value problems.

The bitter lesson favors compute-scaling over heuristics

History shows that general algorithms scaling with compute consistently beat clever human-crafted heuristics. Academic research must shift away from short-term publication hacks toward techniques that effectively leverage massive computational scale.

Q&A explores academic barriers and search generality

Discussion addresses the stark compute disparity between industry and academia, as well as comparisons to classical logic and SAT solvers. The speaker emphasizes that o1's primary advantage is its broad domain generality rather than narrow search rules.

0:00 / 52:03
k play/pause · f fullscreen · m mute
n next · p previous clip · t highlights/full
< > slower/faster · / search what was said
  • Counterfactual regret minimization1× · first 2:14

    The underlying algorithm used by early poker bots trained via self-play.

  • Monte Carlo tree search4× · first 13:13

    A search algorithm combined with raw neural nets to leverage extra computation at inference time in Go bots.

  • Chain of thought7× · first 21:43

    A step-by-step reasoning process in natural language generated by a model before providing a final answer.

  • Test-time compute18× · first 10:56

    Additional computational resources expended during inference to improve model reasoning and problem-solving.