A Cooperative Lightweight Translation Algorithm Combined with Sparse-ReLU

Xintao Xu,Yi Liu,Gang Chen,Junbin Ye,Zhigang Li,Huaxiang Lu
DOI: https://doi.org/10.1155/2022/4398839
IF: 3.12
2022-05-29
Computational Intelligence and Neuroscience
Abstract:In the field of natural language processing (NLP), machine translation algorithm based on Transformer is challenging to deploy on hardware due to a large number of parameters and low parametric sparsity of the network weights. Meanwhile, the accuracy of lightweight machine translation networks also needs to be improved. To solve this problem, we first design a new activation function, Sparse-ReLU, to improve the parametric sparsity of weights and feature maps, which facilitates hardware deployment. Secondly, we design a novel cooperative processing scheme with CNN and Transformer and use Sparse-ReLU to improve the accuracy of the translation algorithm. Experimental results show that our method, which combines Transformer and CNN with the Sparse-ReLU, achieves a 2.32% BLEU improvement in prediction accuracy and reduces the number of parameters of the model by 23%, and the sparsity of the inference model increases by more than 50%.
mathematical & computational biology,neurosciences
What problem does this paper attempt to address?