Skip to main content
QUICK REVIEW

[论文解读] Multi-Level Matching and Aggregation Network for Few-Shot Relation Classification

Zhi-Xiu Ye, Zhen-Hua Ling|arXiv (Cornell University)|Jun 16, 2019
Domain Adaptation and Few-Shot Learning参考文献 29被引用 20
一句话总结

本文提出了一种多层级匹配与聚合网络(MLMAN),用于少样本关系分类,通过在局部和实例层级上联合建模查询-支持样本之间的交互关系。通过采用交互式匹配与基于注意力的聚合机制,MLMAN在FewRel基准上实现了最先进性能,通过增强表示学习与一致性正则化,超越了先前的原型网络。

ABSTRACT

This paper presents a multi-level matching and aggregation network (MLMAN) for few-shot relation classification. Previous studies on this topic adopt prototypical networks, which calculate the embedding vector of a query instance and the prototype vector of each support set independently. In contrast, our proposed MLMAN model encodes the query instance and each support set in an interactive way by considering their matching information at both local and instance levels. The final class prototype for each support set is obtained by attentive aggregation over the representations of its support instances, where the weights are calculated using the query instance. Experimental results demonstrate the effectiveness of our proposed methods, which achieve a new state-of-the-art performance on the FewRel dataset.

研究动机与目标

  • 为解决因罕见关系标注数据不足而导致的关系分类长尾问题。
  • 通过建模查询样本与支持集之间的交互依赖关系,而非将其独立处理,以改进少样本关系分类。
  • 通过根据查询与支持实例的匹配分数加权,采用基于注意力的聚合方式,增强原型学习。
  • 通过引入辅助一致性损失,对每个类别内支持实例表示进行对齐,提升模型鲁棒性。

提出的方法

  • 使用受句子匹配框架启发的软注意力机制,在查询与支持实例表示之间执行局部匹配。
  • 通过最大池化与平均池化操作,将局部表示聚合为查询与支持实例的嵌入表示。
  • 使用多层感知机(MLP)计算实例层级的匹配分数,以衡量查询与每个支持实例之间的相似性。
  • 通过基于注意力的聚合方式生成类别原型,其中注意力权重由实例层级的匹配分数生成。
  • 使用可学习的MLP作为类别匹配函数,计算查询与每个类别原型之间的最终分类得分。
  • 引入辅助一致性损失,以最小化每个类别内支持实例表示之间的不一致性,从而提升原型质量。

实验结果

研究问题

  • RQ1在多个层级上对查询与支持实例进行交互建模,是否能超越独立编码方式,显著提升少样本关系分类性能?
  • RQ2与直接计算原型相比,查询与支持实例之间的局部匹配是否能提升表示质量?
  • RQ3基于注意力的实例层级聚合方式与简单平均相比,在构建类别原型时表现如何?
  • RQ4在少样本关系分类中,可学习的MLP匹配函数是否能超越基于度量的替代方法(如欧氏距离)?
  • RQ5辅助一致性损失是否能提升模型在低资源关系上的泛化能力与性能?

主要发现

  • MLMAN在FewRel数据集上实现了新的最先进性能,展现出在少样本关系分类任务中的优越泛化能力。
  • 消融实验表明,移除局部匹配会导致性能显著下降,凸显其在表示学习中的关键作用。
  • 当引入局部匹配时,使用可学习MLP进行类别匹配的表现优于欧氏距离,表明其能更好地建模复杂依赖关系。
  • 辅助一致性损失有效降低了支持实例表示之间的方差,提升了原型稳定性与分类准确率。
  • 不包含拼接或局部匹配的模型变体性能劣于完整模型,证实结构化交互在表示学习中的关键作用。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。