← 最新论文
🤖 machine learning

Radial Basis Function Networks as Projection Heads in Self-Supervised Learning

本文提出通过使用径向基函数网络(RBFN)取代自监督学习中的标准 MLP 投影头,以通过一种新颖的尺度归一化分离(SNS)指标实现无标签的表示质量评估,同时在多种架构和数据集上保持具有竞争力的性能。

原作者: Andreas Schliebitz, Heiko Tapken, Martin Atzmueller

发布于 2026-06-23
📖 1 分钟阅读☕ 轻松阅读

原作者: Andreas Schliebitz, Heiko Tapken, Martin Atzmueller

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

大局观:那个“没用”的头部

想象一下,你正在训练一个机器人来识别不同类型的动物。为了做到这一点,你给机器人提供了一个“大脑”(称为骨干网络/backbone),它通过观察图片来理解重要的特征。

在目前的自监督学习(即机器人无需老师告知答案即可自主学习)中,研究人员会在大脑上方添加一个小的“辅助帽”。这个帽子被称为投影头(projection head)。它的任务是在训练阶段帮助大脑更好地学习。

问题所在: 一旦机器人完成学习,科学家通常会把这个“辅助帽”扔掉。然后,他们必须花费大量的时间和金钱去教大脑一种新的、简单的方法来对动物进行分类,仅仅是为了看看这个大脑到底表现得有多好。本文的作者说:“等等!把帽子扔掉太浪费了。为什么不保留它,并利用它来检查大脑是否在做好工作呢?”

解决方案:用“聪明帽”取代“笨帽”

通常,这种辅助帽是一个标准的、略显复杂的机器(MLP),它很难被理解。你无法轻易判断它为什么做出某个决定。

作者提议将这种标准的帽子更换为径向基函数网络(RBFN)

  • 类比: 把标准的帽子想象成一个只给你一个数字的黑匣子。而新的 RBFN 帽子就像一张带有特定地标的地图。
  • 它是如何工作的: RBFN 不仅仅是处理数字,它还学习特定的“中心点”(地标)和“形状”(每个地标的影响范围有多宽)。
  • 优势: 因为这些地标和形状非常清晰且具有可解释性,你可以通过观察帽子本身来得出结论:“啊,地标分布得很均匀,所以大脑在组织数据方面做得非常好。”你不需要扔掉帽子,也不需要雇佣新的老师来检查工作。

新工具:“SNS”(质量评分)

为了证明他们的想法,作者发明了一种名为**尺度归一化分离度(Scale-Normalized Separation, SNS)**的新质量衡量方法。

  • 隐喻: 想象你正在组织一场派对,宾客(数据点)需要坐在桌子(簇/clusters)旁。
    • 如果所有人都挤在一个角落里,派对就会一团糟(质量差)。
    • 如果每个人都分布得非常完美,且桌子之间留有恰到好处的空间,那么派对就很棒(质量好)。
  • SNS 如何工作: 它观察 RBFN 帽子学习到的“地标”(中心点)。它会检查这些地标之间的距离是否与“桌子的大小”(形状参数)相匹配。
  • 结果: 如果地标分布得非常完美,SNS 得分就会很高。作者发现,高 SNS 得分几乎总是意味着机器人在识别事物方面确实非常出色,即使在没有任何标注答案的情况下也是如此。

他们测试了什么

研究人员在五种不同的流行机器人学习系统(如 MoCo、SimCLR 和 BYOL)以及四组不同的图像数据集(包括他们基于 Google Open Images 创建的新数据集)上测试了这个想法。

他们提出了两个主要问题:

  1. 这个新的“聪明帽”是否和旧的“笨帽”一样好用?
    • 答案: 是的。机器人使用 RBFN 帽子学习得一样好。在某些情况下,它甚至表现得稍好一些。它是一个完美的“即插即用”替代品。
  2. 我们能否信任 SNS 得分来告诉我们机器人是否聪明?
    • 答案: 可以。他们将 SNS 得分与传统的、昂贵的方法(在带标签的数据上训练一个新的分类器)进行了对比。他们发现两者之间存在极强的关联:当 SNS 得分较高时,机器人的实际表现也同样很高。

建议

根据实验结果,作者提出了以下建议:

  • 使用 3 层结构: 构建 RBFN 帽子时,使用三层“地标”而不是通常的两层。这似乎效果最好。
  • 使用高斯形状: 使用特定的数学形状(高斯函数)作为地标,因为它比其他形状效果更好。
  • 无需担心归一化: 你不需要对帽子进行额外的数学“归一化”处理;它在原样状态下就能很好地工作。

总结

本文认为,我们在训练结束后扔掉“辅助帽”是一种资源的浪费。通过将标准帽子更换为径向基函数网络(RBF),我们得到了一个既能帮助机器人学习,又能作为内置质量报告单的工具。我们可以通过观察帽子的内部地图(中心点和形状),瞬间了解机器人的学习情况,从而节省时间和金钱。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →