skip to main content
10.1145/3644116.3644179acmotherconferencesArticle/Chapter ViewAbstractPublication PagesisaimsConference Proceedingsconference-collections
research-article

An Entity Prediction Method for Chinese Medical Knowledge Graph via Bert Sentence Embedding and Classification

Published:05 April 2024Publication History

ABSTRACT

Automatic Knowledge Graph Completion is becoming the main research direction of Knowledge graph construction. Among, the entity prediction method can complete the complement of entities in RDF triples, and is widely used in the generation process of the Knowledge graph. In order to complete the Chinese medical Knowledge graph, this paper proposes an entity prediction method based on BRRT sentence embedding and classification. This method needs three steps, the first step is to introduce a large-scale medical corpus to fine tune the basic BERT model into a BERT Domain model. The second step is obtaining the sentence embedding through the model for candidate triples. The third step is to obtain the top N candidate entity lists according to the ranking of classifier probabilities of all candidate. In order to verify the effectiveness of this method, a series of experiments are conducted on the BIOS. The experimental results show that the optimal accuracy of the entity prediction method in this paper is 20.5%, which is 7.2% higher than that using Word embedding+distance.

References

  1. Quan Wang, Zhendong Mao, Bin Wang, Li Guo. 2017. Knowledge graph embedding: a survey of approaches and applications. IEEE Transactions on Knowledge & Data Engineering, , 29 (12):2724-2743.Google ScholarGoogle ScholarCross RefCross Ref
  2. Antoine Bordes, Nicolas Usunier, Alberto Garcia-Duranl, Jason Weston, Oksana Yakhnenko. 2013. Translating embeddings for modeling multi-relational data. Proceedings of the 26th International Conference on Neural Information Processing Systems - Volume2. Curran Associates Inc, Red Hook, NY, United States, 2787–2795.Google ScholarGoogle ScholarDigital LibraryDigital Library
  3. Zhen Wang, Jianwen Zhang, Jianlin Feng, Zheng Chen. 2014. Knowledge graph embedding by translating on hyperplanes. Proceedings of the Twenty-Eighth AAAI Conference on Artificial Intelligence. AAAI Press,1112–1119.Google ScholarGoogle ScholarCross RefCross Ref
  4. Bishan Yang, Wen-tau Yih, Xiaodong He, Jianfeng Gao, Li Deng. 2014. Embedding entities and relations for learning and inference in knowledge bases. https://arxiv.org/pdf/1412.6575.pdf.Google ScholarGoogle Scholar
  5. Richard Socher, Danqi Chen, Christopher D. Manning, Andrew Y. Ng. 2013. Reasoning with neural tensor networks for knowledge base completion. NIPS'13: Proceedings of the 26th International Conference on Neural Information Processing Systems - Volume 1. Curran Associates Inc. 926–934.Google ScholarGoogle Scholar
  6. Zhen Wang , Jianwen Zhang, Jianlin Feng, Zheng Chen. 2014. Knowledge graph and text jointly Embedding. Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP). ACL, Stroudsburg,PA, 1591–1601.Google ScholarGoogle Scholar
  7. Ruobing Xie, Zhiyuan Liu, Jia Jia, Huanbo Luan, Maosong Sun. 2016. Representation learning of knowledge graphs with entity descriptions. AAAI'16: Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence. AAAI Press, 2659–2665.Google ScholarGoogle ScholarCross RefCross Ref
  8. Han Xiao, Minlie Huang, Xiaoyan Zhu. 2016. SSP: Semantic space projection for knowledge graph embedding with text descriptions. https://doi.org/10.48550/arXiv.1604.04835.Google ScholarGoogle ScholarCross RefCross Ref
  9. Mingxia Gao, Jianguo Lu, Furong Chen. 2022. Medical knowledge graph completion based on word embeddings. Information (Switzerland), 12(4):205.https://doi.org/10.3390/info13040205 .Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. An Entity Prediction Method for Chinese Medical Knowledge Graph via Bert Sentence Embedding and Classification

    Recommendations

    Comments

    Login options

    Check if you have access through your login credentials or your institution to get full access on this article.

    Sign in
    • Published in

      cover image ACM Other conferences
      ISAIMS '23: Proceedings of the 2023 4th International Symposium on Artificial Intelligence for Medicine Science
      October 2023
      1394 pages
      ISBN:9798400708138
      DOI:10.1145/3644116

      Copyright © 2023 ACM

      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      • Published: 5 April 2024

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited

      Acceptance Rates

      Overall Acceptance Rate53of112submissions,47%
    • Article Metrics

      • Downloads (Last 12 months)0
      • Downloads (Last 6 weeks)0

      Other Metrics

    PDF Format

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format .

    View HTML Format