PALT: Parameter-Lite Transfer of Language Models for Knowledge Graph Completion
Shen, Jianhao, Wang, Chenguang, Yuan, Ye, Han, Jiawei, Ji, Heng, Sen, Koushik, Zhang, Ming, Song, Dawn
–arXiv.org Artificial Intelligence
This paper presents a parameter-lite transfer learning approach of pretrained language models (LM) for knowledge graph (KG) completion. Instead of finetuning, which modifies all LM parameters, we only tune a few new parameters while keeping the original LM parameters fixed. We establish this via reformulating KG completion as a "fill-in-the-blank" task, and introducing a parameter-lite encoder on top of the original LMs. We show that, by tuning far fewer parameters than finetuning, LMs transfer non-trivially to most tasks and reach competitiveness with prior state-of-the-art approaches. For instance, we outperform the fully finetuning approaches on a KG completion benchmark by tuning only 1% of the parameters. The code and datasets are available at \url{https://github.com/yuanyehome/PALT}.
arXiv.org Artificial Intelligence
Oct-24-2022
- Country:
- Asia > China (0.04)
- North America > United States
- Illinois (0.04)
- Europe
- Germany (0.04)
- Romania > Sud - Muntenia Development Region
- Giurgiu County > Giurgiu (0.04)
- Genre:
- Research Report (1.00)
- Industry:
- Health & Medicine (0.68)
- Technology: