Biblioteca122.739 documentos en línea

Artículo

High Performance Implementation of 3D Convolutional Neural Networks on a GPUImplementación de alto rendimiento de redes neuronales convolucionales 3D en una GPU

Resumen

Las redes neuronales convolucionales han demostrado ser muy exitosas en aplicaciones como la clasificación de imágenes, el seguimiento de objetos y muchas otras tareas basadas en entradas 2D. Recientemente, los investigadores han empezado a aplicar las redes neuronales convolucionales a la clasificación de vídeos, que constituyen una entrada 3D y requieren cantidades mucho mayores de memoria y mucho más cálculo. Los métodos basados en la FFT pueden reducir la cantidad de cálculo, pero esto suele tener como contrapartida un aumento de las necesidades de memoria. Por otro lado, el algoritmo de filtrado mínimo de Winograd (WMFA) puede reducir el número de operaciones necesarias y, por tanto, acelerar el cálculo, sin aumentar la memoria necesaria. Esta estrategia ha demostrado su eficacia en redes neuronales 2D. Implementamos el algoritmo para redes neuronales convolucionales 3D y lo aplicamos a una popular red neuronal convolucional 3D que se utiliza para clasificar vídeos y la comparamos con cuDNN. Para nuestra implementación altamente optimizada del algoritmo, observamos un aumento de velocidad del doble para la mayoría de las capas de convolución 3D de nuestra red de prueba en comparación con la versión cuDNN.

  • Tipo de documento:
  • Formato:pdf
  • Idioma:Inglés
  • Tamaño: Kb

Cómo citar el documento

Esta es una versión de prueba de citación de documentos de la Biblioteca Virtual Pro. Puede contener errores. Lo invitamos a consultar los manuales de citación de las respectivas fuentes.

Este contenido no est� disponible para su tipo de suscripci�n

Información del documento