HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

PromptRank: Unsupervised Keyphrase Extraction Using Prompt

Aobo Kong Shiwan Zhao Hao Chen Qicheng Li Yong Qin Ruiqi Sun Xiaoyan Bai

PromptRank: Unsupervised Keyphrase Extraction Using Prompt

Abstract

The keyphrase extraction task refers to the automatic selection of phrases from a given document to summarize its core content. State-of-the-art (SOTA) performance has recently been achieved by embedding-based algorithms, which rank candidates according to how similar their embeddings are to document embeddings. However, such solutions either struggle with the document and candidate length discrepancies or fail to fully utilize the pre-trained language model (PLM) without further fine-tuning. To this end, in this paper, we propose a simple yet effective unsupervised approach, PromptRank, based on the PLM with an encoder-decoder architecture. Specifically, PromptRank feeds the document into the encoder and calculates the probability of generating the candidate with a designed prompt by the decoder. We extensively evaluate the proposed PromptRank on six widely used benchmarks. PromptRank outperforms the SOTA approach MDERank, improving the F1 score relatively by 34.18%, 24.87%, and 17.57% for 5, 10, and 15 returned results, respectively. This demonstrates the great potential of using prompt for unsupervised keyphrase extraction. We release our code at https://github.com/HLT-NLP/PromptRank.

Code Repositories

nku-hlt/promptrank
pytorch
Mentioned in GitHub
hlt-nlp/promptrank
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
keyphrase-extraction-on-inspecPromptRank
F1@10: 37.88
keyphrase-extraction-on-krapivinPromptRank
F1@10: 16.71
keyphrase-extraction-on-nusPromptRank
F1@10: 20.13

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
PromptRank: Unsupervised Keyphrase Extraction Using Prompt | Papers | HyperAI