Information Materials and Intelligent Sensing Laboratory of Anhui Province, School of Computer Science and Technology, Anhui University, Hefei, China.
Information Materials and Intelligent Sensing Laboratory of Anhui Province, School of Computer Science and Technology, Anhui University, Hefei, China; Institute of Artificial Intelligence, Hefei Comprehensive National Science Center, China.
Neural Netw. 2022 Nov;155:50-57. doi: 10.1016/j.neunet.2022.08.003. Epub 2022 Aug 6.
Graph Neural Networks (GNNs) have been employed for few-shot learning (FSL) tasks. The aim of GNN based FSL is to transform the few-shot learning problem into a graph node classification or edge labeling tasks, which can thus fully explore the relationships among samples in support and query sets. However, existing works generally consider the graph learned by node features which ignore the initial pairwise label constraints and thus are generally not guaranteed to be optimal for FSL tasks. Also, existing works generally learn graph edges independently based on node's own features which lack of considering the consistent relationships among different edges. To address these issues, we propose a novel Label Guided Graph Learning-Neural network (LGLNN) model for FSL tasks. The aim of LGLNN is to incorporate the label information to learn an optimal metric graph for GNN by employing the pairwise constraint propagation. The main advantage of LGLNN is that it can learn the metrics (both similarity and dissimilarity) for each graph edge by aggregating the metric information from its neighboring edges and thus can conduct metric learning of all edges cooperatively and consistently. Experimental results demonstrate the effectiveness and better performance of the proposed LGLNN method.
图神经网络 (GNN) 已被用于少样本学习 (FSL) 任务。基于 GNN 的 FSL 的目的是将少样本学习问题转化为图节点分类或边标记任务,从而可以充分探索支持集和查询集中样本之间的关系。然而,现有工作通常考虑基于节点特征学习的图,忽略了初始的成对标签约束,因此通常不能保证对 FSL 任务是最优的。此外,现有工作通常基于节点自身的特征独立学习图边,缺乏对不同边之间一致关系的考虑。为了解决这些问题,我们提出了一种新的用于 FSL 任务的标签引导图学习神经网络 (LGLNN) 模型。LGLNN 的目的是通过使用成对约束传播将标签信息纳入到图神经网络中,以学习最优的度量图。LGLNN 的主要优势在于它可以通过聚合来自相邻边的度量信息来学习每条图边的度量(相似性和相异性),从而可以协作一致地进行所有边的度量学习。实验结果证明了所提出的 LGLNN 方法的有效性和更好的性能。