摘要 - 机器学习系统在实时,关键的决策领域(例如自动驾驶和工业自动化)中获得了突出性。他们的实现应通过不确定性估计来避免过度自信的预测。贝叶斯神经网络(Baynns)是估计预测不确定性的原则方法。但是,它们的计算成本和功耗阻碍了它们在Edge AI中的广泛部署。利用辍学作为后验分布的近似值,将贝恩的参数进行二进制,以及在基于Spintronics基于旋转的计算中(CIM)硬件阵列中实现它们的进一步,可以提供可行的解决方案。但是,设计用于卷积神经网络(CNN)拓扑的硬件辍学模块是具有挑战性且昂贵的,因为它们可能需要大量的辍学模块,并且需要使用空间信息来删除某些元素。在本文中,我们引入了MC-SpatialDropout,这是一种基于空间辍学的近似贝恩,带有Spintronics的新兴设备。我们的方法利用Spintronic设备的固有随机性与现有实现相比有效地实现了空间辍学模块。此外,每个网络层的辍学模块的数量减少了9倍,能量消耗量为94。11×,同时与相关工作相比仍能实现可比的预测性能和不确定性估计。
主要关键词