基于胶囊网络的变分期望最大化路由算法

    公开(公告)号:CN109583563A

    公开(公告)日:2019-04-05

    申请号:CN201811250635.1

    申请日:2018-10-25

    Abstract: 本发明公开了一种基于胶囊网络的变分期望最大化路由算法,把低级胶囊的pose矩阵视为GMM的数据点、高级胶囊的pose矩阵视为高斯分布,由VBEM路由算法将数据点聚类为一个个高斯分布并计算其分布参数,即在运行时将低级胶囊分组形成一个高级别的胶囊,然后根据高斯分布参数更新计算激活值a。本发明在胶囊网络中使用VBEM算法,与EM算法相比,这种方法几乎不需要额外的计算量,并且它解决了最大似然方法中的计算难题,基于变分推断对每个分解因子进行最优化来完成整体的最优化过程,既可以得到近似解,又可以避免当高斯分量“退化”到一个具体的数据点时产生的奇异性,也使得隐变量类别数k可以自动在算法中确定,并且在k较大的时候也避免了过拟合。

Patent Agency Ranking