CodeZine(コードジン)

特集ページ一覧

Facebook AI Research、ディープラーニングに最適化したフレームワーク「Torch」向けモジュールをOSSで公開

  • LINEで送る
  • このエントリーをはてなブックマークに追加
2015/01/19 19:45

 Facebook AI Research(FAIR)は、ディープラーニングに最適化した科学計算フレームワーク「Torch」向けのモジュールを、1月16日(現地時間)にオープンソースで公開した。

 今回公開したモジュールは、Torchデフォルトのモジュールよりも著しく高速で、FAIRが行っている研究における、ニューラルネットワーク接続の高速化を実現する。

 また、自然言語処理に用いられている畳み込みネットワーク(ConvNet)向けの、GPU最適化モジュールもあわせて公開している。ConvNet向けモジュールは、NVIDIAのcuFFTライブラリを使用したカスタムCUDAカーネルによる、高速フーリエ変換をベースにした畳み込みレイヤーを収録する。

 このほか、データ多重化モデルやモデル多重化モデルを用いて、学習を複数のGPUに分散するコンテナや、単語などの個別のオブジェクトや自然言語の学習に使用される最適化ルックアップテーブル、外部の多数のクラスを利用して学習速度を向上する階層化SoftMaxモジュールなど、複数のCUDAゲースのモジュールやコンテナをあわせて公開している。


【関連リンク】
Facebook AI Research(英語)
Torch(英語)

  • LINEで送る
  • このエントリーをはてなブックマークに追加

あなたにオススメ

All contents copyright © 2005-2021 Shoeisha Co., Ltd. All rights reserved. ver.1.5