Machine Learnin backed FAQ Chatbot

The goal of this project was to develop an AI backend for a FAQ chatbot. A client had collected significant amount of questions and answers that were used to train machine learning models. Besides text, the questions had assigned tags which were used to cluster questions info topic-based segments.  

The overall algorithm was trained as follows:

  • first we trained a topic classifier based on the assigned tags; for this we performed TF-IDF transformation and we trained an XGBoost classifier
  • second, for each topic we built a Doc2Vec embeddings targeting only the specific topic. Such approach allowed us to save computational costs and memory footprint, as only one relatively small model was loaded to the main memory when processing a question. 

An arriving question was processed as follows:

  • first we had to use a generic classifier to find the right topic,
  • then we loaded the embeddings for the topic,
  • finally, we found the most similar embedding to the question

The model was deployed and served as a REST API within a docker container, what enabled horizontal scalability of the solution. 

Avatar
Marcin Wylot, PhD
Data Scientist & Machine Learning Engineer

20 years of experience in data processing from A to Z

Related