arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.01647cs.AIcs.CLcs.IRcs.LG

面向知识图谱构建的生物医学关系抽取方法与模型比较

Comparison of biomedical relationship extraction methods and models for knowledge graph creation

University of Manchester(曼彻斯特大学);Bayer Pharmaceuticals R&D(拜耳制药研发)

Nikola Milosevic, Wolfgang Thielemann

AI总结:

本文比较了基于规则、传统机器学习与多种Transformer模型在生物医学关系抽取中的表现,发现基于PubMedBERT的模型在平衡数据上微调后F1达0.92,且Transformer模型对小规模和不平衡数据具有较好鲁棒性。

摘要:

生物医学研究正以指数级速度增长,科学家、研究人员和从业者已无法应对该领域已发表文献的数量。文献中呈现的知识需要被系统化,以便能够轻松查找、访问和验证各种主张与假设。知识图谱可以为文献中的语义知识表示提供这样的框架。然而,为了构建知识图谱,有必要以生物医学实体之间的关系形式抽取知识,并对实体和关系类型进行规范化。在本文中,我们提出并比较了几种基于规则的方法和基于机器学习的方法(朴素贝叶斯、随机森林作为传统机器学习方法的示例,以及基于DistilBERT、PubMedBERT、T5和SciFive的模型作为现代深度学习Transformer的示例),用于从生物医学文献中进行可扩展的关系抽取,并用于集成到知识图谱中。我们考察了这些不同方法对不平衡且相当小的数据集的鲁棒性。我们的实验表明,基于Transformer的模型能够很好地处理小数据集(由于在大数据集上的预训练)和不平衡数据集。表现最好的模型是基于PubMedBERT并在平衡数据上微调的模型,其报告的F1分数为0.92。基于DistilBERT的模型紧随其后,F1分数为0.89,且运行速度更快、资源需求更低。基于BERT的模型表现优于基于T5的生成式模型。

英文摘要:

Biomedical research is growing at such an exponential pace that scientists, researchers, and practitioners are no more able to cope with the amount of published literature in the domain. The knowledge presented in the literature needs to be systematized in such a way that claims and hypotheses can be easily found, accessed, and validated. Knowledge graphs can provide such a framework for semantic knowledge representation from literature. However, in order to build a knowledge graph, it is necessary to extract knowledge as relationships between biomedical entities and normalize both entities and relationship types. In this paper, we present and compare few rule-based and machine learning-based (Naive Bayes, Random Forests as examples of traditional machine learning methods and DistilBERT, PubMedBERT, T5 and SciFive-based models as examples of modern deep learning transformers) methods for scalable relationship extraction from biomedical literature, and for the integration into the knowledge graphs. We examine how resilient are these various methods to unbalanced and fairly small datasets. Our experiments show that transformer-based models handle well both small (due to pre-training on a large dataset) and unbalanced datasets. The best performing model was the PubMedBERT-based model fine-tuned on balanced data, with a reported F1-score of 0.92. DistilBERT-based model followed with F1-score of 0.89, performing faster and with lower resource requirements. BERT-based models performed better then T5-based generative models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑