適用製品
畳み込みニューラルネットワーク学習装置およびそのプログラム
目的
動画翻訳ニューラルネットワークに用いるCNNを、動画とラベル列とのペアから、直接学習することが可能な畳み込みニューラルネットワーク学習装置およびそのプログラムを提供する。
効果
静止画と単独のラベルとを対応付けることができない動画からでも、CNNを学習することができる。
技術概要
動画を構成する静止画をラベル列に翻訳するCNNとRNNとを連結した動画翻訳ニューラルネットワークに用いられる前記CNNを事前学習する畳み込みニューラルネットワーク学習装置であって、
学習データである前記動画から前記静止画を順次切り出す静止画切り出し手段と、
前記CNNを用いて、前記静止画切り出し手段が切り出した前記静止画から、全ラベル数に対応する次元数のラベル確率分布を生成するラベル確率分布生成手段と、
前記ラベル確率分布生成手段が生成した複数のラベル確率分布から、ラベルごとに最大の確率値を選択し、ラベル最大値列を生成する最大値選択手段と、
前記学習データである前記動画に対応する正解データであるラベル列と、前記最大値選択手段で生成したラベル最大値列との誤差を計算する誤差計算手段と、
前記誤差計算手段が計算した前記誤差を小さくする方向に前記CNNのパラメータを更新するパラメータ更新手段と、
を備えることを特徴とする畳み込みニューラルネットワーク学習装置。