Adaptive Linear Unit for Accurate Binary Neural Networks

Ruchan Mo,Ke Xu,Li Liu,Lingzhi Liu,Dong Wang
DOI: https://doi.org/10.1109/icsp56322.2022.9965306
2022-01-01
Abstract:Binary Neural Networks (BNNs) have show huge advantages over full-precision floating-point neural networks in both computing efficiency and storage cost. However, there is still a considerable accuracy gap between BNNs and their floating-point counterparts. In order to close the gap, we propose Adaptive Linear Units (AdaLU) which can stabilize the training process of BNN and improve the accuracy of BNNs effectively. AdaLU introduces learnable offsets, slopes and saturation value to the traditional linear rectification function. The learnable offsets can enhance useful feature information, while learnable slopes are effective for stabilizing the training process, and the learnable saturation value introduces smoother loss landscapes, indicating these networks are more generalizable. Extensive experiments on the CIFAR-10 and ImageNet datasets have shown that the proposed AdaLU delivered superior performance improvements over current advanced training methods for BNNs, especially on ImageNet dataset, when Bi-Real Net was used as the benchmark model, the proposed scheme improved the accuracy by 4.6% over the baseline network.
What problem does this paper attempt to address?