Abstract:Supervised learning methods excel in traditional relation extraction tasks. However, the quality and scale of the training data heavily influence their performance. Few-shot relation extraction is gradually becoming a research hotspot whose objective is to learn and extract semantic relationships between entities with only a limited number of annotated samples. In recent years, numerous studies have employed prototypical networks for few-shot relation extraction. However, these methods often suffer from overfitting of the relation classes, making it challenging to generalize effectively to new relationships. Therefore, this paper seeks to utilize a diffusion model for data augmentation to address the overfitting issue of prototypical networks. We propose a diffusion model-enhanced prototypical network framework. Specifically, we design and train a controllable conditional relation generation diffusion model on the relation extraction dataset, which can generate the corresponding instance representation according to the relation description. Building upon the trained diffusion model, we further present a pseudo-sample-enhanced prototypical network, which is able to provide more accurate representations for prototype classes, thereby alleviating overfitting and better generalizing to unseen relation classes. Additionally, we introduce a pseudo-sample-aware attention mechanism to enhance the model's adaptability to pseudo-sample data through a cross-entropy loss, further improving the model's performance. A series of experiments are conducted to prove our method's effectiveness. The results indicate that our proposed approach significantly outperforms existing methods, particularly in low-resource one-shot environments. Further ablation analyses underscore the necessity of each module in the model. As far as we know, this is the first research to employ a diffusion model for enhancing the prototypical network through data augmentation in few-shot relation extraction.

Few-Shot Relation Extraction with Dual Graph Neural Network Interaction.

Introducing Graph Neural Networks for Few-Shot Relation Prediction in Knowledge Graph Completion Task

Long-tail Relation Extraction via Knowledge Graph Embeddings and Graph Convolution Networks.

Adapting Distilled Knowledge for Few-shot Relation Reasoning over Knowledge Graphs

Co-Attention Based Few-Shot Relation Classification Model with Dynamic Routing

Transfer Learning for Relation Extraction Via Relation-Gated Adversarial Learning

Heterogeneous graph neural networks for noisy few-shot relation classification

Sample Feature Enhancement Model Based on Heterogeneous Graph Representation Learning for Few-shot Relation Classification

Gated graph convolutional network with enhanced representation and joint attention for distant supervised heterogeneous relation extraction

Bridging Text and Knowledge with Multi-Prototype Embedding for Few-Shot Relational Triple Extraction.

Dual Attention Graph Convolutional Network for Relation Extraction

Learning Discriminative Semantic and Multi-view Context for Domain Adaptive Few-Shot Relation Extraction.

Adaptive Attentional Network for Few-Shot Relational Learning of Knowledge Graphs

Transductive Relation-Propagation With Decoupling Training for Few-Shot Learning

DiffFSRE: Diffusion-Enhanced Prototypical Network for Few-Shot Relation Extraction

Adaptive class augmented prototype network for few-shot relation extraction

Learning Dual-Pooling Graph Neural Networks for Few-Shot Video Classification

Heterogeneous Representation Learning and Matching for Few-Shot Relation Prediction

Graph Enhanced Dual Attention Network for Document-Level Relation Extraction.

Relation-oriented few-shot knowledge graph prototype networks

Few-Shot Relation Prediction of Knowledge Graph via Convolutional Neural Network with Self-Attention