Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Playing Psychic: Using Thought Trees to Predict Reasoning Models Accuracy on Coding Tasks

About

Recent advances in large language models (LLMs) have shown that test-time scaling can substantially improve model performance on complex tasks, particularly in the coding domain. Under this paradigm, models use a larger token budget during inference to generate intermediate reasoning traces before producing a final answer. However, current evaluations primarily rely on competitive programming benchmarks, which may not capture the full range of reasoning abilities. In this work, we perform a systematic study of frontier reasoning models to understand their performance on real-world coding benchmarks. To gain more insights into the performance of such models, we devise a programmatic way to {\em automatically generate} coding tasks of arbitrary difficulty and structure from existing benchmarks. Using this framework, our analysis reveals that the structure of a reasoning trace, not just its contents, is a strong predictor of correctness. Motivated by this, we propose structured thought-trees as means to represent reasoning traces. To illustrate their use, we train a lightweight classifier on features extracted from thought-trees to predict trace correctness, and demonstrate that flagging and retrying structurally anomalous traces based on the extracted features yields consistent gains at lower complexity levels.

Jiaxin Fang, Runyuan He, Sahil Bhatia, Neel Gajare, Alvin Cheung• 2026

Related benchmarks

TaskDatasetResultRank
Block InfillingSAFIM--
6
Code TranslationCodeLingua Java → C--
6
Code TranslationCodeLingua Python → C--
6
Code TranslationCodeLingua Java → Fortran--
6
Output PredictionCRUXEval--
6
Reasoning failure prediction and recoveryCRUXEval L2
Accuracy77
4
Reasoning failure predictionCodeLingua (L1)
Accuracy73
2
Reasoning failure predictionCodeLingua (L2)
Accuracy75
2
Reasoning failure predictionCodeLingua (L3)
Accuracy76
2
Reasoning failure prediction and recoveryCRUXEval L1
Accuracy89
2
Showing 10 of 13 rows

Other info

Follow for update