The Simple Scaling lab introduces s1-32B, a reasoning model that achieves strong performance through supervised fine-tuning on a curated dataset of 1,000 samples and a technique called budget forcing. This approach controls test-time compute by either terminating the model's thinking process or appending "Wait" to extend it, allowing the model to double-check its answers.

  • The s1K dataset contains 1,000 questions paired with reasoning traces distilled from Google's Gemini Thinking Experimental model, selected for difficulty, diversity, and quality.
  • Budget forcing allows precise control over test-time compute; suppressing the end-of-thinking token and adding "Wait" encourages further exploration of the solution path.
  • s1-32B exceeds OpenAI’s o1-preview on competition math questions by up to 27% on MATH and AIME24 benchmarks.
  • Scaling test-time compute allows the model to extrapolate its performance from 50% to 57% on AIME24.

The authors demonstrate that simple supervised fine-tuning combined with direct control over thinking duration is a highly effective and sample-efficient method for achieving test-time scaling and strong reasoning capabilities.