HyperAIHyperAI

Command Palette

Search for a command to run...

长源序列生成的分层学习

Tobias Rohde Xiaoxia Wu Yinhan Liu

摘要

当前序列到序列(seq2seq)模型面临的一个主要挑战是如何处理长序列,例如在摘要生成和文档级机器翻译任务中。这些任务要求模型能够在词元(token)、句子乃至段落等多个层次上进行推理。为此,我们设计并研究了一种新型的分层注意力Transformer架构(Hierarchical Attention Transformer, HAT),该架构在多个序列到序列任务中表现优于标准Transformer模型。此外,我们的模型在四项摘要任务(包括PubMed、arXiv、CNN/Daily Mail、SAMSum和AMI)上取得了当前最优的ROUGE得分。在WMT20英语到德语的文档级机器翻译任务中,该模型也超越了现有的基线模型。通过可视化分层编码器-解码器注意力机制,我们进一步探究了各分层所学习到的特征表示。最后,我们还研究了仅编码器架构下的分层预训练方法,并分析其在分类任务中的表现。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供