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.

Task-Adaptive Feature Fusion for Generalized Few-Shot Relation Classification in an Open World Environment

Adaptive Prototypical Networks With Label Words and Joint Representation Learning for Few-Shot Relation Classification

Large Margin Prototypical Network for Few-shot Relation Classification with Fine-grained Features

Hybrid Attention-Based Prototypical Networks for Noisy Few-Shot Relation Classification.

Boosting Knowledge Base Automatically via Few-Shot Relation Classification

Adaptive class augmented prototype network for few-shot relation extraction

DPNet: domain-aware prototypical network for interdisciplinary few-shot relation classification

Multi-Channel Convolutional Neural Networks with Adversarial Training for Few-Shot Relation Classification (student Abstract)

Prototypical attention network for few-shot relation classification with entity-aware embedding module

Hybrid Enhancement-based prototypical networks for few-shot relation classification

FewRel: A Large-Scale Supervised Few-Shot Relation Classification Dataset with State-of-the-Art Evaluation

FewRel 2.0: Towards More Challenging Few-Shot Relation Classification

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

ReNAP: Relation Network with Adaptive Prototypical Learning for Few-Shot Classification

Adversarial Learning with Domain-Adaptive Pretraining for Few-Shot Relation Classification Across Domains

Few-shot learning based on hierarchical feature fusion via relation networks

Heterogeneous Representation Learning and Matching for Few-Shot Relation Prediction

Multi-view Interaction Learning for Few-Shot Relation Classification

Enhance Prototypical Network with Text Descriptions for Few-shot Relation Classification

Relation Adversarial Network for Low Resource Knowledge Graph Completion

Novel target attention convolutional neural network for relation classification