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

Reasoning-Intensive Regression

About

AI researchers and practitioners increasingly apply large language models (LLMs) to what we call reasoning-intensive regression (RiR), i.e., deducing subtle numerical scores from text. Unlike standard language regression tasks such as sentiment or similarity analysis, RiR often appears instead in ad-hoc applications such as rubric-based scoring, modeling dense rewards in complex environments, or domain-specific retrieval, where much deeper analysis of context is required while only limited task-specific training data and computation are available. We cast four realistic problems as RiR tasks to establish an initial benchmark, and use that to test our hypothesis that prompting frozen LLMs and fine-tuning Transformer encoders via gradient descent will both often struggle in RiR. We then propose MENTAT, a simple and lightweight method that combines batch-reflective prompt optimization with neural ensemble learning. MENTAT achieves up to 65% improvement over both baselines, though substantial room remains for future advances.

Diane Tchuindjo, Omar Khattab• 2025

Related benchmarks

TaskDatasetResultRank
Essay GradingEssay Grading (test)
NMSE0.53
26
Mathematical Error DetectionMathematical Error Detection (test)
NMSE0.42
26
Pairwise RAG ComparisonPairwise RAG Comparison (test)
NMSE0.77
26
Instruction FollowingInstruction Following (test)
NMSE0.9
11
Showing 4 of 4 rows

Other info

Follow for update