2020年度 人工知能学会全国大会(第34回)

講演情報

一般セッション

一般セッション » J-2 機械学習

[1J5-GS-2] 機械学習: 基礎理論 (2)

2020年6月9日(火) 17:20 〜 19:00 J会場 (jsai2020online-10)

座長:中口悠輝(NEC)

17:20 〜 17:40

[1J5-GS-2-01] TensorShader : 高次元ニューラルネットワーク深層学習フレームワーク

〇吉村 拓馬1 (1. 株式会社ポコアポコネットワークス)

キーワード:深層学習、高次元ニューラルネットワーク、GPGPU

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

講演PDFパスワード認証
論文PDFの閲覧にはログインが必要です。参加登録者の方は「参加者用ログイン」画面からログインしてください。あるいは論文PDF閲覧用のパスワードを以下にご入力ください。

パスワード