mlsanityData to 14 Oct 2026

Where to Place the Query? Unveiling and Mitigating Positional Bias in In-Context Learning for Diffusion LLMs via Decoding Dynamics

arXiv 2606.19349 ↗First posted 26 Apr 2026cs.CL cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

While In-Context Learning (ICL) is extensively studied in Autoregressive (AR) LLMs, its mechanism within Diffusion Large Language Models (dLLMs) remains largely unexplored. Unlike AR models restricted by unidirectional causal masking, dLLMs intrinsically utilize bidirectional attention, offering extensive spatial flexibility for query placement. Unfortunately, current practices conventionally inherit AR-style trailing-query templates, often overlooking the structural paradigm shift. This paper presents a comprehensive analysis unveiling that query position is actually a first-order variable in dLLMs. Through empirical decoupling, we demonstrate that positional variance impacts generation quality on par with example semantic quality. Internally, this positional sensitivity stems from a spatial ``Recency Effect'' in attention flow and task-dependent shifts in decoding trajectories. To mitigate this instability without ground-truth labels, we reveal that traditional single-step confidence (Cdecoded) fails in dLLMs. Instead, we propose Average Confidence (overlineC), a novel metric tracking the iterative decoding process. By establishing the foundational spatial ICL baselines, we introduce Auto-ICL, a training-free adaptive routing strategy that dynamically optimizes query placement, robustly approaching oracle performance across heterogeneous reasoning and perception tasks.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 32 of its 36 references that are in the map. First posted on arXiv 26 Apr 2026.

Builds on

32 of 36 references.

References · 32

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  3. 2021Program Synthesis with Large Language Modelscited by 3,637
  4. 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
  5. 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
  6. 2021What Makes Good In-Context Examples for GPT-3?cited by 1,195
  7. 2021Fantastically Ordered Prompts and Where to Find Them: Overcoming Few-Shot Prompt Order Sensitivitycited by 1,135
  8. 2025Large Language Diffusion Modelscited by 912
  9. 2020Quantifying Attention Flow in Transformerscited by 775
  10. 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 629
  11. 2025Dream 7B: Diffusion Large Language Modelscited by 544
  12. 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
20 more
  1. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
  2. 2022Self-Adaptive In-Context Learning: An Information Compression Perspective for In-Context Example Selection and Orderingcited by 161
  3. 2022Ground-Truth Labels Matter: A Deeper Look into Input-Label Demonstrationscited by 116
  4. 2025Diffusion LLMs Can Do Faster-Than-AR Inference via Discrete Diffusion Forcingcited by 116
  5. 2018Sharp Nearby, Fuzzy Far Away: How Neural Language Models Use Contextcited by 113
  6. 2025Diffusion Language Models Know the Answer Before Decodingcited by 70
  7. 2023Large Language Models Can be Lazy Learners: Analyze Shortcuts in In-Context Learningcited by 69
  8. 2023In-context Example Selection with Influencescited by 61
  9. 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
  10. 2022Data Curation Alone Can Stabilize In-context Learningcited by 59
  11. 2024Let's Learn Step by Step: Enhancing In-Context Learning Ability with Curriculum Learningcited by 47
  12. 2024From Redundancy to Relevance: Information Flow in LVLMs Across Reasoning Taskscited by 38
  13. 2025Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Modelscited by 33
  14. 2025LoPA: Scaling dLLM Inference via Lookahead Parallel Decodingcited by 17
  15. 2025Where to show Demos in Your Prompt: A Positional Bias of In-Context Learningcited by 13
  16. 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusingcited by 7
  17. 2025OptiSeq: Ordering Examples On-The-Fly for In-Context Learningcited by 6
  18. 2025Order Matters: Rethinking Prompt Construction in In-Context Learningcited by 3
  19. 2025Unveiling Effective In-Context Configurations for Image Captioning: An External & Internal Analysiscited by 2
  20. 2026DIP: Dynamic In-Context Planner For Diffusion Language Modelscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.