English

Convolutional neural network compression for natural language processing

Computation and Language 2018-05-29 v1 Machine Learning Neural and Evolutionary Computing

Abstract

Convolutional neural networks are modern models that are very efficient in many classification tasks. They were originally created for image processing purposes. Then some trials were performed to use them in different domains like natural language processing. The artificial intelligence systems (like humanoid robots) are very often based on embedded systems with constraints on memory, power consumption etc. Therefore convolutional neural network because of its memory capacity should be reduced to be mapped to given hardware. In this paper, results are presented of compressing the efficient convolutional neural networks for sentiment analysis. The main steps are quantization and pruning processes. The method responsible for mapping compressed network to FPGA and results of this implementation are presented. The described simulations showed that 5-bit width is enough to have no drop in accuracy from floating point version of the network. Additionally, significant memory footprint reduction was achieved (from 85% up to 93%).

Keywords

Cite

@article{arxiv.1805.10796,
  title  = {Convolutional neural network compression for natural language processing},
  author = {Krzysztof Wróbel and Marcin Pietroń and Maciej Wielgosz and Michał Karwatowski and Kazimierz Wiatr},
  journal= {arXiv preprint arXiv:1805.10796},
  year   = {2018}
}

Comments

7 pages, 4 figures, 6 tables