第37回 (2023)
セッションID: 1O4-GS-7-03
会議情報
主催: 一般社団法人 人工知能学会
会議名: 2023年度人工知能学会全国大会（第37回）
回次: 37
開催地: 熊本城ホール＋オンライン
開催日: 2023/06/06 - 2023/06/09
シーングラフ生成における最適輸送損失を用いた物体間関係予測
詳細
主催: 一般社団法人 人工知能学会
会議名: 2023年度人工知能学会全国大会（第37回）
回次: 37
開催地: 熊本城ホール＋オンライン
開催日: 2023/06/06 - 2023/06/09
シーングラフ生成における画像内の物体間の関係予測に対して, 最適輸送を用いた損失で学習することで, 従来のモデルを超える性能を一部で達成し, 学習データに少ない関係ラベルの再現率を高めた. シーングラフ生成には, データセットの関係ラベルに重度の不均衡があり, 一般的な交差エントロピーを損失とする学習ではその分布に偏った予測を行ってしまうという問題がある. そこで我々は, ラベル間の類似性を輸送コストとして明示的に反映させやすい最適輸送を損失とした学習を, シーングラフ生成における画像内の物体間の関係予測に対して行い, その効果を評価した. 最適輸送の輸送コストは, 事前学習済みモデルから得られる単語の類似度を使って定義した.
