News for hackers. Stuff that matters.
Re:Facebook open-sources deep-learning modules by contingencies on Friday January 16, 2015 @02:37PM (#8901139)
Paper with proper summary: http://arxiv.org/abs/1412.7580
Re:Facebook open-sources deep-learning modules by ajtulloch on Friday January 16, 2015 @02:42PM (#8901168)
This paper describes one part of the fbcunn release (the fast convolution layers implemented via FFT, with the source available at https://github.com/facebook/fbcunn/tree/master/src/cuda/fft). There's a lot more in fbcunn if you want to check it out.
Re:Facebook open-sources deep-learning modules by Hydraulix989 on Friday January 16, 2015 @03:15PM (#8901391)
In my experience, the fully connected layers are the bottleneck. The other issue was the alternating compute-heavy convolution and the IO-heavy pooling. I'm curious how this FFT implementation stacks up against cuDNN (what's the speedup like for just the convolutional layers? and then what's the overall speedup like?).
Re:Facebook open-sources deep-learning modules by ajtulloch on Friday January 16, 2015 @03:39PM (#8901531)
http://arxiv.org/pdf/1412.7580v2.pdf compares the convolutional implementation with the cuDNN layers. For the FC layers, it's just CuBLAS `sgemm`.