Critic-Guided Decoding for Controlled Text Generation
Not on the map. Why?
Abstract (arXiv)
Steering language generation towards objectives or away from undesired content has been a long-standing goal in utilizing language models (LM). Recent work has demonstrated reinforcement learning and weighted decoding as effective approaches to achieve a higher level of language control and quality with pros and cons. In this work, we propose a novel critic decoding method for controlled language generation (CriticControl) that combines the strengths of reinforcement learning and weighted decoding. Specifically, we adopt the actor-critic framework to train an LM-steering critic from non-differentiable reward models. And similar to weighted decoding, our method freezes the language model and manipulates the output token distribution using called critic, improving training efficiency and stability. Evaluation of our method on three controlled generation tasks, namely topic control, sentiment control, and detoxification, shows that our approach generates more coherent and well-controlled texts than previous methods. In addition, CriticControl demonstrates superior generalization ability in zero-shot settings. Human evaluation studies also corroborate our findings.
Key facts. Cited by 45 papers in the map (data through 7 Oct 2026). Builds on 50 of its 68 references that are in the map. First posted on arXiv 21 Dec 2022.
Builds on
50 of 68 references.
References · 50
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2020Learning to summarize from human feedbackcited by 2,562
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,124
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 1,916
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
38 more
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2016Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translationcited by 1,471
- 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,235
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,227
- 2018Hierarchical Neural Story Generationcited by 1,182
- 2015A Diversity-Promoting Objective Function for Neural Conversation Modelscited by 1,068
- 2018Neural Network Acceptability Judgmentscited by 952
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 793
- 2019CTRL: A Conditional Transformer Language Model for Controllable Generationcited by 754
- 2015Sequence Level Training with Recurrent Neural Networkscited by 568
- 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 434
- 2020GeDi: Generative Discriminator Guided Sequence Generationcited by 377
- 2021FUDGE: Controlled Text Generation With Future Discriminatorscited by 374
- 2017A Deep Reinforced Model for Abstractive Summarizationcited by 364
- 2021Recursively Summarizing Books with Human Feedbackcited by 287
- 2016Deep Reinforcement Learning for Dialogue Generationcited by 286
- 2017Toward Controlled Generation of Textcited by 271
- 2022Is Reinforcement Learning (Not) for Natural Language Processing: Benchmarks, Baselines, and Building Blocks for Natural Language Policy Optimizationcited by 244
- 2022Quark: Controllable Text Generation with Reinforced Unlearningcited by 236
- 2016An Actor-Critic Algorithm for Sequence Predictioncited by 158
- 2022Offline RL for Natural Language Generation with Implicit Language Q Learningcited by 137
- 2018Learning to Write with Cooperative Discriminatorscited by 119
- 2017Deceiving Google's Perspective API Built for Detecting Toxic Commentscited by 99
- 2019Transforming Delete, Retrieve, Generate Approach for Controlled Text Style Transfercited by 81
- 2021Towards Facilitating Empathic Conversations in Online Mental Health Support: A Reinforcement Learning Approachcited by 75
- 2021What Changes Can Large-scale Language Models Bring? Intensive Study on HyperCLOVA: Billions-scale Korean Generative Pretrained Transformerscited by 75
- 2022Controllable Natural Language Generation with Contrastive Prefixescited by 75
- 2017Reinforcement Learning for Bandit Neural Machine Translation with Simulated Human Feedbackcited by 71
- 2021Controlled Text Generation as Continuous Optimization with Multiple Constraintscited by 70
- 2022Mix and Match: Learning-free Controllable Text Generation using Energy Language Modelscited by 66
- 2018Learning to Extract Coherent Summary via Deep Reinforcement Learningcited by 63
- 2022Controllable Text Generation with Neurally-Decomposed Oraclecited by 41
- 2019Reinforcement Learning Based Text Style Transfer without Parallel Training Corpuscited by 38
- 2022DIRECTOR: Generator-Classifiers For Supervised Language Modelingcited by 38
- 2019Towards Coherent and Engaging Spoken Dialog Response Generation Using Automatic Conversation Evaluatorscited by 34
- 2022A Distributional Lens for Multi-Aspect Controllable Text Generationcited by 33
- 2022Efficient Reinforcement Learning for Unsupervised Controlled Text Generationcited by 7
- 2020Discrete Action On-Policy Learning with Action-Value Criticcited by 3
Built on by
Citing papers per year: 2023: 6; 2024: 20; 2025: 9; 2026: 10.
45 papers.
Sort:
By area:
- 2023Bias and Fairness in Large Language Models: A Surveycited by 626
- 2024From r to Q^*: Your Language Model is Secretly a Q-Functioncited by 242
- 2023Controlled Decoding from Language Modelscited by 133
- 2024Bias and Unfairness in Information Retrieval Systems: New Challenges in the LLM Eracited by 97
- 2023Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Modelcited by 79
- 2023Controlled Text Generation via Language Model Arithmeticcited by 51
- 2025FairSteer: Inference Time Debiasing for LLMs with Dynamic Activation Steeringcited by 37
- 2024Weak-to-Strong Search: Align Large Language Models via Searching over Small Language Modelscited by 32
- 2024Can LLMs Recognize Toxicity? A Structured Investigation Framework and Toxicity Metriccited by 23
- 2024SafeInfer: Context Adaptive Decoding Time Safety Alignment for Large Language Modelscited by 22
- 2024Mitigating Hallucinations in Large Vision-Language Models via Summary-Guided Decodingcited by 21
- 2024Automatic Metrics in Natural Language Generation: A Survey of Current Evaluation Practicescited by 19
33 more
- 2024Inference-Time Language Model Alignment via Integrated Value Guidancecited by 19
- 2024Metasql: A Generate-then-Rank Framework for Natural Language to SQL Translationcited by 17
- 2024Controlled Text Generation for Large Language Model with Dynamic Attribute Graphscited by 16
- 2023LifeTox: Unveiling Implicit Toxicity in Life Advicecited by 14
- 2025Drift: Decoding-time Personalized Alignments with Implicit User Preferencescited by 13
- 2024AdvisorQA: Towards Helpful and Harmless Advice-seeking Question Answering with Collective Intelligencecited by 12
- 2023Tackling Bias in Pre-trained Language Models: Current Trends and Under-represented Societiescited by 11
- 2024Potential and Challenges of Model Editing for Social Debiasingcited by 10
- 2024Controllable Text Generation in the Instruction-Tuning Eracited by 7
- 2024Large Language Models can be Strong Self-Detoxifierscited by 7
- 2024Guaranteed Generation from Large Language Modelscited by 7
- 2025Speculative Safety-Aware Decodingcited by 7
- 2024FreeCtrl: Constructing Control Centers with Feedforward Layers for Learning-Free Controllable Text Generationcited by 6
- 2024Decoding Game: On Minimax Optimality of Heuristic Text Generation Strategiescited by 6
- 2025BiasFilter: An Inference-Time Debiasing Framework for Large Language Modelscited by 6
- 2025Evolutionary Guided Decoding: Iterative Value Refinement for LLMscited by 3
- 2026UniARM: Towards a Unified Autoregressive Reward Model for Multi-Objective Test-Time Alignmentcited by 3
- 2024A Framework for Real-time Safeguarding the Text Generation of Large Language Modelcited by 2
- 2024LaSEr-Edit: Localized Span-level Error Editing with Energy-based Localizationcited by 2
- 2024Learning Goal-Conditioned Representations for Language Reward Modelscited by 2
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignmentcited by 2
- 2026Test-Time Detoxification without Training or Learning Anythingcited by 2
- 2026DLM-SWAI: Steering Diffusion Language Models Before They Unmaskcited by 2
- 2025Mimicking How Humans Interpret Out-of-Context Sentences Through Controlled Toxicity Decodingcited by 1
- 2026Limits of n-gram Style Control for LLMs via Logit-Space Injectioncited by 1
- 2025Accelerating Reinforcement Learning Algorithms Convergence using Pre-trained Large Language Models as Tutors With Advice Reusingcited by 0
- 2025Test-time reward-guided alignment of language models by importance sampling on pre-logit spacecited by 0
- 2026OLLM: Options-based Large Language Modelscited by 0
- 2026Guidance Is Not a Hyperparameter: Learning Dynamic Control in Diffusion Language Modelscited by 0
- 2026A Comparative Study of Controlled Text Generation Systems Using Level-Playing-Field Evaluation Principlescited by 0
- 2026Latent Action Reparameterization for Efficient Agent Inferencecited by 0
- 2026Robust Critics: Defending LLMs Against Multi-Turn Attackscited by 0
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Modelscited by 0
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Models14 Aug 2026
- 2026DLM-SWAI: Steering Diffusion Language Models Before They Unmask28 May 2026
- 2026Robust Critics: Defending LLMs Against Multi-Turn Attacks24 May 2026
- 2026Latent Action Reparameterization for Efficient Agent Inference18 May 2026
- 2026A Comparative Study of Controlled Text Generation Systems Using Level-Playing-Field Evaluation Principles12 May 2026
- 2026Guidance Is Not a Hyperparameter: Learning Dynamic Control in Diffusion Language Models8 May 2026
- 2026OLLM: Options-based Large Language Models21 Apr 2026
- 2026UniARM: Towards a Unified Autoregressive Reward Model for Multi-Objective Test-Time Alignment10 Feb 2026
- 2026Test-Time Detoxification without Training or Learning Anything14 Jan 2026
- 2026Limits of n-gram Style Control for LLMs via Logit-Space Injection12 Jan 2026
- 2025Test-time reward-guided alignment of language models by importance sampling on pre-logit space30 Oct 2025
- 2025Accelerating Reinforcement Learning Algorithms Convergence using Pre-trained Large Language Models as Tutors With Advice Reusing10 Sep 2025
33 more
- 2025Speculative Safety-Aware Decoding25 Aug 2025
- 2025BiasFilter: An Inference-Time Debiasing Framework for Large Language Models28 May 2025
- 2025FairSteer: Inference Time Debiasing for LLMs with Dynamic Activation Steering20 Apr 2025
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignment9 Apr 2025
- 2025Mimicking How Humans Interpret Out-of-Context Sentences Through Controlled Toxicity Decoding11 Mar 2025
- 2025Evolutionary Guided Decoding: Iterative Value Refinement for LLMs4 Mar 2025
- 2025Drift: Decoding-time Personalized Alignments with Implicit User Preferences20 Feb 2025
- 2024Mitigating Hallucinations in Large Vision-Language Models via Summary-Guided Decoding17 Oct 2024
- 2024Guaranteed Generation from Large Language Models9 Oct 2024
- 2024Decoding Game: On Minimax Optimality of Heuristic Text Generation Strategies4 Oct 2024
- 2024Large Language Models can be Strong Self-Detoxifiers4 Oct 2024
- 2024Inference-Time Language Model Alignment via Integrated Value Guidance26 Sep 2024
- 2024Automatic Metrics in Natural Language Generation: A Survey of Current Evaluation Practices17 Aug 2024
- 2024Learning Goal-Conditioned Representations for Language Reward Models18 Jul 2024
- 2024LaSEr-Edit: Localized Span-level Error Editing with Energy-based Localization30 Jun 2024
- 2024SafeInfer: Context Adaptive Decoding Time Safety Alignment for Large Language Models18 Jun 2024
- 2024FreeCtrl: Constructing Control Centers with Feedforward Layers for Learning-Free Controllable Text Generation14 Jun 2024
- 2024Weak-to-Strong Search: Align Large Language Models via Searching over Small Language Models29 May 2024
- 2024Controllable Text Generation in the Instruction-Tuning Era2 May 2024
- 2024A Framework for Real-time Safeguarding the Text Generation of Large Language Model29 Apr 2024
- 2024From r to Q^*: Your Language Model is Secretly a Q-Function18 Apr 2024
- 2024AdvisorQA: Towards Helpful and Harmless Advice-seeking Question Answering with Collective Intelligence18 Apr 2024
- 2024Bias and Unfairness in Information Retrieval Systems: New Challenges in the LLM Era17 Apr 2024
- 2024Metasql: A Generate-then-Rank Framework for Natural Language to SQL Translation27 Feb 2024
- 2024Potential and Challenges of Model Editing for Social Debiasing21 Feb 2024
- 2024Controlled Text Generation for Large Language Model with Dynamic Attribute Graphs17 Feb 2024
- 2024Can LLMs Recognize Toxicity? A Structured Investigation Framework and Toxicity Metric10 Feb 2024
- 2023Tackling Bias in Pre-trained Language Models: Current Trends and Under-represented Societies3 Dec 2023
- 2023Controlled Text Generation via Language Model Arithmetic24 Nov 2023
- 2023LifeTox: Unveiling Implicit Toxicity in Life Advice16 Nov 2023
- 2023Controlled Decoding from Language Models25 Oct 2023
- 2023Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model14 Oct 2023
- 2023Bias and Fairness in Large Language Models: A Survey2 Sep 2023
- 2024From r to Q^*: Your Language Model is Secretly a Q-Functioncited by 242
- 2023Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Modelcited by 79
- 2024Weak-to-Strong Search: Align Large Language Models via Searching over Small Language Modelscited by 32
- 2024SafeInfer: Context Adaptive Decoding Time Safety Alignment for Large Language Modelscited by 22
- 2024Inference-Time Language Model Alignment via Integrated Value Guidancecited by 19
- 2025Drift: Decoding-time Personalized Alignments with Implicit User Preferencescited by 13
- 2024Large Language Models can be Strong Self-Detoxifierscited by 7
- 2026UniARM: Towards a Unified Autoregressive Reward Model for Multi-Objective Test-Time Alignmentcited by 3
- 2024Learning Goal-Conditioned Representations for Language Reward Modelscited by 2
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignmentcited by 2
- 2025Test-time reward-guided alignment of language models by importance sampling on pre-logit spacecited by 0
- 2026UniARM: Towards a Unified Autoregressive Reward Model for Multi-Objective Test-Time Alignment10 Feb 2026
- 2025Test-time reward-guided alignment of language models by importance sampling on pre-logit space30 Oct 2025
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignment9 Apr 2025
- 2025Drift: Decoding-time Personalized Alignments with Implicit User Preferences20 Feb 2025
- 2024Large Language Models can be Strong Self-Detoxifiers4 Oct 2024
- 2024Inference-Time Language Model Alignment via Integrated Value Guidance26 Sep 2024
- 2024Learning Goal-Conditioned Representations for Language Reward Models18 Jul 2024
- 2024SafeInfer: Context Adaptive Decoding Time Safety Alignment for Large Language Models18 Jun 2024
- 2024Weak-to-Strong Search: Align Large Language Models via Searching over Small Language Models29 May 2024
- 2024From r to Q^*: Your Language Model is Secretly a Q-Function18 Apr 2024
- 2023Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model14 Oct 2023
- 2025Speculative Safety-Aware Decodingcited by 7
- 2025Speculative Safety-Aware Decoding25 Aug 2025
- 2023Bias and Fairness in Large Language Models: A Surveycited by 626
- 2023Controlled Decoding from Language Modelscited by 133
- 2024Bias and Unfairness in Information Retrieval Systems: New Challenges in the LLM Eracited by 97
- 2023Controlled Text Generation via Language Model Arithmeticcited by 51
- 2025FairSteer: Inference Time Debiasing for LLMs with Dynamic Activation Steeringcited by 37
- 2024Can LLMs Recognize Toxicity? A Structured Investigation Framework and Toxicity Metriccited by 23
- 2024Automatic Metrics in Natural Language Generation: A Survey of Current Evaluation Practicescited by 19
- 2024Metasql: A Generate-then-Rank Framework for Natural Language to SQL Translationcited by 17
- 2024Controlled Text Generation for Large Language Model with Dynamic Attribute Graphscited by 16
- 2023LifeTox: Unveiling Implicit Toxicity in Life Advicecited by 14
- 2024AdvisorQA: Towards Helpful and Harmless Advice-seeking Question Answering with Collective Intelligencecited by 12
- 2023Tackling Bias in Pre-trained Language Models: Current Trends and Under-represented Societiescited by 11
12 most cited of 26 All 26 (JSON)
- 2026A Comparative Study of Controlled Text Generation Systems Using Level-Playing-Field Evaluation Principles12 May 2026
- 2026OLLM: Options-based Large Language Models21 Apr 2026
- 2026Test-Time Detoxification without Training or Learning Anything14 Jan 2026
- 2026Limits of n-gram Style Control for LLMs via Logit-Space Injection12 Jan 2026
- 2025Accelerating Reinforcement Learning Algorithms Convergence using Pre-trained Large Language Models as Tutors With Advice Reusing10 Sep 2025
- 2025BiasFilter: An Inference-Time Debiasing Framework for Large Language Models28 May 2025
- 2025FairSteer: Inference Time Debiasing for LLMs with Dynamic Activation Steering20 Apr 2025
- 2025Mimicking How Humans Interpret Out-of-Context Sentences Through Controlled Toxicity Decoding11 Mar 2025
- 2025Evolutionary Guided Decoding: Iterative Value Refinement for LLMs4 Mar 2025
- 2024Guaranteed Generation from Large Language Models9 Oct 2024
- 2024Decoding Game: On Minimax Optimality of Heuristic Text Generation Strategies4 Oct 2024
- 2024Automatic Metrics in Natural Language Generation: A Survey of Current Evaluation Practices17 Aug 2024
12 newest of 26 All 26 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024Beyond Sparse Rewards: Enhancing Reinforcement Learning with Language Model Critique in Text Generation
- 2025Teaching Language Models to Critique via Reinforcement Learning
- 2024A Critical Look At Tokenwise Reward-Guided Text Generation
- 2026Adaptive Decoding via Test-Time Policy Learning for Self-Improving Generation
- 2024On the Low-Rank Parametrization of Reward Models for Controlled Language Generation
- 2025Text2Grad: Reinforcement Learning from Natural Language Feedback
- 2023The Consensus Game: Language Model Generation via Equilibrium Search
- 2025Towards Cost-Effective Reward Guided Text Generation