arXiv:2510.11471cs.LGcs.AI2025-10被引 2

统一了上下文学习与自适应优化器,提出可扩展的迭代推理框架。

Iterative Amortized Inference: Unifying In-Context Learning and Learned Optimizers

  • 按任务适应方式分三类:参数化、隐式、显式,揭示不同方法本质差异
  • 提出迭代推理机制,通过小批量逐步优化,支持大规模数据处理
  • 适合研究通用任务适应的学者,尤其关注高效模型泛化与可扩展性

现代学习系统越来越多依赖于摊销学习——即复用计算或跨任务共享的归纳偏置,以实现对新问题的快速泛化。这一思想涵盖元学习、上下文学习、提示调优、自学习优化器等多种方法。尽管目标相似,这些方法在如何编码和利用任务特定信息(如上下文示例)上存在差异。本文提出一个统一框架,指出这些方法主要区别在于摊销的学习内容(如初始化、学习更新、预测映射)以及推理时如何融入任务数据。我们构建了一个分类体系,将摊销模型分为参数化、隐式和显式三类,依据任务适应是否外化、内化或联合建模。基于此,我们识别出当前方法的关键局限:多数难以扩展到大规模数据集,因推理时处理任务数据的能力(如上下文长度)受限。为此,我们提出迭代摊销推理,一类通过小批量逐步优化解决方案的模型,灵感来自随机优化。该框架连接基于优化的元学习与大模型中的前向传播摊销,为通用任务适应提供可扩展、可拓展的基础。

原文摘要 · Abstract (English)

Modern learning systems increasingly rely on amortized learning - the idea of reusing computation or inductive biases shared across tasks to enable rapid generalization to novel problems. This principle spans a range of approaches, including meta-learning, in-context learning, prompt tuning, learned optimizers and more. While motivated by similar goals, these approaches differ in how they encode and leverage task-specific information, often provided as in-context examples. In this work, we propose a unified framework which describes how such methods differ primarily in the aspects of learning they amortize - such as initializations, learned updates, or predictive mappings - and how they incorporate task data at inference. We introduce a taxonomy that categorizes amortized models into parametric, implicit, and explicit regimes, based on whether task adaptation is externalized, internalized, or jointly modeled. Building on this view, we identify a key limitation in current approaches: most methods struggle to scale to large datasets because their capacity to process task data at inference (e.g., context length) is often limited. To address this, we propose iterative amortized inference, a class of models that refine solutions step-by-step over mini-batches, drawing inspiration from stochastic optimization. Our formulation bridges optimization-based meta-learning with forward-pass amortization in models like LLMs, offering a scalable and extensible foundation for general-purpose task adaptation.

元学习上下文学习优化器可扩展性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。