中文

用于二进制神经网络的设计抖动激活

计算机视觉与模式识别 2024-05-10 v2 机器学习

摘要

二进制神经网络通过对网络权重或激活值进行二值化,成为计算机视觉任务中一种具有成本效益和能源效率的解决方案。然而,常见的二进制激活函数,如 Sign 激活函数,仅以单个阈值进行突变性二值化,丢失了特征输出中细粒度信息。本文提出一种激活函数,采用多阈值遵循抖动原则的做法,依据每个像素的空间周期性阈值核,将 Sign 激活函数进行偏移。与文献中方法不同,此处的偏移是为一组相邻像素联合定义的,充分利用了空间相关性。在分类任务上进行实验,表明所设计的抖动 Sign 激活函数作为二进制神经网络的替代激活函数具有有效性,而且不会增加计算成本。进一步地,DeSign 在保留细节与二进制操作效率之间实现了平衡。

引用

@article{arxiv.2405.02220,
  title  = {Designed Dithering Sign Activation for Binary Neural Networks},
  author = {Brayan Monroy and Juan Estupiñan and Tatiana Gelvez-Barrera and Jorge Bacca and Henry Arguello},
  journal= {arXiv preprint arXiv:2405.02220},
  year   = {2024}
}

备注

7 pages