Document Classification for COVID-19 Literature

Bernal Jimenez Gutierrez, Juncheng Zeng, Dongdong Zhang, Ping Zhang, Yu Su

Keywords: COVID-19, BERT, LitCovid, Document Classification, Biomedical NLP, BioBERT, Longformer, Coronavirus

Abstract: The global pandemic has made it more important than ever to quickly and accurately retrieve relevant scientific literature for effective consumption by researchers in a wide range of fields. We provide an analysis of several multi-label document classification models on the LitCovid dataset, a growing collection of 8,000 research papers regarding the novel 2019 coronavirus. We find that pre-trained language models fine-tuned on this dataset outperform all other baselines and that the BioBERT and novel Longformer models surpass all others with almost equivalent micro-F1 and accuracy scores of around 81% and 69% on the test set. We evaluate the data efficiency and generalizability of these models as essential features of any system prepared to deal with an urgent situation like the current health crisis. Finally, we explore 50 errors made by the best performing models on LitCovid documents and find that they often (1) correlate certain labels too closely together and (2) fail to focus on discriminative sections of the articles; both of which are important issues to address in future work. Both data and code are available on GitHub.

Community Implementations: [ 3 code implementations](https://www.catalyzex.com/paper/arxiv:2006.13816/code)