4 个月前

将多个贪婪依存句法分析器蒸馏为一个MST分析器

将多个贪婪依存句法分析器蒸馏为一个MST分析器

摘要

我们介绍了两种基于图的一阶依存句法分析器,它们达到了新的技术水平。第一种是一个共识分析器,由一组具有不同随机初始化的独立训练的贪婪LSTM转换分析器组成。我们将这种方法视为最小贝叶斯风险解码(在汉明成本下),并认为集合内部较弱的共识是有用的难度或歧义信号。第二种分析器是将该集合“蒸馏”为单个模型。我们使用结构化铰链损失目标函数训练蒸馏分析器,其中包含了一种新颖的成本计算方法,该方法结合了每个可能依存关系的集合不确定性估计,从而避免了将标准蒸馏目标应用于结构化输出问题时所需的复杂交叉熵计算。一阶蒸馏分析器在英语、汉语和德语上的表现与现有技术水平相当或更优。

代码仓库

adhigunasurya/distillation_parser
官方
GitHub 中提及

基准测试

基准方法指标
dependency-parsing-on-penn-treebankDistilled neural FOG
LAS: 92.06
POS: 97.44
UAS: 94.26

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
将多个贪婪依存句法分析器蒸馏为一个MST分析器 | 论文 | HyperAI超神经