本仓库记录关于LLM (large language models)和VLM (vision-language models)的文章,特别是关于In-context Learning (ICL)的。看过的文章会至少用一句话概括内容,有些还会有notes。只有标题的就是还没看过的,只是先存档到这里。
有关OOD generalization的paper list请移步:link
- 2024-12 接下来主要关注VLM的hallucination、reasoning问题。同时也会follow ICL的最新进展。
- 2024-05 接下来主要关注探究ICL机制的相关工作
-
Explore Spurious Correlations at the Concept Level in Language Models for Text Classification (Arxiv Jan 2024) [paper] 发现了LLM在文本分类中会依赖的concept-label spurious correlation,提出使用ChatGPT来扩充数据来消除虚假关联。
-
Positional Information Matters for Invariant In-Context Learning: A Case Study of Simple Function Classes (ongoing work) [[paper]](Positional Information Matters for Invariant In-Context Learning: A Case Study of Simple Function Classes) 发现模型对于demonstration的permutation invariance或许是ICL OOD的关键。提出使用相同的positional encoding来提升ICL OOD性能。
-
Simple synthetic data reduces sycophancy in large language models (Arxiv Feb 2024) [paper] LLMs会迎合提问者的观点而罔顾事实。提出合成一些用户的观点和正确性无关的新prompt,然后在这些数据上fine-tune来解决sycophancy问题。
-
Understanding In-Context Learning in Transformers and LLMs by Learning to Learn Discrete Functions (ICLR 2024 Oral) [paper] 探究transformer在一系列离散任务上的能力。特别地,发现经过预训练的模型相比随机初始化的模型获得了更强的最近邻、disjunction和conjunction的能力。
-
Batch-ICL: Effective, Efficient, and Order-Agnostic In-Context Learning (Arxiv Jan 2024) 发现使用batch ICL,将N个example设置为N个one-shot inference,再把每个inference得到的token做平均,替换到query sample做aggregation最终再预测能带来提升。一个奇特的发现是做aggregation时从某一层往后做性能会突增,在那之前性能接近零。对此解释是transformer的低层是在学语义信息。
-
RefuteBench: Evaluating Refuting Instruction-Following for Large Language Models (Arxiv Feb 2024) [paper] 评估模型的改变它们的原始输出并遵循和一开始相违背的指令的能力。主要观察:1)大部分模型都会倾向于遵守它们的预训练知识 2)模型很难根据人类后续的反馈泛化到新的问题 3)所有模型都会逐步忘记人类反馈并落回到它们的内部知识里 4)模型是不是第一时间遵守了人类的反馈,对于后续的行为起到关键作用
-
Function Vectors in Large Language Models (ICLR 2024) [paper] 发现context prompt的最后一个token的隐层表示encode了这个任务的信息,称为function vector(FV)。将其加到zero-shot的prompt上,发现有显著提升。
-
A Data Generation Perspective to the Mechanism of In-Context Learning (Arxiv Feb 2024) [paper] 有关task recognition和task learning的综述
-
Identifying and Analyzing Task-Encoding Tokens in Large Language Models (Arxiv Feb 2024) [paper] 探究了context中的template词("data:","answer:")/stopword(标点、连词等无意义词)/content对performance的意义。结果发现template词对ICL性能提升最有用,content反而没什么用;还探究了template词的什么特征使得它有别于context中的其他成分,结果发现template词本身的语义、其重复性、其分隔x和y的格式作用这三者都对ICL性能有显著的作用。
-
Whispers that Shake Foundations: Analyzing and Mitigating False Premise Hallucinations in Large Language Models (Arxiv Feb 2024) [paper] 发现,问题中的错误前提而导致的回答中的幻觉是由于模型中特定的head的激活所引起的。提出了一种强行消除这些head对于问题中的错误前提对应的token的attention的方法。
-
In-context Vectors: Making In Context Learning More Effective and Controllable Through Latent Space Steering (Arxiv Feb 2024) [paper] 提出用context的第L层表示构造一个表征任务信息的vector(ICV),然后再加到query时的第L层所有token的表示上。
-
The mechanistic basis of data dependence and abrupt learning in an in-context classification task (ICLR 2024 Oral) [paper] 有关transformer 的IWL(in-weights learning)和ICL学习过程的实验性分析。在一个两层toy transformer中揭示了induction head学习机制。
-
Understanding In-context Learning From Repetitions (ICLR 2024) [paper] 揭示了context中重复出现的pattern会导致模型更倾向于输出这个pattern的现象。
-
In-context Learning Learns Label Relationships but is not Conventional Learning (ICLR 2024) [paper] 以更大的模型和更长的context重新审视以往的ICL讨论,并得出了以下三个结论:1)ICL会学x-y映射,正确的label是有用的,且模型越大这一效应越明显 2)ICL能学预训练时没见过的新任务 3)即使context很长,ICL也不能彻底覆盖预训练获得的preference 4)LLM更关注更靠近query的example
-
How do Large Language Models Learn In-Context? Query and Key Matrices of In-Context Heads are Two Towers for Metric Learning (Arxiv Feb 2024) [paper] 在简单的word classification任务上,首先按照类似Function Vector的做法,提取出对输出正确预测贡献最大的head。然后分析这些head并发现了如下机制:label的V encode了label的特征,label的K encode了demonstration的特征;last token的Q encode了query的特征;last token query和正确label的K的attention score比其他head的显著大;last token Q与在context中出现更多的label/更靠近query的label的K的attention score更大。
-
Locating Factual Knowledge in Large Language Models: Exploring the Residual Stream and Analyzing Subvalues in Vocabulary Space (Arxiv Jan 2024) [paper] 提出了一种定位transformer中对输出某一label贡献最大的attention或FFN layer(或其subvalue)的方法。
-
In-Context Learning State Vector with Inner and Momentum Optimization (NeurIPS 2024) [paper] 提了一种新的用vector压缩信息的技术(State Vector SV):是将前L层的每层的attention输出concat起来。然后提了三种技术(aggregate每一个example的SV、用momentum、分组提取SV再聚合)来进一步优化SV,取得了一些性能提升。
-
GNNavi: Navigating the Information Flow in Large Language Models by Graph Neural Network (Arxiv Feb 2024) [paper] 提出将GNN插在LLM的某一层后面,强行使得information flow(token representation就是node representation)是从x->y和y->:连边,然后得到的node representation输给LLM的下一层(每个token的都保留着,因为GNN的输出也是所有node的输出)。最后只在ICL数据集上微调GNN,能够实现和lora媲美的速度和更好的acc。
-
Decomposing Label Space, Format and Discrimination: Rethinking How LLMs Respond and Solve Tasks via In-Context Learning (Arxiv April 2024) [paper] 将ICL能力分成1)正则化输出的label space、2)正则化输出的label format,和3)提升label space/format分布内的判别能力三个方面。结论:ICL的能力主要来自前两者。同时也在实验上间接证明了ICL会倾向于预测出context和test更像的样本的label。
-
The Evolution of Statistical Induction Heads: In-Context Learning Markov Chains (Arxiv Feb 2024) [paper] 在预测Markov序列任务上,揭示了存在一个学习出从简单到复杂function的过程(uniform -> unigram -> bigrams (optimal))。此外,也验证了类似retrieval(n-gram),即找最相似的context token然后取它后面的token作为预测的机制
-
In-Context Language Learning: Architectures and Algorithms (Arxiv Jan 2024) [paper] 构造了一个模拟的language token ICL任务,给了一系列实验证据说明transformer实现了和n-gram类似的retrieval过程
-
Trusting Your Evidence: Hallucinate Less with Context-aware Decoding (Arxiv May 2024) [paper] 为了增强对context的关注能力,提出在推理时加权以context为条件的预测和不含context的预测:$y=\text{softmax}((1+\alpha) p_\theta(y|c,x)-\alpha p_\theta(y|x))$ 。背后的理论基础是朴素贝叶斯 [blog]
-
How In-Context Learning Emerges from Training on Unstructured Data: On the Role of Co-Occurrence, Positional Information, and Noise Structures (Arxiv Jun 2024) [paper] 在非ICL格式的数据上训练,探究了“国家-首都”类任务(预训练常见)和输出首字母任务(不常见),发现pattern在训练数据里的重复性和位置信息分别是这两种任务的关键。
-
Benefits of Transformer: In-Context Learning in Linear Regression Tasks with Unstructured Data (Arxiv Feb 2024) [paper] 分析多层、PE、multi head等模块对于提升ICL在线性回归任务上性能的作用。
-
Do pretrained Transformers Learn In-Context by Gradient Descent? (ICML 2024) [paper] 讨论了一下目前ICL工作的不切实际的setting,从一些实验指标上说明了ICL和GD有显著不同。
-
Rectifying Demonstration Shortcut in In-Context Learning (NAACL 2024) [paper] 发现context单词的字面意思会影响ICL分类的结果(一种shortcut)。提出了一种calibration的策略。
-
Investigating the Pre-Training Dynamics of In-Context Learning: Task Recognition vs. Task Learning (Arxiv June 2024) [paper] 训练过程中task learning和task recognition存在竞争现象
-
Transformers Can Perform Distributionally-robust Optimisation through In-context Learning (ICML 2024 workshop on ICL) [paper] ICL有一定的DRO的能力
-
How Do In-Context Examples Affect Compositional Generalization? (ACL 2024) [paper] 发现context example对于组合泛化能力影响显著。具体来说,context example和query越像、example越多样、每个样本越简单,泛化能力越好。
-
What Do Language Models Learn in Context? The Structured Task Hypothesis (ACL 2024) [paper] 通过实验验证了ICL能够对预训练见过的任务进行复合的假设,否定了ICL仅仅能够进行分布内任务的试别以及ICL能够泛化到某些训练时没见过的任务的假设。
-
What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formation (ICML 2024) [paper] 识别了transformer在解决ICL的copy-and-paste任务中存在的三种circuit
-
In-Context Learning of Energy Functions (ICML 2024 ICL workshop) [paper] 提出了将next-token的条件分布建模为能量函数的形式,发现transformer也能在这种形式下展现出ICL能力
-
From Words to Numbers: Your Large Language Model Is Secretly A Capable Regressor When Given In-Context Examples (Arxiv April 2024) [paper] 发现诸如GPT-4,Claude-3之类的LLM能够在不重新训练的情况下做linear和non-linear regression,甚至有时能超过supervised training的方法(但仅限于很大的LLM)。
-
Disentangling Latent Shifts of In-Context Learning Through Self-Training (Arxiv Oct 2024) [paper] 针对ICL不稳定的问题,提出为student LLM训练一个adapter用来从teacher LLM那里获取context的知识。【insight】认为之前的vector系列工作只考虑attn head,不够全面。
-
Learning Task Representations from In-Context Learning (ICML 2024 ICL workshop) [[paper]](Learning Task Representations from In-Context Learning) 提出learnable task vector(LTV),为所有head增加可学习的权重,然后加权组合每一个head的activation来得到每一层function vector。发现其可以增强ICL的长度泛化能力。
-
Task Diversity Shortens the ICL Plateau (Arxiv Oct 2024) [paper] synthetic setting,在更多的function class上训练可以加快收敛。发现A任务训练到loss正在逃离plateau的checkpoint在B任务上继续训,可以加快B的训练,说明不同任务之间有一些common structure,提供了为什么多任务训练能更快收敛的一个解释。
-
Many-Shot In-Context Learning (ICML 2024 ICL workshop) [paper] ICL的潜力被few-shot限制了
-
Out-of-distribution generalization via composition: a lens through induction heads in Transformers (Arxiv Aug 2024) [papaer] 在OOD的copy任务上,发现了OOD性能源于执行不同功能层的composition(并没有测复杂的组合泛化任务)。还发现了induction head和previous token head的各自内部的表示的相似性。
-
Context-Scaling versus Task-Scaling in In-Context Learning (Arxiv Oct 2024) [paper] 核心发现:kernel smoothing的特征映射是能够进行context scaling的关键
-
Bayesian scaling laws for in-context learning (Arxiv Oct 2024) [paper] 推导了一种基于贝叶斯的scaling law。在模拟数据集上效果比exponetial scaling law好,在真实LLM和数据集上效果还行。
-
Learning to grok: Emergence of in-context learning and skill composition in modular arithmetic tasks (NeurIPS 2024) [paper] 探究在modular加法问题上的ICL的OOD能力,并解释了模型组件是如何实现OOD的能力的
-
Improving In-Context Learning with Small Language Model Ensembles (NeurIPS 2024 Workshop on Adaptive Foundation Models) [paper] 将在下游任务上fine-tune的多个小模型预测的label和confidence与原始label组合到一起,再输给大模型来做ICL,发现可以提升性能
-
Algorithmic Phases of In-context Learning (ICLR 2025 Ratings 10 8 6 6) [paper] 在一个马尔可夫链上,识别了ICL的四种推理模式:unigram/bigram-inference/retrieval,这几种模式之间的切换可以解释目前的一系列ICL现象,如task diversity threshold, transient nature, task retreival/task learning, early ascent等。
-
Can In-context Learning Really Generalize to Out-of-distribution Tasks? (ICLR 2025 Ratings 8665) [paper] 通过一系列实验分析发现了ICL在OOD任务上只能实现从预训练任务中寻找一个最优任务来拟合下游任务。并从理论上论证了ICL的算法选择机制的存在。
- Rethinking the Role of Demonstrations: What Makes In-Context Learning Work? [paper] 做了一系列消融实验来对ICL进行解释。主要结论:即使input和label不是一一对应,只要label的分布合理,那么ICL同样能给出较为正确的答案.
- Symbol tuning improves in-context learning in language models (EMNLP 2023) [paper] 将demonstration的label换为无意义的symbol,然后微调,以此强迫模型学习input-label mapping。
- In-context Learning Generalizes, But Not Always Robustly: The Case of Syntax (Arxiv Nov 2023) [[paper] ](In-context Learning Generalizes, But Not Always Robustly: The Case of Syntax) 本文通过构建一些语法任务来测试模型对于句子结构的理解能力,以及OOD泛化性能。总的说来,LLM还是会用到一些spurious correlation。
- A Closer Look at In-Context Learning under Distribution Shifts (Arxiv May 2023) [paper] 在一定的分布偏移下,transformer比set-based MLP的性能好;在严重的分布偏移下,两种模型的ICL能力都丧失了。
- Few-shot Fine-tuning vs. In-context Learning: A Fair Comparison and Evaluation (Arxiv May 2023) [paper] 在参数量相当的情况下,ICL的OOD不如FT。30B的ICL跟6.7B的FT性能相当。大部分情况下ICL不如FT。
- Instruction-following Evaluation through Verbalizer Manipulation (Arxiv July 2023) [paper] 发现LLM遵循flipped-label instructions的能力很差,说明ICL可能只是直接利用了预训练语料的知识,而不是学习了context。即使是强如GPT-4的模型也不能很好地遵循flipped-label instructions。
- Reasoning or Reciting? Exploring the Capabilities and Limitations of Language Models Through Counterfactual Tasks (Arxiv Aug 2023) [paper] 一些主要发现:①模型在counterfactual的setting中性能会变差,且setting和常见的、符合事实的setting相差越远,性能越差,说明了模型可能的记忆现象。②在算术任务上,ICL能提升counterfactual(不同进制的计算)性能,但和default setting的差距难以抹平。
- What In-Context Learning "Learns" In-Context: Disentangling Task Recognition and Task Learning (Findings of ACL 2023) [paper] 分别用随机label(x-y映射关系被破坏)和非自然语言label(x-y映射关系保留)来检验模型的从预训练知识中识别任务和从context中学习input-label映射关系的能力,发现:这两种能力同时存在;任务识别能力基本不随模型规模变化;in-context学习能力会随模型变大而上升。
- Larger language models do in-context learning differently (Arxiv Mar 2023) [paper] 和disentanglement TR and TL 那篇差不多,发现了:小模型会倾向于用prior,随着模型增大,覆盖prior而从context学习映射关系的能力会越来越强。
- In-Context Learning Creates Task Vectors (Arxiv Oct 2023) [paper] 同样发现context的最后一个token的表示encode了该任务的信息。通过实验发现ICL近似是在实现如下过程:1)从context学出一个映射函数 2)将这个映射函数用到query上来预测。一个重要观察是:说明模型更倾向于使用vector里的信息,而不是原始context
- Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learning (EMNLP 2023) [paper] 浅层网络从text到label聚合信息,深层网络从label到last token聚合信息。
- Pretraining Data Mixtures Enable Narrow Model Selection Capabilities in Transformer Models (Arxiv Nov 2023) [paper] 发现ICL在测试和预训练任务不相同时,性能不好。
- Pretraining task diversity and the emergence of non-Bayesian in-context learning for regression (NeurIPS 2023) [paper] 发现预训练学习的任务越多,ICL在新任务上的泛化越强(不同任务:不同线性回归的W)
- The Transient Nature of Emergent In-Context Learning in Transformers (NeurIPS 2023) [paper] 训练任务:每个序列的token都有一个label。该任务既可以用ICL解决也可以用In-weights Learning (IWL)解决。实验发现随着训练epoch增加,ICL性能先上升再下降,而IWL能力逐渐上升。
- THE EFFECTS OF PRETRAINING TASK DIVERSITY ON IN-CONTEXT LEARNING OF RIDGE REGRESSION (ICLR 2023 workshop) [paper] 随着预训练时见到的线性回归w(都来自同一分布)越来越多,ICL表现逐渐从MMSE(预训练w的加权组合)变为岭回归(test理论最优)。
- Birth of a Transformer: A Memory Viewpoint (NeurIPS 2023) [paper] 构建了一个bigram任务,在简化setting下推导出了两层transformer要解决这个任务所应具备的参数闭式解,以此计算模型参数和最优解的差距来分析训练过程中的ICL能力的变化
- What Can Transformers Learn In-Context? A Case Study of Simple Function Classes (NeurIPS 2022) [paper] 实验发现:1)linear function是能通过transformer学到的(性能能逼近最小二乘估计)2)ICL有一定的OOD泛化能力(train -> test, context -> test)3)ICL也能学到更复杂的函数,比如sparse linear functions、ReLU NNs、decision trees。
- Rethinking the Role of Demonstrations: What Makes In-Context Learning Work? (EMNLP 2022) [paper] 探究ICL work的因素。
- On the Compositional Generalization Gap of In-Context Learning (Arxiv 2022) [paper] 在CFQ等组合泛化任务上测,发现大模型的OOD(query和context不一致)和ID之间的组合泛化能力的gap相比小模型更小。
-
How do Transformers perform In-Context Autoregressive Learning? (Arxiv Feb 2024) [paper] 在限定linear attention、diagonal weight matrix等条件下,对于序列预测任务$s_{T+1}=Ws_T$(文章考虑的$W$是酉矩阵和正交矩阵两种情况),从理论上给出了取到全局最优解时,transformer 参数所应满足的性质。
-
On Mesa-Optimization in Autoregressively Trained Transformers: Emergence and Capability (Arxiv May 2024) [paper] 理论证明了,不同于直接在ICL目标上进行预训练,经过自回归预训练的one-layer linear attention不能在简单如服从高斯分布的序列上实现ICL。
-
How Do Nonlinear Transformers Learn and Generalize in In-Context Learning? (ICML 2024) [paper] 在进行ICL预训练的情况下,给出了非线性attention的ID和OOD的泛化保证
-
Why Larger Language Models Do In-context Learning Differently? (ICML 2024) [paper] 本文对于更大的模型更容易在flipped label任务上失败给了理论解释:大模型更容易受到prompt中noise的影响,而小模型只会关注更重要的feature所以不容易受到noise影响,进而使pretrain feature发挥更大的作用。
-
Dual Operating Modes of In-Context Learning (ICML 2024) [paper] 理论setting:在混合高斯的线性回归上预训练,分析了给定test context时的后验概率,解释了task recognition和task learning:发现context较短时以task recognition(调整后验的混合高斯的各分量的权重)为主。context变长之后以task learning为主。
-
In-Context Learning with Transformers: Softmax Attention Adapts to Function Lipschitzness (Arxiv May 2024) [paper] softmax能adaptively学一个attention window来实现将context
$y_i$ 进行插值作为预测,将分类任务中见到的retrieval机制拓展到了回归任务上。 -
Towards Better Understanding of In-Context Learning Ability from In-Context Uncertainty Quantification (Arxiv May 2024) [paper] 理论,多头SoftMax attention,任务是估计p(y|x)和Var(y|x),给出了分布内泛化error bound。
-
An Information-Theoretic Analysis of In-Context Learning (Arxiv Jan 2024) [paper] 在信息论视角下,将ICL泛化误差拆解为多项。
- What learning algorithm is in-context learning? Investigations with linear models (ICLR 2023) [paper] 还没看,理论理解ICL机制的文章,linear regression任务,但它的理论设定是模型要在ICL任务上预训练,与实际的Auto Regressive预训练有较大gap。它的证明思路也是通过网络参数构造解,和A Theoretical Understanding of Self-Correction through In-context Alignment这篇类似。
- Transformers as Algorithms: Generalization and Stability in In-context Learning (ICML 2023) [paper] 考虑了context为一系列独立pair和前后样本有关联两种模式,在进行ICL预训练的条件下,给了一个non-linear transformer的excess risk的upper bound
- In-Context Convergence of Transformers (Arxiv Oct 2023) [paper] linear regression任务,需要预训练,一层非线性attention,但是做了其他简化使得transforer就是在根据x之间的attention weight来加权组合各个context y作为最终预测。
- Trained Transformers Learn Linear Models In-Context (Arxiv Oct 2023) [paper] linear regression任务,需要预训练,一层线性attention。证明了预训练loss收敛到全局最优解时,当训练和测试context足够长时,能学到测试prompt上的正确解W。
-
What and How Does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalization (Arxiv Oct 2023) [paper] 数据生成模型是隐马尔可夫模型(和An Explanation of In-context Learning as Implicit Bayesian Inference这篇如出一辙),理论证明了ICL能先根据context推断一个“任务概念”
$\theta$ ,然后根据$\theta$ ,query和context来推断y。 - Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selection (NeurIPS 2023) [paper] 证明了存在一个L-层线性transformer在线性回归、lasso、ridge问题上error有上界。同时在理论和实验上发现了会自动选择最优预训练知识的现象。
- The Learnability of In-Context Learning (NeurIPS 2023) [paper] 证明了当预训练分布包含下游任务的分布的mixuture,ICL能逼近下游任务上的贝叶斯最优分类器。
- An Explanation of In-context Learning as Implicit Bayesian Inference (Arxiv 2022) [paper] 早期经典之作,隐马尔可夫模型,证明ICL能实现bayesian-optimal prediction。
- Scaling LLM Test-time Compute Optimally can be More Effective than Scaling Model Parameters [paper] 研究了两种scaling test-time compute的策略:1)基于verifier(process reward model)的;2)基于模型的self-revision的。发现了根据具体任务(不同难度)来选择最优scaling策略能在达到相同性能时相比best-of-N降低四倍计算量
- LET’S VERIFY STEP BY STEP (ICLR 2024) 发现PRM比ORM好
- The Unlocking Spell on Base LLMs: Rethinking Alignment via In-Context Learning (ICLR 2024) [paper] 通过ICL,添加system prompt和风格化的输出,实现只用很少的样本(3个)来提升LLM alignment。
- The Instruction Hierarchy: Training LLMs to Prioritize Privileged Instructions (Arxiv April 2024) [paper] 构造训练数据来教模型学习不同指令的优先级来防御有害指令。具体方法为,对于不同的任务,分别构造与最高指令aligned/misaligned的指令,然后训练模型输出期望的回答。
- (DPO) Direct Preference Optimization: Your Language Model is Secretly a Reward Model (NeurIPS 2023) [paper]
- (RLHF) Deep reinforcement learning from human preferences (NeurIPS 2017) [paper]
- (PPO) Proximal Policy Optimization Algorithms (Arxiv 2017) [paper]
- LLMs Know More Than They Show: On the Intrinsic Representation of LLM Hallucinations (ICLR 2025 Ratings:8666) [paper] 用一个线性probe来根据模型中间层表示判断模型输出的正确与否。然后让LLM对同一个问题生成多个答案,并用该分类器筛选出正确概率最高的答案,发现能相比原本的答案正确率更高。
- Insights into LLM Long-Context Failures: When Transformers Know but Don't Tell (EMNLP 2024 Findings) [paper] 用一个线性probe来根据模型中间层表示来直接预测问题的答案。发现probe acc比直接生成的acc好。
-
Model Editing with Canonical Examples [paper] 提出了一个新任务:让模型学习几个特定的文本例子,以实现某些纠正,同时还不能让模型改变很多。
-
Evaluating Large Language Models at Evaluating Instruction Following [paper] (ICLR 2024)
-
Not all Layers of LLMs are Necessary during Inference (Arxiv April 2024) 训练一个对LLM中间层feature的分类器判断是否应该早停来获取早停层数,来加速LLM推理。还发现中间层预测的top prob和top prob-second top prob在各个任务上都呈现出随着层数加深而增加并逐渐稳定的趋势(但在不同任务上层数不一样)。[paper]
-
Demonstrating Mutual Reinforcement Effect through Information Flow (Arxiv March 2024) [paper] 研究了同时进行word分类和text分类的MRE(Mutual Reinforcement Effect)任务,也观察到了anchor那篇中的三种attention activation随layer的分布趋势。
-
A Theoretical Understanding of Self-Correction through In-context Alignment (Arxiv May 2024) [paper] 理论分析transformer中的各个模块在self-correction中发挥的作用
-
Mechanics of Next Token Prediction with Self-Attention (AISTATS 2024) [paper] 构造了一个graph来描述next token prediction任务,在简化setting下理论分析出last token更倾向于给更经常作为label的token分配更高的attention。
-
The pitfalls of next-token prediction (Arxiv April 2024) [paper] 指出了自回归模型的缺陷:错误滚雪球效应和在一个单一token路径上只能学出一个类似induction head的shortcut模型
-
A Law of Next-Token Prediction in Large Language Models (Arxiv Aug 2024) [paper]
-
SEMIEVOL: Semi-supervised Fine-tuning for LLM Adaptation (Arxiv Oct 2024) [paper] 提出了半监督fine-tuning框架SEMIEVOL。
- Instruction-following Evaluation through Verbalizer Manipulation (Arxiv July 2023) [paper] 发现LLM遵循flipped-label instructions的能力很差,说明ICL可能只是直接利用了预训练语料的知识,而不是学习了context。即使是强如GPT-4的模型也不能很好地遵循flipped-label instructions。
- Reasoning or Reciting? Exploring the Capabilities and Limitations of Language Models Through Counterfactual Tasks (Arxiv Aug 2023) [paper] 一些主要发现:①模型在counterfactual的setting中性能会变差,且setting和常见的、符合事实的setting相差越远,性能越差,说明了模型可能的记忆现象。②在算术任务上,ICL能提升counterfactual(不同进制的计算)性能,但和default setting的差距难以抹平。
- Can the Inference Logic of Large Language Models be Disentangled into Symbolic Concepts? (Arxiv Apr 2023) [paper] 提出了一种empirical的指标来衡量输入句子里的某些词和词组对某一特定输出的决定程度。
- Contrastive Chain-of-Thought Prompting (Arxiv Nov 2023) [paper] 使用对比CoT,即一个正确CoT搭配一个错误CoT能相比常规的CoT带来提升.
- Same Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Models [paper]
LORA: LOW-RANK ADAPTATION OF LARGE LANGUAGE MODELS 将对模型权重矩阵的更新限制为低秩矩阵乘积$BA$的形式,极大减少了pre-trained model迁移到新任务的代价(不用fine-tune所有参数) [paper]
- Are Sixteen Heads Really Better than One? (NeurIPS 2019) [paper] 在某些层上,只用一个head性能也能保持不变。同时提出了使用attention梯度来衡量head的重要性,提出了剪枝策略。
- Is A Picture Worth A Thousand Words? Delving Into Spatial Reasoning for Vision Language Models (NeurIPS 2024) [paper] 在三个合成的空间理解任务上评测LLM和LVM,主要发现:1)该任务的总体表现并不好 2)对于VLM而言,更依赖于语言信息而不是视觉信息做决策,去掉/扰乱视觉信息甚至会有提升 3)VLM中的language encoder比同样的单独LLM性能更好,说明多模态pretrain对于language有用。【insight】现有的将视觉信息转化到language space再进行推理的范式不够好。
- Can Vision Language Models Learn from Visual Demonstrations of Ambiguous Spatial Reasoning? (Arxiv Sep 2024) [paper]
- TOPVIEWRS: Vision-Language Models as Top-View Spatial Reasoners (Arxiv June 2024) [paper] 提了一个新的俯视图理解的数据集,发现VLM的俯视图理解能力仍然很差
- Decomposing Complex Visual Comprehension into Atomic Visual Skills for Vision Language Models [paper] 原子视觉任务benchmark Atomic Visual Skills Benchmark (AVSBench)
- **DOES SPATIAL COGNITION EMERGE IN FRONTIER MODELS? ** (Arxiv Oct 2024) [paper] 提出了空间理解任务 SPACE benchmark。发现目前最强的模型在简单的空间任务上性能很差
- Towards Interpreting Visual Information Processing in Vision-Language Models (ICLR 2025 886)
- Visual Sketchpad: Sketching as a Visual Chain of Thought for Multimodal Language Models (NeurIPS 2024) [paper] 让模型生成代码来调用工具根据现有的视觉输入产生新的视觉图像来作为推理的辅助,可以提升在各种视觉相关任务上的能力。
- Task Navigator: Decomposing Complex Tasks for Multimodal Large Language Models (CVPR 2024) [paper] 工程文章,借助LLM根据历史子问题和模型回答,迭代产生多个子问题,提升MLLM完成复杂视觉理解任务的能力。提出了VersaChallenge benchmark,包括常识推理、物理关系推理、未来预测等。
- SpatialVLM: Endowing Vision-Language Models with Spatial Reasoning Capabilities (CVPR 2024) [paper] 构建数据集,训了一个spatial-VLM用以解决空间任务
- SpatialRGPT: Grounded Spatial Reasoning in Vision Language Models (NeurIPS 2024) [paper] 构建空间位置关系数据集,添加了一个深度图->语言模块,来增强几何推理
- Multimodal Chain-of-Thought Reasoning in Language Models (TMLR 2024) [paper] 两阶段训练,第一阶段接受文本和视觉的融合特征输出一个rationale(推理过程的文本描述),第二阶段将生成的rationale和原始文本结合,再与视觉特征融合重新输入模型产生预测。
- Thinking Before Looking: Improving Multimodal LLM Reasoning via Mitigating Visual Hallucination (Arxiv Nov 2024) [paper] 对于VQA任务,提出thinking-before-looking范式,先利用一个LLM根据文本问题生成一堆更细致的问题,然后将这些问题和图片一起输给MLLM让其生成推理步骤。最终将原始问题、图片、推理步骤一起输给MLLM让其生成答案。
- Link-Context Learning for Multimodal LLMs (CVPR 2024) [paper] 提出一种新的fine-tune MLLM的方法:让context和query具有一定的causal联系,发现能提升模型通过context学习新概念的能力
- Lever LM: Configuring In-Context Sequence to Lever Large Vision Language Models (NeurIPS 2024) [paper] 先构建一个优质的ICL数据集,然后将该数据集中的image-text对视作token,用CLIP抽取特征作为token embedding,训练一个很小的Transformer(lever-LM)来在该数据集上进行next-token prediction(序列是从query到context这样倒着来的)。测试时,最后给定测试样本,拿lever-LM从该预先挑选好的数据集中预测后续的example来构成context。
- Thinking Before Looking: Improving Multimodal LLM Reasoning via Mitigating Visual Hallucination (Arxiv Nov 2024) [paper] 对于VQA任务,提出thinking-before-looking范式,先利用一个LLM根据文本问题生成一堆更细致的问题,然后将这些问题和图片一起输给MLLM让其生成推理步骤。最终将原始问题、图片、推理步骤一起输给MLLM让其生成答案。
- Mitigating Hallucination in Large Vision-Language Models via Modular Attribution and Intervention (ICLR 2025 8866) [paper] 发现幻觉的产生是由于某些特定的attention head,这些head是源自VLM的LM部分。他们会给文本分配更高的attention。提出了在推理时关闭这些幻觉head和在instruction tunning时专门调这些head两种改进方法。
- Reducing Hallucinations in Large Vision-Language Models via Latent Space Steering (ICLR 2025 886) [paper] 动机:发现使用扰动后再平均的vision feature能降低幻觉,认为幻觉来自vision encoder的不够鲁棒。提出使用in-context vector的做法,计算从正常feature到扰动平均后的feature的主成分,加到推理的时候。
- Analyzing and Mitigating Object Hallucination in Large Vision-Language Models (ICLR 2024) [paper] 发现了幻觉产生的几个触发因素:1)训练数据中的某两种对象的spurious共现关系 2)decoding过程的不确定性会将幻觉词采样出来(即使幻觉词的生成概率本不应该是最高) 3)幻觉更容易出现在生成文本中靠后的位置
-
Debiasing Multimodal Large Language Models (Arxiv Mar 2024) [paper] 同样发现了VLM关注text token的问题。提出了两种decoding的策略。其中一种类似Trusting Your Evidence那篇增强对于context的关注的contrastive decoding方法:
$y=\text{softmax}((1+\alpha) p_\theta(y|v,x)-\alpha p_\theta(y|v',x))$ ,其中第一项和第二项分别表示正常的图文输入和仅文本输入时的输出。 -
IBD: Alleviating Hallucinations in Large Vision-Language Models via Image-Biased Decoding (Arxiv Feb 2024) [paper] 也提出了contrastive decoding的方法,用一个更加关注视觉token的模型
$\hat{\theta}$ 的logit减去原始模型$\theta$ 的logit,该项称为CD score。构建“更加关注视觉token的模型”的方法:增大对视觉token的attention score。同时使用两个自适应权重来调节该contrastive decoding的程度:1)$\hat{\theta}$ 和$\theta$ 的预测越像,CD score权重越小;2) 由于发现生成content token(有实际意义的)相比function token(无实际意义的连词等)的CD score更大,也就是说更加关注image只对content token的正确生成更有利,所以对content token添加更大的权重,而对function token添加较小的权重。 - Paying More Attention to Image: A Training-Free Method for Alleviating Hallucination in LVLMs (ECCV 2024) [paper] 发现当去掉图像,且让模型在其在有图像的情况下所生成的文本的基础上继续生成,仍然会出现相同的幻觉。这种现象被称为text inertia(文本惯性)幻觉。提出的方法也是contrastive decoding:用正常的prediction减去纯文本的prediction
- Mitigating object hallucinations in large vision-language models through visual contrastive decoding (CVPR 2024) Visual Contrastive Decoding (VCD)
- Mitigating hallucinations in large vision-language models with instruction contrastive decoding (ACL Findings 2024) Instruction Contrastive Decoding (ICD)
- OPERA: Alleviating Hallucination in Multi-Modal Large Language Models via Over-Trust Penalty and Retrospection-Allocation (CVPR 2024) [paper] 发现生成回答中的summary token(指attn都集中在其上的token,且往往是无意义token,无法蕴含丰富的视觉信息)越多,幻觉越严重。提出了识别生成token中的summary token并据此减轻幻觉的策略
- Self-Introspective Decoding: Alleviating Hallucinations for Large Vision-Language Models (ICLR 2025 Ratings: 8665) [paper] 首先指出了过往的contrastive decoding方法的问题:有可能所减去的幻觉输出“不够幻觉”,导致正常输出减去它之后反而不准确了。本文认为低attention score的vision token更容易导致幻觉,因此为了更好地引发幻觉输出再减去它,提出在推理时仅保留低attention score的token。
-
Intervening Anchor Token: Decoding Strategy in Alleviating Hallucinations for MLLMs (ICLR 2025 Ratings: 8866) [paper] 先定义了一种分析工具:token propagation probability
$\rho$ ,来描述一个token在前传时的贡献。发现幻觉和$\rho$ 的低熵有关(attention都集中在summary token上了,从而丢失了视觉token的信息)。理论证明了将QK矩阵的二范数控制在一个合理范围内可以增大$\rho$ 的熵,提了一个启发式策略来实现这一目标。 - Visual Description Grounding Reduces Hallucinations and Boosts Reasoning in LVLMs (ICLR 2025 Ratings: 8666) [paper] 现有的解决幻觉的方法难以提升在视觉推理benchmark上的能力。VLM能识别视觉元素,但难以利用它们进行推理。
- Towards Interpreting Visual Information Processing in Vision-language Models (ICLR 2025 Ratings: 8866) 发现object token(图像中对应于物体的token)去掉之后模型掉点最严重。且发现阻塞object token到last token的attention之后掉点最严重。说明在识别物体时,信息直接从object token传递到last token。
- Explainable and Interpretable Multimodal Large Language Models: A Comprehensive Survey (Arxiv Dec 2024) [paper] Survey
- Emu3: Next-Token Prediction is All You Need (Arxiv September 2024) [paper] 将文本、图片、视频都转化为token,进行next-token prediction的预训练。能同时做图片视频的生成、视觉-语言理解。训练模型:包含文本encoder(T5)、视觉encoder(ViT-large)和文本decoder(T5,输入为视觉-文本融合特征,输出为文本)。训练资源:8*32G V100。
- Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generation (Arxiv Oct 2024) [paper] 用一个自回归transformer统一实现多模态的理解和生成任务
- Link-Context Learning for Multimodal LLMs (CVPR 2024) [paper] 提出一种新的fine-tune MLLM的方法:让context和query具有一定的causal联系,发现能提升模型通过context学习新概念的能力
- Can Vision Language Models Learn from Visual Demonstrations of Ambiguous Spatial Reasoning? (Arxiv Sep 2024)
- Finding Visual Task Vectors (ECCV 2024) [paper]
- Lever LM: Configuring In-Context Sequence to Lever Large Vision Language Models (NeurIPS 2024) [paper] 先构建一个优质的ICL数据集,然后将该数据集中的image-text对视作token,用CLIP抽取特征作为token embedding,训练一个很小的Transformer(lever-LM)来在该数据集上进行next-token prediction(序列是从query到context这样倒着来的)。测试时,最后给定测试样本,拿lever-LM从该预先挑选好的数据集中预测后续的example来构成context。
- Towards Global Optimal Visual In-Context Learning Prompt Selection (NeurIPS 2024) [paper] 没细看,也是做ICL example排序的。base idea都是与测试样本越相似的example效果越好。训练一个用于排序的transformer进行局部排序,再根据局部排序训练一个全局排序信息的向量。
- What Factors Affect Multi-Modal In-Context Learning? An In-Depth Exploration (NeurIPS 2024) [paper] 从demo选择、demo顺序和context的构建三个角度探究了影响多模态ICL的因素
- What Makes Multimodal In-Context Learning Work? (CVPR 2024 Workshop on Prompting in Vision) [paper] 对Multimodal ICL的实验性分析,主要发现:文本和图像同时输入时,MLLM更依赖文本;目前的MICL基本上是在做从context co'p'y
- What Makes Good Examples for Visual In-Context Learning? [paper] 纯vision ICL。找和query最相近的样本来做ICL,类似Link-context learning。
- Conditional Prompt Learning for Vision-Language Models (CoCoOp, CVPR2022) 将图片特征直接加到context token上,获得sample-wise的prompt,以实现instance的generalization。其实就是希望通过引入图像信息来使得prompt描述得更贴切。不过感觉还是有点怪,因为所有class都加上了同样的可学习prefix,为什么能提高预测为正确类的概率?
- MaPLe: Multi-modal Prompt Learning, CVPR2023
- Prompt-aligned Gradient for Prompt Tuning, ICCV2023
- Compound Text-Guided Prompt Tuning via Image-Adaptive Cues, AAAI2024
- MmAP : Multi-modal Alignment Prompt for Cross-domain Multi-task Learning, AAAI2024
- Improving Zero-Shot Generalization for CLIP with Synthesized Prompts (ICCV 2023)
- Domain Adaptation via Prompt Learning, arxiv 2022
- AD-CLIP: Adapting Domains in Prompt Space Using CLIP, ICCV2023
- Multi-Prompt Alignment for Multi-Source Unsupervised Domain Adaptation, NIPS2023
- Prompt-based Distribution Alignment for Unsupervised Domain Adaptation, AAAI2024
- StyLIP: Multi-Scale Style-Conditioned Prompt Learning for CLIP-based Domain Generalization, arxiv2023
- VisionLLaMA: A Unified LLaMA Interface for Vision Tasks (Arxiv Mar 2024) [paper] Vision LLaMa
- Are We on the Right Way for Evaluating Large Vision-Language Models? (Arxiv April 2024) [paper] 现有的vision-language数据集质量不够好,很多问题都是只看语言部分就能解决,或者问题在类似的训练语料中见过,根本不需要图片;构建了一个高质量的vision-language数据集。
- Visual Instruction Tuning (NeurIPS 2023) [paper] LLaVA