fadelmuli / covid-faq-bert

COVID-19 FAQ Retrieval System with BERT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

covid-faq-bert

This is a final project task from our information retrieval class where we built an Indonesian COVID-19 Frequently Asked Questions (FAQ) retrieval system. When a user asks a question related to COVID-19 in Indonesian language, our system will look up the answer on our dataset or corpus, which consists of thousands of question and answer pairs, and then retrieve the answer to the user based on the highest similarity between the question provided by the user and the question-answer pair in the corpus. For our system to be able to do this, we embed the question provided by the user and the corpus using the state-of-the-art pre-trained language model BERT and calculate the similarity using cosine similarity.

Dataset

We create the corpus by collecting a question-answer pairs in Indonesian related to COVID-19 from indonesian health forum website Alodokter. In this forum, the question was asked by a patient, and the answer was replied by a qualified, registered doctor. From this website, we obtained around 1800 question-answer pairs

Word Embedding

As mentioned before, we use pre-trained BERT to convert the questions and QA pairs to a dense vector representation. We evaluate several BERT variations and compare which one is better on processing Indonesian language. This variations are:

  • multilingual-BERT (m-BERT)
  • sentence-BERT (SBERT)
  • IndoBERT
  • A lite BERT (ALBERT)
  • XLM-RoBERTa

About

COVID-19 FAQ Retrieval System with BERT


Languages

Language:Jupyter Notebook 100.0%