訓練事例の影響の軽量な推定

DOI Web Site 参考文献12件 オープンアクセス
  • 小林 颯介
    東北大学情報科学研究科 株式会社 Preferred Networks
  • 横井 祥
    東北大学情報科学研究科 理化学研究所革新知能統合研究センター
  • 鈴木 潤
    東北大学情報科学研究科 理化学研究所革新知能統合研究センター
  • 乾 健太郎
    東北大学情報科学研究科 理化学研究所革新知能統合研究センター

書誌事項

タイトル別名
  • Efficient Estimation of Influence of a Training Instance

抄録

<p>機械学習モデルの挙動の解釈において,各訓練事例がもたらす影響を理解することは重要である.単純にはデータセットから対象の訓練事例を除いて再訓練してモデルの変化を解析することもできるが,必要な計算量が非常に大きくなってしまい,特に膨大なパラメータのニューラルネットワークモデルへの適用が困難であった.本論文では,ニューラルネットワークモデルへの各訓練事例の影響の推定手法として,既存手法に比べて非常に効率的な方法を提案する.提案手法では,各事例の学習時に dropout を用いて,事例ごとに固有のサブネットワークのパラメータを更新せずに訓練を進めることで,訓練終了後には各訓練事例の影響を受けていないサブネットワークを自由に抽出し,それを影響値の推定に活かすことができる.実験では,提案手法を,文書分類と画像物体認識において,BERT および VGGNet に適用し訓練事例への紐付けを行うことで,解釈性の高い形でモデルの予測を解析できることを示した.また,サブネットワークの学習曲線の解析やデータフィルタリングの実験を通して,提案手法が事例間の関係性を適切に捉えていることを定量的に示した.</p>

収録刊行物

  • 自然言語処理

    自然言語処理 28 (2), 573-597, 2021

    一般社団法人 言語処理学会

参考文献 (12)*注記

もっと見る

関連プロジェクト

もっと見る

キーワード

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

問題の指摘

ページトップへ