再发注意单位

内置注意力门控

将注意力机制无缝集成到GRU单元内部,使网络能够自适应关注序列中的重要部分,增强长期记忆能力。

Niu, Zhaoyang · 仲国强 · 岳国华 · Wang, Lina · Yu, Hui · Ling, Xiao · 董军宇

Neurocomputing 2023

技术优势

能关注重点

通过新增注意力门控,RAU能够自适应关注序列中的重要部分,而非平等对待所有输入。

无缝集成

注意力门控直接加在GRU单元内部,无需改变整体网络结构即可替换原有循环单元。

性能一致领先

在四类任务上与GRU及其他相关模型相比均取得更好结果,证明了模型的泛化能力。

应用场景