TensorShader : 高次元ニューラルネットワーク深層学習フレームワーク

DOI

書誌事項

タイトル別名
  • TensorShader : Deep Learning Framework for High-Dimensional Neural Networks

抄録

<p>近年, 複素数や四元数などの多元数を基とする高次元ニューラルネットワークの研究が進められている. 一方で, 高次元ニューラルネットワークをGPUで扱うことのできる深層学習フレームワークは未だに少なく, 実験の障壁となっている. そこで本研究では,複素数,四元数,3次元ベクトルを基とする深層学習フレームワークを開発した.このフレームワークでは, 高次元ニューラルネットワークを実装する際に課題となる一時計算領域の肥大を解消するため専用のCUDAカーネルを実装し, さらに丸め誤差の蓄積を避けるためFP32-FP32演算を採用した.これらにより,既存のフレームワークに比べ空間計算量と計算誤差の低減について優位であることを示す.</p>

収録刊行物

詳細情報 詳細情報について

  • CRID
    1390003825189220992
  • NII論文ID
    130007856785
  • DOI
    10.11517/pjsai.jsai2020.0_1j5gs201
  • 本文言語コード
    ja
  • データソース種別
    • JaLC
    • CiNii Articles
  • 抄録ライセンスフラグ
    使用不可

問題の指摘

ページトップへ