Skip to content

Solving MNIST Dataset's Prediction using CNN or MLP, written in OpenCL

Notifications You must be signed in to change notification settings

qizy09/MNIST-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

30 Commits
 
 
 
 
 
 
 
 

Repository files navigation

MNIST-CNN

This is a repository for GPU acceleration of Machine Learning algorithms, including CNN and MLP. This repository is expected to continuously be maintained.

First Author

Ziyang Qi

Co-authors

Shuangchen Li, Yuan Xie

Abstract

Leading increased performance in benchmark tasks and to enable discovery of complex
high-level features, scaling up deep learning algorithms have earned researchers’
interests. Recent works train deep neural networks with deep layers and very high
dimension of parameters with the assistance of vast amount of computing power. The
Graphics Processing Units (GPUs) have been applied successfully in many areas for parallel
computing in recent years. Compared with the traditional CPU cluster, GPU has an obvious
advantage of low cost of hardware and electricity consumption.
Deep learning algorithms including Deep Neural Networks (DNN), whose forward and
backward propagation contains many inner products or matrix multiplications, which can
utilize the parallelism of GPU.
In this work, we are going to perform optimizations on the implementations of DNN on
GPU and heterogeneous platforms including AMD APU, Intel CPU + Nvidia GPU, AMD CPU,
and Intel CPU. Evaluation and analysis of the implementations are compared on respective
platforms.

License

Copyright 2016.
For any questions, feel free to let me know.

About

Solving MNIST Dataset's Prediction using CNN or MLP, written in OpenCL

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published