以检索为中心的深度学习:把权重矩阵换成向量数据库

kaixhin · x · 2026-11-04

Google Research PI 小组的 Johannes von Oswald 在 Sakana AI 的 FIG 研讨会上介绍了新论文《Retrieval-Centric Deep Learning in Growing Nonparametric Neural Networks》(arXiv:2610.03858)。核心思路源自 Irie 等人的对偶视角:GD 训练的线性层等价于对训练数据做线性注意力,本文更进一步——训练时不把数据压缩进固定大小的权重矩阵,而是为每个数据点存一对 key-value 表示(即向量数据库),推理时用注意力检索与重组,得到不断生长的非参数网络。论文指出把线性注意力直接换成更强核(RBF、softmax 类)并不能带来有原则的优化,因此提出了基于函数梯度的学习规则,建立了「检索为中心的深度学习」(RCDL) 范式,并在图像分类上展示了有前景的性能与学习效率。演讲还讨论了与 mech interp、优化器设计、数据高效学习的关联。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →