当前位置: 首页 > article >正文

<论文>什么是胶囊神经网络?

一、摘要

        本文介绍2017年的论文《Dynamic Routing Between Capsules》,论文介绍了胶囊之间的动态路由机制,至此胶囊神经网络才算是完整地被提出来(此前几年Hinton老爷子有给出过概念,但是没有给出官方的实现)。

5e0a4e4fff7744ce88d8fb6f67b0842c.png

译文:

        胶囊是一组神经元,其活动向量表示特定类型实体(如物体或物体部分)的实例化参数。我们使用活动向量的长度来表示实体存在的概率,并使用其方向来表示实例化参数。一个层级的活跃胶囊通过变换矩阵对更高层级胶囊的实例化参数进行预测。当多个预测一致时,更高层级的胶囊变得活跃。我们展示了一种经过判别性训练的多层胶囊系统在MNIST数据集上达到了最先进的性能,并且在识别高度重叠的数字方面明显优于卷积神经网络。为了实现这些结果,我们使用了一种迭代的通过一致性路由的机制:低层级胶囊倾向于将其输出发送到那些活动向量与来自低层级胶囊的预测有较大标量积的高层级胶囊。

二、核心创新点

a591db89016749758a979802f12959ff.png

        胶囊网络最初被Hinton老爷子提出来是用于解决CNN的缺陷的。我们拿一张网上广为流传的示例图来解释胶囊网络相对于CNN的改进之处。对于下面这张图,CNN则会认为它是一张人脸的图像,而胶囊网络则可以辨别出来这不是正常的人脸。

70338014b84f44a0a5244a2a19248c70.png

        要学习胶囊网络,我们还需要简要了解下基本概念。胶囊(Capsule)是胶囊神经网络中的基本单元,它由一组神经元组成,用于表示特定类型的实体及其属性。每个胶囊不仅输出实体存在的概率,还可以表示实体的姿态参数(如位置、方向、大小等)。对比常规的神经网络,常规神经网络的基本单元是神经元,其输出是一个标量值,而胶囊的输出是一个向量,这使得胶囊能够捕捉更丰富的特征信息。通过引入胶囊和动态路由机制,胶囊网络改进了卷积神经网络在特征表示和特征关系捕捉方面的局限性,从而能够更好地处理物体的变形和旋转。

1、胶囊的输入输出和动态路由机制

        对于胶囊的输入和输出,作者使用胶囊的输出向量长度来表示当前胶囊所代表的实体存在于当前输入中的概率。看起来很绕,通俗来说就是如果一个胶囊对应的特征在输入中有出现,那么它对应的输出向量长度就应该比没出现的胶囊输出向量要长。因此,作者使用了一个非线性的squash函数,这个函数确保短向量的长度被压缩到接近0,二长向量的长度则压缩到略小于1的取值。这个思路通过使用判别式学习来实现:

eq?v_%7Bj%7D%20%3D%20%5Cfrac%7B%7C%7Cs_%7Bj%7D%7C%7C%5E%7B2%7D%7D%7B1+%7C%7Cs_%7Bj%7D%7C%7C%5E%7B2%7D%7D%20%5Cfrac%7Bs_%7Bj%7D%7D%7B%7C%7Cs_%7Bj%7D%7C%7C%7D

        其中,eq?v_%7Bj%7D是胶囊j的输出向量,而eq?s_%7Bj%7D是其总输入。对于第一层胶囊之外的其余胶囊,总输入就是所有来自下一层的胶囊的预测向量eq?%5Chat%7Bu%7D_%7Bj%7Ci%7D加权和,这个预测向量则是使用一个权重矩阵eq?W_%7Bij%7D与下一层胶囊的输出eq?u_%7Bi%7D相乘得到。

eq?s_%7Bj%7D%20%3D%20%5Csum_%7Bi%7Dc_%7Bij%7D%5Chat%7Bu%7D_%7Bj%7Ci%7D

eq?%5Chat%7Bu%7D_%7Bj%7Ci%7D%20%3D%20W_%7Bij%7Du_%7Bi%7D

        其中,eq?c_%7Bij%7D是耦合系数,由动态路由迭代确定。胶囊i与上一层中的所有胶囊之间的耦合系数总和为1。并且,耦合系数由“routing softmax”函数来决定,该函数的初始logits(记为eq?b_%7Bij%7D)则由胶囊i和胶囊j之间的对数先验耦合概率来确定。

eq?c_%7Bij%7D%20%3D%20%5Cfrac%7Bexp%28b_%7Bij%7D%29%7D%7B%5Csum_%7Bk%7Dexp%28b_%7Bik%7D%29%7D

        对于动态路由机制,作者也给出了对应的算法流程。

1b9dbf3571254d5f9cf481141880665d.png

2、损失函数

        论文使用实例化向量的长度来表示胶囊实体存在的概率,当且仅当图像中存在数字类k时(论文案例是图像数字识别任务),作者希望数字类k的top-level胶囊具有较长的实例化向量。为了允许多个数字的情况,作者为每个digit胶囊使用单独的边界损失(margin loss)eq?L_%7Bk%7D

eq?L_%7Bk%7D%20%3D%20T_%7Bk%7Dmax%280%2C%20m%5E%7B+%7D-%7C%7Cv_%7Bk%7D%7C%7C%29%5E%7B2%7D+%20%5Clambda%281-T_%7Bk%7D%29max%280%2C%20%7C%7Cv_%7Bk%7D%7C%7C-m%5E%7B-%7D%29%5E%7B2%7D

        其中,当数字类别k存在时,eq?T_%7Bk%7D等于1,另外eq?m%5E%7B+%7D是0.9,eq?m%5E%7B-%7D则是0.1。对于不存在的数字类别,λ则可以防止初始学习过程缩小所有digit胶囊的活动向量长度,λ取值设定为0.5。

 


http://www.kler.cn/a/473516.html

相关文章:

  • 【vue3封装element-plus的反馈组件el-drawer、el-dialog】
  • 优化提示词改善答疑机器人回答质量
  • selenium合集
  • Jenkins内修改allure报告名称
  • xss-labs关卡记录15-20关
  • 数据库回滚:大祸临头时
  • 使用java springboot 使用 Redis 作为限流工具
  • 使用 SQL 和表格数据进行问答和 RAG(7)—将表格数据(CSV 或 Excel 文件)加载到向量数据库(ChromaDB)中
  • MySql---进阶篇(十一)----游标,条件处理程序,存储函数
  • Bash语言的计算机基础
  • 【优选算法】Binary-Blade:二分查找的算法刃(下)
  • 一款FPGA芯片开发的核心板(EP4CE6核心板)
  • WebRTC 的优缺点详细解析
  • 怎麼在iPhone iOS(Wi-Fi/蜂窩數據)上查找IP地址?
  • vue js实现时钟以及刻度效果
  • HTML5 波动动画(Pulse Animation)详解
  • 微信小程序中使用weui组件库
  • 基于知识蒸馏的跨模态目标检测方法总结
  • 【问题记录】npm create vue@latest报错
  • 后勤管理系统|Java|SSM|VUE| 前后端分离
  • 系统分析师笔记
  • 上门按摩系统架构与功能分析
  • PHP语言的正则表达式
  • 面向强化学习的状态空间建模:RSSM的介绍和PyTorch实现
  • STM32之一种双通路CAN总线消息备份冗余处理方法(十三)
  • 工业级千兆路由器 5G+WIFI6 高速稳定串口采集