HyperAI
Command Palette
Search for a command to run...
ニューラル機械翻訳におけるアテンションヘッドの不平等の軽減
ニューラル機械翻訳におけるアテンションヘッドの不平等の軽減
Zewei Sun Shujian Huang Xin-Yu Dai Jiajun Chen
概要
最近の研究では、Transformerにおけるアテンションヘッドが均等ではないことが示されている。この現象は、マルチヘッドアテンションの訓練における不均衡さおよびモデルが特定のヘッドに依存していることと関連していると考察される。この問題に対処するために、我々は2つの具体的なアプローチに基づいた単純なマスキング手法「HeadMask」を提案する。実験の結果、複数の言語対において翻訳性能の向上が確認された。さらに、後続の実証的分析により、本研究の仮説が支持され、提案手法の有効性が裏付けられた。