When Correct Isn't Usable: Improving Structured Output Reliability in Small Language Models
Not on the map. Why?
Abstract (arXiv)
Deployed language models must produce outputs that are both correct and format-compliant. We study this structured-output reliability gap using two mathematical benchmarks -- GSM8K and MATH -- as a controlled testbed: ground truth is unambiguous and the output contract is strict (JSON with required fields). We evaluate three 7-9B models under five prompting strategies and report output accuracy -- the joint event of mathematical correctness and valid JSON structure -- as the primary metric. A systematic format failure emerges: NAIVE prompting (no system prompt) achieves up to 85% task accuracy on GSM8K but 0% output accuracy across all models and datasets. REFERENCE prompting (a minimal hand-written JSON format prompt) fares little better, yielding 0% output accuracy for two of four models tested. Constrained decoding enforces syntactic validity but incurs 3.6x-8.2x latency overhead and in several settings degrades task performance substantially. To overcome this limitation, we developed AloLab, an iterative system-prompt optimizer (meta-agent: Claude Sonnet 4.5) requiring only black-box API access to the target model; it reaches 84-87% output accuracy on GSM8K and 34-40% on MATH across five independent runs per model, with 29/30 paired McNemar comparisons against the best static prompt significant at p < 0.05, at near-NAIVE inference latency and without model fine-tuning. The same format failure extends to GPT-4o (OpenAI, 2024), a proprietary closed-source model: REFERENCE achieves 0% output accuracy due to systematic markdown-fence wrapping, while AloLab reaches 95.2% [94.8, 95.6]. An ablation replacing the Sonnet 4.5 meta-agent with Claude 3 Haiku reduces mean output accuracy to 61.0% and increases run-to-run standard deviation from <1 pp to 21.8 pp, confirming that meta-agent capability is a primary driver of optimization quality.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 23 of its 26 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
23 of 26 references.
References · 23
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2024Qwen2 Technical Reportcited by 2,076
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2022Large Language Models Are Human-Level Prompt Engineerscited by 1,020
- 2023Large Language Models as Optimizerscited by 830
- 2023Automatic Prompt Optimization with "Gradient Descent" and Beam Searchcited by 628
- 2024GSM-Symbolic: Understanding the Limitations of Mathematical Reasoning in Large Language Modelscited by 533
- 2023Efficient Guided Generation for Large Language Modelscited by 192
- 2024SynCode: LLM Generation with Grammar Augmentationcited by 66
11 more
- 2025JSONSchemaBench: A Rigorous Benchmark of Structured Outputs for Language Modelscited by 66
- 2024Grammar-Aligned Decodingcited by 54
- 2025Towards Reasoning Ability of Small Language Modelscited by 29
- 2025Verifiable Format Control for Large Language Model Generationscited by 16
- 2025Constrained Decoding of Diffusion LLMs with Context-Free Grammarscited by 13
- 2026Thinking Before Constraining: A Unified Decoding Framework for Large Language Modelscited by 5
- 2026LLMStructBench: Benchmarking Large Language Model Structured Data Extractioncited by 4
- 2026Generalizable Self-Evolving Memory for Automatic Prompt Optimizationcited by 4
- 2025Earley-Driven Dynamic Pruning for Efficient Structured Decodingcited by 3
- 2026In-Context Reinforcement Learning for Tool Use in Large Language Modelscited by 3
- 2025Lost in Space: Finding the Right Tokens for Structured Outputcited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.