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