cs.CLOct 19, 2026

Shaping the Document, Not the Model: Optimized Suffixes for Context Following

Authors: Gia-Bao Dinh Ho, Sekh Mainul Islam, Masahiro Kaneko, Timothy Baldwin, Isabelle Augenstein

Organizations: Mohamed bin Zayed University of Artificial Intelligence · University of Copenhagen

Abstract

When a user supplies a document and wants the answer drawn from it, a language model is expected to follow it even when it contradicts its parametric knowledge. Prior work has focused on making the model rely on context by steering activations, contrasting logits during decoding, or instructing it in plain language. We ask whether the document alone, with no instruction written into it and no access to the model at inference time, is enough. We optimize a document suffix by discrete token search, starting from the answer to a single question and extending the objective to a set of passages, to multi-hop questions that follow from the planted fact, and to later turns of a conversation. On counterfactual passages that contradict parametric knowledge of the model, the suffix flips nearly all items the model resists across three instruction-tuned model families, in most settings outperforms the decoding baseline and the strongest instruction we could write, and matches activation steering. Controls show that the suffix largely acts through the passage rather than injecting the answer, and a single suffix per model transfers to unseen passages. Each extended objective carries the effect to its target, so the objective, not only the document, sets how far the effect reaches.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Cache2LoRA: Understanding and Improving Context Internalization

    Oct 16, 2026Yi Fang, Gaurav Srivastava, Zhenyu Bi +2LLM Context CompressionKnowledge Augmentation for Language Models

  2. Accept or Repair: Preserving Passage-Supported Answers in Contrastive Decoding

    Oct 21, 2026Jungseob Lee, Seungyoon Lee, Hyeonseok Moon +6Evidence-Grounded QALanguage Model Decoding