Понижение размерности методом градиентного спуска с использованием графических ускорителей

Abstract

В статье рассматриваются различные способы реализации алгоритма градиентного спуска, используемого для понижения размерности данных, предназначенные для графических процессоров. Представлены 4 варианта реализации градиентного спуска для графических процессоров, написанные на языке HIP. Удалось достичь 6-кратного улучшения производительности при использовании графического процессора AMD Radeon RX Vega 56 относительно многопоточной реализации, исполняемой на центральном процессоре. In this papeг we discuss possible realizations of GPU-targeted gradient descent algorithm used for dimensionality reduction. 4 realizations of gradient descent algorithm were created using HIP - a new framework for GPGPU programming. We got 6 times perfomance improvement over multithreaded CPU version using AMD Radeon RX Vega 56.