设为首页收藏本站

LUPA开源社区

 找回密码
 注册
文章 帖子 博客
LUPA开源社区 首页 业界资讯 开源资讯 查看内容

Facebook开源DL模块带给我们什么?

2015-1-22 12:36| 发布者: joejoe0332| 查看: 2391| 评论: 0|原作者: 周建丁|来自: CSDN

摘要: 日前,Facebook人工智能研究院(FAIR)宣布开源了一组深度学习软件库,是针对Torch机器学习框架的插件,基于NVIDIA的GPU,大大提升了神经网络的性能,可以用于计算机视觉和自然语言处理(NLP)等场景。 ...


多GPU之上的并行化

  从工程方面,Facebook一直努力实现在多GPU上并行训练神经网络模型的能力。Facebook致力于最小化并行的开销,同时也使研究人员极易使用,通过数据并行和模型并行模块(fbcunn的一部分)。当研究人员把它们的模型推到这些易于使用的容器,代码自动调度模型在多个GPU上最大化加速。Facebook使用多GPU在ImageNet上训练一个ConvNet的一个例子已经展示了这一点。


fbcunn是什么

  这个库包含Facebook用于GPU的高度工程化深度学习模块,以加快深度学习的速度。它插入到Torch7的框架之中,通过luarocks无缝安装,并且与Torch的NN封装完全兼容。

  总体来说,Facebook发布了用于Convnets和一般神经网络的快速NN模块:

  • 快速空间卷积模块,使用FFT加速卷积。
  • 快速Temporal卷积,速度相比Torch的cunn实现快1.5倍至10倍。
  • nn.DataParallel和nn.ModelParallel容器。将您的模型插入,在多个GPU上加速效果立竿见影。
  • 采用FFT/ IFFT为NN模块。
  • 用于神经语言模块(Neural Language Models)和文字嵌入的快速LookupTable。比在Torch/ NN下快很多。
  • Hierarchical SoftMax模块,使得现在分类百万类成为实际可行的策略。
  • 特征映射上的LP和Max Pooling (用于MaxOut)。
  • 更多的好东西。

示例:

  • 在Torch-7中使用多GPU训练基于分类的imagenet(展示FFT卷积以及ModelParallel容器)


fbcunn如何使用

  • DataParallel和ModelParallel这两个模块超级简单易用。这个unit-test同时作为例子和测试。examples/imagenet下面还有ModelParallel的例子。
m = nn.DataParallel():add(nn.SpatialConvolution(...)):add(nn.ReLU())-- see, so simple<br>
  • 卷积模块更加简单易用。它们与NN的API完全兼容。
<p>CONV= nn.SpatialConvolutionCuFFT(...)-- fast spatial convolutions!</p><p>CONV= nn.TemporalConvolutionFB(...)-- fast temporal convolutions!</p>
  • LookupTable和Hierarchical SoftMax分别被命名为nn.LookupTableGPU和nn.HSM,也是超级简单易用。它们的使用,以及fbcunn的安装步骤、API文档等更多的细节,您都可以在这个链接的完整的文档和规范中找到答案:http://facebook.github.io/fbcunn/fbcunn/。


编后语

  极客们说,人工智能是世界的未来,深度学习算法是让机器拥有智能的最佳途径。我们看到,从的相册处理到无人驾驶汽车,深度学习的模型已经在逐步应用。尽管目前的技术成果距离S.W.Hawking和Elon Musk所担忧的“机器政变”还非常遥远,但在硬件加速技术的支撑和开源社区的贡献之下,深度学习算法的普及应用门槛确实大大降低,模式识别、NLP变得更加简单,我们没有理由不加以学习和利用。

  因此,2015年,人工智能也将成为CSDN的一大重点内容,我们致力于为大家提供最新鲜、最实用的干货,服务于大家的工作和生活。人工智能最新动态,敬请留意CSDN 官网、CSDN人工智能社区、CSDN云计算微博和CSDN大数据微信的更新。此外,如果您有好的内容、好的意见和建议可以提供,欢迎致邮:zhoujd at csdn dot net。


原文链接: FAIR open sources deep-learning modules for Torch(文/周建丁)

酷毙

雷人

鲜花

鸡蛋

漂亮
  • 快毕业了,没工作经验,
    找份工作好难啊?
    赶紧去人才芯片公司磨练吧!!

最新评论

关于LUPA|人才芯片工程|人才招聘|LUPA认证|LUPA教育|LUPA开源社区 ( 浙B2-20090187 浙公网安备 33010602006705号   

返回顶部