mlsanityData to 14 Oct 2026

Limits of n-gram Style Control for LLMs via Logit-Space Injection

arXiv 2601.16224 ↗First posted 12 Jan 2026cs.CLcited by 1

Not on the map. Why?

Abstract (arXiv)

Large language models (LLMs) are typically personalized via prompt engineering or parameter-efficient fine-tuning such as LoRA. However, writing style can be difficult to distill into a single prompt, and LoRA fine-tuning requires computationally intensive training and infrastructure. We investigate a possible lightweight alternative: steering a frozen LLM with n-gram style priors injected in logit space at decoding time. We train an n-gram model on stylistically distinct corpora -- including Don Quixote, CNN/DailyMail news headlines, and arXiv abstracts -- constructing an interpolated 1-to-3-gram prior over next-token probabilities. During generation we modify the LLM's logits by adding a weighted sum of style log-probabilities from each n-gram order that matches the current context, scaled by a control parameter lambda in [0, 1]. We sweep lambda and style corpora and report style perplexity under the n-gram model, base-model perplexity as a proxy for fluency, Jensen-Shannon (JS) divergence between the original and steered token distributions, and token-overlap statistics. On TinyLlama-1.1B we identify a single narrow regime (for the Don Quixote corpus at lambda=0.1) where style perplexity improves by 24.7% and base-model perplexity improves by 51.4% relative to the frozen model. Outside this regime, and for multi-author corpora such as CNN/DailyMail and arXiv abstracts, even small nonzero lambda values generally result in worse style and fluency, and larger lambda values lead to collapse with extreme perplexities and incoherent text. Logit-space injection of n-gram style priors provides lightweight, tunable style control, but it is fragile: it operates effectively only within a narrow range of low lambda values and is consistently outperformed by prompting and LoRA.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 7 of its 8 references that are in the map. First posted on arXiv 12 Jan 2026.

Builds on

7 of 8 references.

References · 7

  1. 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
  2. 2024Infini-gram: Scaling Unbounded n-gram Language Models to a Trillion Tokenscited by 153
  3. 2024Decoding-time Realignment of Language Modelscited by 67
  4. 2024PAD: Personalized Alignment of LLMs at Decoding-Timecited by 60
  5. 2022Critic-Guided Decoding for Controlled Text Generationcited by 45
  6. 2022N-gram Is Back: Residual Learning of Neural Text Generation with n-gram Language Modelcited by 3
  7. 2025Transferring Extreme Subword Style Using Ngram Model-Based Logit Scalingcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.