Few-shot relation extraction method based on self-supervised learning prototypical networks
Few-shot relation extraction (FSRE) is a key task in Information extraction that aims to train models using a small number of labeled samples to identify relation categories in sentences. Existing research widely adopts meta-learning approaches, with prototypical networks being popular due to their simplicity and efficiency. However, these methods mainly focus on distinguishing samples of different categories, establishing inter-class boundaries based solely on simple calculations of relation category samples, while ignoring the differences between instances and failing to fully utilize instance feature information. To address this issue, this paper proposes self-supervised learning prototypical networks (SSLPN), a self-supervised learning-based prototypical network method for FSRE. This model utilizes self-supervised learning to enhance sample discriminability and effectively learn category representations. SSLPN consists of three stages, and this three-stage architecture synergistically enhances sample discriminability, mitigates overfitting, and optimizes model performance for FSRE tasks. The FewRel1.0 dataset was used to evaluate SSLPN, and the outcomes demonstrated that the proposed method considerably enhanced the FSRE performance, confirming its efficacy.
Authors
- Yu Fu (ORCID: https://orcid.org/0000-0002-9795-7807)
- Shuang Qian (ORCID: https://orcid.org/0000-0002-2313-0268)
- Jia Cheng (ORCID: https://orcid.org/0000-0003-3116-4035)
- Mingqiang Wei (ORCID: https://orcid.org/0000-0003-0429-490X)
- Kunlin Li (ORCID: https://orcid.org/0000-0002-4996-5423)
- D. Li (ORCID: https://orcid.org/0009-0002-3793-1565)
- Yiqiu Sun (ORCID: https://orcid.org/0009-0002-8130-5115)
Institutions
- Zhengzhou University of Light Industry (CN)
- Henan University of Economic and Law (CN)
Publication Details
- Published
- 2025-03-18
- DOI
- https://doi.org/10.1117/12.3065060
- Primary Topic
- Technology and Security Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.00