DETAIL: Task DEmonsTration Attribution for Interpretable In-context Learning
Not on the map. Why?
Abstract (arXiv)
In-context learning (ICL) allows transformer-based language models that are pre-trained on general text to quickly learn a specific task with a few "task demonstrations" without updating their parameters, significantly boosting their flexibility and generality. ICL possesses many distinct characteristics from conventional machine learning, thereby requiring new approaches to interpret this learning paradigm. Taking the viewpoint of recent works showing that transformers learn in context by formulating an internal optimizer, we propose an influence function-based attribution technique, DETAIL, that addresses the specific characteristics of ICL. We empirically verify the effectiveness of our approach for demonstration attribution while being computationally efficient. Leveraging the results, we then show how DETAIL can help improve model performance in real-world scenarios through demonstration reordering and curation. Finally, we experimentally prove the wide applicability of DETAIL by showing our attribution scores obtained on white-box models are transferable to black-box models in improving model performance.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 43 of its 71 references that are in the map. First posted on arXiv 22 May 2024.
Builds on
43 of 71 references.
References · 43
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,293
31 more
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2021Fantastically Ordered Prompts and Where to Find Them: Overcoming Few-Shot Prompt Order Sensitivitycited by 1,122
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
- 2022Transformers learn in-context by gradient descentcited by 714
- 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 710
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
- 2021Datasets: A Community Library for Natural Language Processingcited by 526
- 2020Pruning neural networks without any data by iteratively conserving synaptic flowcited by 475
- 2023Trained Transformers Learn Linear Models In-Contextcited by 331
- 2023Evaluating Verifiability in Generative Search Enginescited by 297
- 2018SentEval: An Evaluation Toolkit for Universal Sentence Representationscited by 294
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 291
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2020Influence Functions in Deep Learning Are Fragilecited by 247
- 2023Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learningcited by 235
- 2023Transformers as Algorithms: Generalization and Stability in In-context Learningcited by 220
- 2023DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Modelscited by 121
- 2021Sampling Permutations for Shapley Value Estimationcited by 101
- 2023Inseq: An Interpretability Toolkit for Sequence Generation Modelscited by 78
- 2023In-Context Learning through the Bayesian Prismcited by 78
- 2024Training Dynamics of Multi-Head Softmax Attention for In-Context Learning: Emergence, Convergence, and Optimalitycited by 71
- 2023Automatic Evaluation of Attribution by Large Language Modelscited by 70
- 2023Understanding In-Context Learning in Transformers and LLMs by Learning to Learn Discrete Functionscited by 62
- 2022Data Curation Alone Can Stabilize In-context Learningcited by 58
- 2023Use Your INSTINCT: INSTruction optimization for LLMs usIng Neural bandits Coupled with Transformerscited by 29
- 2023IDEAL: Influence-Driven Selective Annotations Empower In-Context Learners in Large Language Modelscited by 26
- 2024Localized Zeroth-Order Prompt Optimizationcited by 23
- 2023Prompt Valuation Based on Shapley Valuescited by 10
- 2022Probably Approximate Shapley Fairness with Applications in Machine Learningcited by 6
Built on by
Citing papers per year: 2025: 4; 2026: 1.
5 papers.
- 2025From Compression to Expression: A Layerwise Analysis of In-Context Learningcited by 5
- 2026Interpretability Can Be Actionablecited by 5
- 2025In-Context Probing for Membership Inference in Fine-Tuned Language Modelscited by 3
- 2025Broaden your SCOPE! Efficient Multi-turn Conversation Planning for LLMs with Semantic Spacecited by 2
- 2025Uncovering Scaling Laws for Large Language Models via Inverse Problemscited by 0
- 2026Interpretability Can Be Actionable11 May 2026
- 2025In-Context Probing for Membership Inference in Fine-Tuned Language Models18 Dec 2025
- 2025Uncovering Scaling Laws for Large Language Models via Inverse Problems9 Sep 2025
- 2025From Compression to Expression: A Layerwise Analysis of In-Context Learning22 May 2025
- 2025Broaden your SCOPE! Efficient Multi-turn Conversation Planning for LLMs with Semantic Space14 Mar 2025