中文

大型语言模型中少样本学习与微调的表示景观

计算与语言 2024-09-10 v2 机器学习

摘要

上下文学习(ICL)和监督微调(SFT)是提升现代大型语言模型(LLM)在特定任务上性能的两种常见策略。尽管它们的本质不同,但这些策略通常能带来可比的性能提升。然而,关于它们是否在 LLM 内部诱导相似的表示,人们知之甚少。我们通过分析这两种情况下其隐藏表示的概率景观来探讨这一问题。更具体地说,我们比较了 LLM 如何解决相同的问答任务,发现 ICL 和 SFT 创造了非常不同的内部结构,且两者都在网络中间经历了急剧的转变。在网络的前半部分,ICL 根据语义内容分层组织,塑造了可解释的表示。相比之下,SFT 获得的概率景观更模糊且语义混合。在模型的后半部分,微调表示发展出更好地编码答案身份的概率模式,而 ICL 表示的景观特征是较不明确的峰值。我们的方法揭示了 LLM 内部在不同条件下解决相同任务所发展出的多样化计算策略,使我们朝着设计从语言模型中提取信息的最佳方法迈进了一步。

关键词

引用

@article{arxiv.2409.03662,
  title  = {The representation landscape of few-shot learning and fine-tuning in large language models},
  author = {Diego Doimo and Alessandro Serra and Alessio Ansuini and Alberto Cazzaniga},
  journal= {arXiv preprint arXiv:2409.03662},
  year   = {2024}
}