LEARNING APPARATUS, METHOD AND PROGRAM

To improve learning precision.SOLUTION: A learning apparatus according to the present embodiment includes a first acquisition unit, a classification unit, a first generation unit, a distribution loss calculation unit, and an update unit. The first acquisition unit acquires first data for learning. The classification unit inputs the first data for learning to a model, and generates a plurality of estimation vectors that are processing results of the model. The first generation unit generates an estimation distribution from the plurality of estimation vectors. The distribution loss calculation unit calculates a distribution loss between the estimation distribution and a target distribution to be a target in an inference using the model. The update unit updates parameters of the model, based on the distribution loss.SELECTED DRAWING: Figure 1 【課題】学習精度を向上させること。【解決手段】本実施形態に係る学習装置は、第１取得部と、分類部と、第１生成部と、分布ロス計算部と、更新部とを含む。第１取得部は、第１学習用データを取得する。分類部は、前記第１学習用データをモデルに入力し、前記モデルの処理結果である複数の推定ベクトルを生成する。第１生成部は、前記複数の推定ベクトルから推定分布を生成する。分布ロス計算部は、前記推定分布と、前記モデルを用いた推論において目標となる目標分布との間の分布ロスを計算する。更新部は、前記分布ロスに基づいて前記モデルのパラメータを更新する。【選択図】図１