Tractable Control for Autoregressive Language Generation
Not on the map. Why?
Abstract (arXiv)
Despite the success of autoregressive large language models in text generation, it remains a major challenge to generate text that satisfies complex constraints: sampling from the conditional distribution Pr(text | α) is intractable for even the simplest lexical constraints α. To overcome this challenge, we propose to use tractable probabilistic models (TPMs) to impose lexical constraints in autoregressive text generation models, which we refer to as GeLaTo (Generating Language with Tractable Constraints). To demonstrate the effectiveness of this framework, we use distilled hidden Markov models, where we can efficiently compute Pr(text | α), to guide autoregressive generation from GPT2. GeLaTo achieves state-of-the-art performance on challenging benchmarks for constrained text generation (e.g., CommonGen), beating various strong baselines by a large margin. Our work not only opens up new avenues for controlling large language models but also motivates the development of more expressive TPMs.
Key facts. Cited by 65 papers in the map (data through 7 Oct 2026). Builds on 31 of its 59 references that are in the map. First posted on arXiv 15 Apr 2023.
Builds on
31 of 59 references.
References · 31
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2014CIDEr: Consensus-based Image Description Evaluationcited by 2,027
- 2015Effective Approaches to Attention-based Neural Machine Translationcited by 1,024
- 2016SPICE: Semantic Propositional Image Caption Evaluationcited by 820
- 2021FUDGE: Controlled Text Generation With Future Discriminatorscited by 374
- 2018Plan-And-Write: Towards Better Automatic Storytellingcited by 232
- 2018Fast Lexically Constrained Decoding with Dynamic Beam Allocation for Neural Machine Translationcited by 181
- 2021NeuroLogic A*esque Decoding: Constrained Text Generation with Lookahead Heuristicscited by 153
- 2020NeuroLogic Decoding: (Un)supervised Neural Text Generation with Predicate Logic Constraintscited by 146
- 2022Semantic Probabilistic Layers for Neuro-Symbolic Learningcited by 103
19 more
- 2020Einsum Networks: Fast and Scalable Learning of Tractable Probabilistic Circuitscited by 102
- 2022On the Paradox of Learning to Reason from Datacited by 93
- 2020Lexically Constrained Neural Machine Translation with Levenshtein Transformercited by 45
- 2018Toward Abstractive Summarization Using Semantic Representationscited by 43
- 2022Controllable Text Generation with Neurally-Decomposed Oraclecited by 41
- 2019Neural Extractive Text Summarization with Syntactic Compressioncited by 36
- 2022Scaling Up Probabilistic Circuits by Latent Variable Distillationcited by 35
- 2019SPFlow: An Easy and Extensible Library for Deep Probabilistic Learning using Sum-Product Networkscited by 26
- 2021Probabilistic Generating Circuitscited by 24
- 2022Zero-shot Sonnet Generation with Discourse-level Planning and Aesthetics Featurescited by 24
- 2020Group Fairness by Probabilistic Modeling with Latent Fair Decisionscited by 22
- 2020Scaling Hidden Markov Language Modelscited by 22
- 2020Joints in Random Forestscited by 21
- 2019On Tractable Computation of Expected Predictionscited by 19
- 2017Boosted Generative Modelscited by 18
- 2022Sparse Probabilistic Circuits via Pruning and Growingcited by 17
- 2019Learning Fair Naive Bayes Classifiers by Discovering and Eliminating Discrimination Patternscited by 14
- 2021InsNet: An Efficient, Flexible, and Performant Insertion-based Text Generation Modelcited by 14
- 2022Predictive Querying for Autoregressive Neural Sequence Modelscited by 5
Built on by
Citing papers per year: 2023: 10; 2024: 21; 2025: 20; 2026: 14.
65 papers.
Sort:
By area:
- 2023LINC: A Neurosymbolic Approach for Logical Reasoning by Combining Language Models with First-Order Logic Proverscited by 161
- 2023Sequential Monte Carlo Steering of Large Language Models using Probabilistic Programscited by 79
- 2025Theoretical Benefit and Limitation of Diffusion Language Modelcited by 46
- 2023Can LLMs Follow Simple Rules?cited by 45
- 2025Syntactic and Semantic Control of Large Language Models via Sequential Monte Carlocited by 45
- 2023Subtractive Mixture Models via Squaring: Representation and Learningcited by 39
- 2024Large Language Models for Medicine: A Surveycited by 38
- 2024DeepEdit: Knowledge Editing as Decoding with Constraintscited by 33
- 2024Adaptable Logical Control for Large Language Modelscited by 30
- 2025Diffusion vs. Autoregressive Language Models: A Text Embedding Perspectivecited by 23
- 2023Image Inpainting via Tractable Steering of Diffusion Modelscited by 22
- 2024What is the Relationship between Tensor Factorizations and Circuits (and How Can We Exploit it)?cited by 22
38 more
- 2023A Pseudo-Semantic Loss for Autoregressive Models with Logical Constraintscited by 21
- 2025Fast Controlled Generation from Language Models with Adaptive Weighted Rejection Samplingcited by 20
- 2024HYSYNTH: Context-Free LLM Approximation for Guiding Program Synthesiscited by 14
- 2024On the Relationship Between Monotone and Squared Probabilistic Circuitscited by 14
- 2025Constrained Sampling for Language Models Should Be Easy: An MCMC Perspectivecited by 14
- 2023Next Steps for Human-Centered Generative AI: A Technical Perspectivecited by 13
- 2024Building Expressive and Tractable Probabilistic Generative Models: A Reviewcited by 12
- 2025Neurosymbolic Diffusion Modelscited by 12
- 2024Controllable Generation via Locally Constrained Resamplingcited by 11
- 2026Breaking the Factorization Barrier in Diffusion Language Modelscited by 11
- 2024A Compositional Atlas for Algebraic Circuitscited by 10
- 2024CoDa: Constrained Generation based Data Augmentation for Low-Resource NLPcited by 9
- 2024Scaling Continuous Latent Variable Models as Probabilistic Integral Circuitscited by 9
- 2024NeSyA: Neurosymbolic Automatacited by 9
- 2025Scaling Probabilistic Circuits via Monarch Matricescited by 9
- 2025Fast and Expressive Multi-Byte Prediction with Probabilistic Circuitscited by 9
- 2025TRACE Back from the Future: A Probabilistic Reasoning Approach to Controllable Language Generationcited by 8
- 2025Decoupling Task-Solving and Output Formatting in LLM Generationcited by 8
- 2026Mitigating Bias in Locally Constrained Decoding via Tractable Proposalscited by 8
- 2024Attribute Controlled Fine-tuning for Large Language Models: A Case Study on Detoxificationcited by 7
- 2024Guaranteed Generation from Large Language Modelscited by 7
- 2024Restructuring Tractable Probabilistic Circuitscited by 7
- 2024Control Large Language Models via Divide and Conquercited by 5
- 2024Jet Expansions of Residual Computationcited by 4
- 2025Tractable Representation Learning with Probabilistic Circuitscited by 4
- 2024Approximately Aligned Decodingcited by 3
- 2025ABS: Enforcing Constraint Satisfaction On Generated Sequences Via Automata-Guided Beam Searchcited by 3
- 2025Tractable Sharpness-Aware Learning of Probabilistic Circuitscited by 3
- 2025Constrained Adaptive Rejection Samplingcited by 3
- 2024DANA: Domain-Aware Neurosymbolic Agents for Consistency and Accuracycited by 2
- 2024Interacting Large Language Model Agents. Interpretable Models and Social Learningcited by 2
- 2024On Faster Marginalization with Squared Circuits via Orthonormalizationcited by 2
- 2025The Limits of Tractable Marginalizationcited by 2
- 2023DiNADO: Norm-Disentangled Neurally-Decomposed Oracles for Controlling Language Modelscited by 1
- 2023BOOST: Harnessing Black-Box Control to Boost Commonsense in LMs' Generationcited by 1
- 2023Towards Consistent Language Models Using Declarative Constraintscited by 1
- 2025Interleaved Gibbs Diffusion: Generating Discrete-Continuous Data with Implicit Constraintscited by 1
- 2026Binary Rewards and Reinforcement Learning: Fundamental Challengescited by 1
50 most cited of 65 All 65 (JSON)
- 2026Provably Tractable NFA-Constrained Language Generation via HMMs30 Sep 2026
- 2026Grab a Coffee: Future-Aware Guidance for Discrete Diffusion with Compiled Objectives28 Sep 2026
- 2026Semantic Prefix Oracles for LLM Decoding: Contracts and Differential Validation28 Sep 2026
- 2026Stay Within Your Bounds: Distance-Guided Decoding for Guaranteed Context-Free Grammar Compliance28 Aug 2026
- 2026Meta-Ctrl: Guaranteed Plan Generation by Decoupling Syntactic and Semantic Constraints23 Aug 2026
- 2026A Compositional Theory of Curvature in Probabilistic Circuits13 Aug 2026
- 2026The Parser Already Knows: Lightweight Bias Correction in Constrained Decoding10 Aug 2026
- 2026Tractable Hierarchical Control of Autoregressive Language Models1 Jun 2026
- 2026Mitigating Bias in Locally Constrained Decoding via Tractable Proposals1 Jun 2026
- 2026The Expressivity Boundary of Probabilistic Circuits: A Comparison with Large Language Models13 May 2026
- 2026Binary Rewards and Reinforcement Learning: Fundamental Challenges4 May 2026
- 2026Precise Robot Command Understanding Using Grammar-Constrained Large Language Models5 Apr 2026
38 more
- 2026Breaking the Factorization Barrier in Diffusion Language Models9 Feb 2026
- 2026REASON: Accelerating Probabilistic Logical Reasoning for Scalable Neuro-Symbolic Intelligence28 Jan 2026
- 2025Learning Tractable Distributions Of Language Model Continuations20 Nov 2025
- 2025Fast and Expressive Multi-Byte Prediction with Probabilistic Circuits14 Nov 2025
- 2025Decoupling Task-Solving and Output Formatting in LLM Generation4 Oct 2025
- 2025Constrained Adaptive Rejection Sampling2 Oct 2025
- 2025Norm-Q: Effective Compression Method for Hidden Markov Models in Neuro-Symbolic Applications29 Sep 2025
- 2025Tractable Sharpness-Aware Learning of Probabilistic Circuits7 Aug 2025
- 2025Tractable Representation Learning with Probabilistic Circuits6 Jul 2025
- 2025Scaling Probabilistic Circuits via Monarch Matrices14 Jun 2025
- 2025ABS: Enforcing Constraint Satisfaction On Generated Sequences Via Automata-Guided Beam Search11 Jun 2025
- 2025Denoising the Future: Top-p Distributions for Moving Through Time9 Jun 2025
- 2025Constrained Sampling for Language Models Should Be Easy: An MCMC Perspective6 Jun 2025
- 2025Diffusion vs. Autoregressive Language Models: A Text Embedding Perspective21 May 2025
- 2025Neurosymbolic Diffusion Models19 May 2025
- 2025TRACE Back from the Future: A Probabilistic Reasoning Approach to Controllable Language Generation25 Apr 2025
- 2025The Limits of Tractable Marginalization17 Apr 2025
- 2025Syntactic and Semantic Control of Large Language Models via Sequential Monte Carlo17 Apr 2025
- 2025Fast Controlled Generation from Language Models with Adaptive Weighted Rejection Sampling7 Apr 2025
- 2025textttSEM-CTRL: Semantically Controlled Decoding3 Mar 2025
- 2025Interleaved Gibbs Diffusion: Generating Discrete-Continuous Data with Implicit Constraints19 Feb 2025
- 2025Theoretical Benefit and Limitation of Diffusion Language Model13 Feb 2025
- 2024On Faster Marginalization with Squared Circuits via Orthonormalization10 Dec 2024
- 2024NeSyA: Neurosymbolic Automata10 Dec 2024
- 2024A Compositional Atlas for Algebraic Circuits7 Dec 2024
- 2024Restructuring Tractable Probabilistic Circuits19 Nov 2024
- 2024Interacting Large Language Model Agents. Interpretable Models and Social Learning2 Nov 2024
- 2024Controllable Generation via Locally Constrained Resampling17 Oct 2024
- 2024Guaranteed Generation from Large Language Models9 Oct 2024
- 2024Jet Expansions of Residual Computation8 Oct 2024
- 2024Attribute Controlled Fine-tuning for Large Language Models: A Case Study on Detoxification7 Oct 2024
- 2024Control Large Language Models via Divide and Conquer6 Oct 2024
- 2024Approximately Aligned Decoding1 Oct 2024
- 2024DANA: Domain-Aware Neurosymbolic Agents for Consistency and Accuracy27 Sep 2024
- 2024What is the Relationship between Tensor Factorizations and Circuits (and How Can We Exploit it)?12 Sep 2024
- 2024On the Relationship Between Monotone and Squared Probabilistic Circuits1 Aug 2024
- 2024Adaptable Logical Control for Large Language Models19 Jun 2024
- 2024Scaling Continuous Latent Variable Models as Probabilistic Integral Circuits10 Jun 2024
50 newest of 65 All 65 (JSON)
- 2023LINC: A Neurosymbolic Approach for Logical Reasoning by Combining Language Models with First-Order Logic Proverscited by 161
- 2024DeepEdit: Knowledge Editing as Decoding with Constraintscited by 33
- 2025Neurosymbolic Diffusion Modelscited by 12
- 2024NeSyA: Neurosymbolic Automatacited by 9
- 2025TRACE Back from the Future: A Probabilistic Reasoning Approach to Controllable Language Generationcited by 8
- 2025Decoupling Task-Solving and Output Formatting in LLM Generationcited by 8
- 2026Binary Rewards and Reinforcement Learning: Fundamental Challengescited by 1
- 2026REASON: Accelerating Probabilistic Logical Reasoning for Scalable Neuro-Symbolic Intelligencecited by 0
- 2026Binary Rewards and Reinforcement Learning: Fundamental Challenges4 May 2026
- 2026REASON: Accelerating Probabilistic Logical Reasoning for Scalable Neuro-Symbolic Intelligence28 Jan 2026
- 2025Decoupling Task-Solving and Output Formatting in LLM Generation4 Oct 2025
- 2025Neurosymbolic Diffusion Models19 May 2025
- 2025TRACE Back from the Future: A Probabilistic Reasoning Approach to Controllable Language Generation25 Apr 2025
- 2024NeSyA: Neurosymbolic Automata10 Dec 2024
- 2024DeepEdit: Knowledge Editing as Decoding with Constraints19 Jan 2024
- 2023LINC: A Neurosymbolic Approach for Logical Reasoning by Combining Language Models with First-Order Logic Provers23 Oct 2023
- 2025Theoretical Benefit and Limitation of Diffusion Language Modelcited by 46
- 2023Image Inpainting via Tractable Steering of Diffusion Modelscited by 22
- 2026Breaking the Factorization Barrier in Diffusion Language Modelscited by 11
- 2026Grab a Coffee: Future-Aware Guidance for Discrete Diffusion with Compiled Objectivescited by 0
- 2026Grab a Coffee: Future-Aware Guidance for Discrete Diffusion with Compiled Objectives28 Sep 2026
- 2026Breaking the Factorization Barrier in Diffusion Language Models9 Feb 2026
- 2025Theoretical Benefit and Limitation of Diffusion Language Model13 Feb 2025
- 2023Image Inpainting via Tractable Steering of Diffusion Models28 Nov 2023
- 2023Sequential Monte Carlo Steering of Large Language Models using Probabilistic Programscited by 79
- 2023Can LLMs Follow Simple Rules?cited by 45
- 2025Syntactic and Semantic Control of Large Language Models via Sequential Monte Carlocited by 45
- 2023Subtractive Mixture Models via Squaring: Representation and Learningcited by 39
- 2024Large Language Models for Medicine: A Surveycited by 38
- 2024Adaptable Logical Control for Large Language Modelscited by 30
- 2025Diffusion vs. Autoregressive Language Models: A Text Embedding Perspectivecited by 23
- 2024What is the Relationship between Tensor Factorizations and Circuits (and How Can We Exploit it)?cited by 22
- 2023A Pseudo-Semantic Loss for Autoregressive Models with Logical Constraintscited by 21
- 2025Fast Controlled Generation from Language Models with Adaptive Weighted Rejection Samplingcited by 20
- 2024HYSYNTH: Context-Free LLM Approximation for Guiding Program Synthesiscited by 14
- 2024On the Relationship Between Monotone and Squared Probabilistic Circuitscited by 14
12 most cited of 51 All 51 (JSON)
- 2026Provably Tractable NFA-Constrained Language Generation via HMMs30 Sep 2026
- 2026Semantic Prefix Oracles for LLM Decoding: Contracts and Differential Validation28 Sep 2026
- 2026Stay Within Your Bounds: Distance-Guided Decoding for Guaranteed Context-Free Grammar Compliance28 Aug 2026
- 2026Meta-Ctrl: Guaranteed Plan Generation by Decoupling Syntactic and Semantic Constraints23 Aug 2026
- 2026A Compositional Theory of Curvature in Probabilistic Circuits13 Aug 2026
- 2026The Parser Already Knows: Lightweight Bias Correction in Constrained Decoding10 Aug 2026
- 2026Tractable Hierarchical Control of Autoregressive Language Models1 Jun 2026
- 2026Mitigating Bias in Locally Constrained Decoding via Tractable Proposals1 Jun 2026
- 2026The Expressivity Boundary of Probabilistic Circuits: A Comparison with Large Language Models13 May 2026
- 2026Precise Robot Command Understanding Using Grammar-Constrained Large Language Models5 Apr 2026
- 2025Learning Tractable Distributions Of Language Model Continuations20 Nov 2025
- 2025Fast and Expressive Multi-Byte Prediction with Probabilistic Circuits14 Nov 2025
12 newest of 51 All 51 (JSON)