Decoupling Task-Solving and Output Formatting in LLM Generation

Read full text
Authors: Haikang Deng, Po-Nien Kung, Nanyun Peng
Cited by: 0

Abstract

Large language models (LLMs) are increasingly adept at solving complex problems, such as mathematical reasoning and automatic evaluation.However, performance often degrades when prompts intertwine task instructions with rigid formatting requirements.This entanglement creates competing goals for the model, hindering its reasoning capabilities.To address this, we introduce DECO-G, a decoding framework that explicitly decouples format adherence from problem solving.DECO-G delegates format adherence to a separate Format Estimation Module (FEM), which performs probabilistic lookahead to estimate future format compliance rate and reweighs token probabilities, allowing the LLM to focus solely on task resolution.To make this approach both practical and efficient, we introduce three key innovations: instruction-aware distillation, a flexible trie-building algorithm, and HMM state pruning.Experiments across mathematical reasoning, event argument extraction, and LLM-asa-judge demonstrate that DECO-G constantly gains over prompting or structured generation baselines, with guaranteed format compliance.

Fields of Study & Subjects

Topic Modeling Natural Language Processing Techniques Text Readability and Simplification

Publication details

Pages
16764–16781
Publication date
1 January 2026
Language
EN