Document Set Expansion with Positive-Unlabeled Learning Using Intractable Density Estimation
Haiyang Zhang, Qiuyi Chen, Yanjie Zou, Jia Wang, Yushan Pan, Mark Stevenson
Abstract
The Document Set Expansion (DSE) task involves identifying relevant documents from large collections based on a limited set of example documents. Previous research has highlighted Positive and Unlabeled (PU) learning as a promising approach for this task. However, most PU methods rely on the unrealistic assumption of knowing the class prior for positive samples in the collection. To address this limitation, this paper introduces a novel PU learning framework that utilizes intractable density estimation models. Experiments conducted on PubMed and Covid datasets in a transductive setting showcase the effectiveness of the proposed method for DSE. Code is available from https://github.com/Beautifuldog01/Document-set-expansion-puDE.
BibTeX
@inproceedings{zhang-etal-2024-document,
title = "Document Set Expansion with Positive-Unlabeled Learning Using Intractable Density Estimation",
author = "Zhang, Haiyang and
Chen, Qiuyi and
Zou, Yanjie and
Wang, Jia and
Pan, Yushan and
Stevenson, Mark",
editor = "Calzolari, Nicoletta and
Kan, Min-Yen and
Hoste, Veronique and
Lenci, Alessandro and
Sakti, Sakriani and
Xue, Nianwen",
booktitle = "Proceedings of the 2024 Joint International Conference on Computational Linguistics, Language Resources and Evaluation (LREC-COLING 2024)",
month = may,
year = "2024",
address = "Torino, Italia",
publisher = "ELRA and ICCL",
url = "https://aclanthology.org/2024.lrec-main.460/",
pages = "5167--5173"
}