Fusing Sentence-Level and Document-Level Representations for Keyphrase Extraction

Marko Pranjić, Kristijan Armeni, Senja Pollak


Abstract
Supervised sequence-labeling approaches to keyphrase extraction operate at token level, but whether a phrase is relevant may depend on context beyond its sentence. Processing an entire document at token resolution can increase sequence length and introduce irrelevant information. We investigate a mixed-resolution context-fusion architecture that retains token representations for the target sentence while compressing available surrounding sentences into pooled representations. Both representation types are placed in document order and processed jointly by a self-attention module before IOB prediction. We evaluate the model on \textsf{kp20k}, \textsf{inspec}, and \textsf{kptimes} against SEKE, a recent state-of-the-art supervised system with approximately 60% more total parameters. The model achieves comparable F1 scores on the two scientific-domain benchmarks, while a larger performance gap remains on the journalistic benchmark. Error analysis shows that the model generates substantially more candidates than there are gold keyphrases. However, many predictions counted as false positives under exact-match evaluation partially overlap with gold keyphrases, while exact matches tend to receive higher scores than partial and non-matches.
Anthology ID:
2026.wasp-1.14
Volume:
Proceedings of the 4th Workshop on Artificial Intelligence for Scientific Publications
Month:
November
Year:
2026
Address:
Hengqin, Zhuhai, China
Editors:
Alberto Accomazzi, Tirthankar Ghosal, Felix Grezes, Kelly Lockhart, Atilla Kaan Alkan
Venues:
WASP | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
122–132
Language:
URL:
https://aclanthology.org/2026.wasp-1.14/
DOI:
Bibkey:
Cite (ACL):
Marko Pranjić, Kristijan Armeni, and Senja Pollak. 2026. Fusing Sentence-Level and Document-Level Representations for Keyphrase Extraction. In Proceedings of the 4th Workshop on Artificial Intelligence for Scientific Publications, pages 122–132, Hengqin, Zhuhai, China. Association for Computational Linguistics.
Cite (Informal):
Fusing Sentence-Level and Document-Level Representations for Keyphrase Extraction (Pranjić et al., WASP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.wasp-1.14.pdf