Facebook AI Research has announced that is open sourcing the deep-learning modules that enable it to train larger neural nets in less time than those already available.
Fortunately for the area of deep learning and convolutional nets it believes that:
Progress in science and technology accelerates when scientists share not just their results, but also their tools and methods.
Hence its decision to do just that with a set of tools that give a 23.5x speed-up over publicly available convolutional layer codes, and the ability to parallelize neural networks training over GPU cards.
The tools are being made available for Torch. an open source development environment for numerics, machine learning, and computer vision widely used at a number of academic labs as well as at Google/DeepMind, Twitter, NVIDIA, AMD, Intel, and many other companies.
The following fast nn modules for Convnets and neural networks in general are provide a a plug-in to the Torch-7 framework:
Fast Temporal convolutions that are 1.5x to 10x faster compared to Torch's cunn implementations.
nn.DataParallel and nn.ModelParallel containers. Plug your model in them and see it accelerate over multiple GPUs,
Wrappers to use FFT/IFFT as nn modules.
Fast LookupTable that is used for Neural Language Models and word embeddings. Much faster than the one in torch/nn
Hierarchical SoftMax module, now classifying 1 million classes is a practically viable strategy
LP and Max Pooling over feature maps (usable for MaxOut).
To use these packages for Torch, visit the fbcunn page which has installation instructions, documentation and examples to train classifiers over ImageNet.
Facebook has also recently released iTorch, an interface for Torch using iPython with visualization and plotting and previously has made available fbnn, extensions to torch/nn, fbcuda, extensions to CUDA, and fblualib libraries and utilities for Lua.
Concluding the announcement on the FAIR Blog, Soumith Chintala notes:
We hope that these high-quality code releases will be a catalyst to the research community and we will continue to update them from time to time.
Last month in front of an enthusiastic audience at Y Combinator, Fei-Fei Li, often called the godmother of AI, talked to Diana Hu about spatial intelligence and why she considers it the next crit [ ... ]
