B-VGG16: Red Neuronal de Convolución cuantizada binariamente para la clasificación de imágenes

• Urbano Pintos, Nicolás ^[1] ; Lacomi, Héctor ^[1] ; Lavorato, Mario ^[1]
  1. [1] Universidad Tecnológica Nacional

     Universidad Tecnológica Nacional

     Argentina

     
• Localización: Elektron: ciencia y tecnología en la electrónica de hoy, ISSN-e 2525-0159, Vol. 6, Nº. 2, 2022, págs. 107-114
• Idioma: español
• Títulos paralelos:
  • B-VGG16: Binary Quantized Convolutional Neuronal Network for image classification
• Enlaces
  • Texto completo (pdf)
• Resumen
  • español

    En este trabajo se entrena y evalúa una red neuronal de convolución cuantizada de forma binaria para la clasificación de imágenes. Las redes neuronales binarizadas reducen la cantidad de memoria, y es posible implementarlas con menor hardware que las redes que utilizan variables de valor real (Floating Point 32 bits). Este tipo de redes se pueden implementar en sistemas embebidos, como FPGA. Se realizó una cuantización consciente del entrenamiento, de modo de poder compensar los errores provocados por la pérdida de precisión de los parámetros. El modelo obtuvo una precisión de evaluación de un 88% con el conjunto de evaluación de CIFAR-10.

  • English

    In this work, a Binary Quantized Convolution neural network for image classification is trained and evaluated. Binarized neural networks reduce the amount of memory, and it is possible to implement them with less hardware than those that use real value variables (Floating Point 32 bits). This type of network can be implemented in embedded systems, such as FPGA. A quantization-aware training was performed, to compensate for the errors caused by the loss of precision of the parameters. The model obtained an evaluation accuracy of 88% with the CIFAR-10 evaluation set.