Skip to content

AI models for automatic job application pipeline (user CV, job description analysis (customized NER/SpaCy) and artificial cover letter generation (trained GPT-2 model) created for Jobzilla project within TechLabs Berlin AI Track programm (03.2021-07.2021).

License

Notifications You must be signed in to change notification settings

pandmi/jobzilla_ai

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Three created AI-models for Jobzilla

1. model:simple without labels and metadata

Based on gpt-2-simple library from Max Woolf - https://minimaxir.com/2019/09/howto-gpt2/

Test finetuning - execution:

  1. Upload notebook file to Google colab
  2. Set the runtime tipe as GPU
  3. Execute all cells from top to bottom

2. model: with Title labels

Model was trained based on cover letters text bodies with job titles and skills as aditional metadata.

3. model: KWs (skills) - extraction

Based on SpyCy NEP with custom keywords matching (json - file with skills - attached). Extraction algorithm was used for job and cv analysis, as well for similarity matching.

API - deployment

Models were containerized (Docker), set up as API (FastAPI) and deployed on Google Cloud Plattform.

(For customized letter creation as prefx varibale should be used -Job Title-skill1, skill2, skill3)

Jobzilla - AI - deep dive into creation workflow

From the AI perspective, my main goal was to create a machine learning model which was able to generate cover letters undistinguished from a human-written one. After reviewing different NLG approaches I have decided for the training already existing transformer model, as the most efficient approach.

The first training of the small 124-M parameters GPT-2 model with the dataset of collected cover letters gave me very impressive results (loss=0.19, avg=0.64, 1500 steps): the newly trained model was able to create the text on a very high qualitative level, but the content was not relevant to the candidate skills and job title. For the model training, I have adjusted gpt-2-simple framework from Max Woolf. [1]

To get more control over the model output I have decided also to add to the training dataset skills extracted from cover letters and job titles. To find this approach helped me the article from Ivan Lai. [2]

It was a challenge to find the most suitable model, which was able to extract correctly the skills from cover letters, but after some experiments with different NLP frameworks (BERT, TFIDF), I decided to use SpaCy Named Entity Recognition model (en_core_web_sm) combined with EntityRuler for custom skills labeling. The skills dataset I found in Microsoft repository on Github. [3]

Although the used dataset had only ca. 2000 listed skills the final model matched with my expectations, most of the candidate skills in the tested cover letters were correctly identified:

Img 1: Example of original cover letter from training dataset for the position Data Scientist

The created NER (named entity recognition) framework helped me to add into the training dataset skills and job titles. The retrained GPT-2 based model was able to create a cover letter by prompting a set of skills, job title, or combination of both. In the results I was able to see that the generated text was relevant to the passed into the model information:

Img 2: Model output after prompt text: Data Scientist~python,sql,algorithms

I need to mention that the description of experiences, the name of the candidate as well as the recipient was created here by the model randomly. My next step will be extending the controlled parameters by those options, but I expect that here might be implemented the same approach as with the skills.

Optimal results were achieved by using minimal numbers of skills as the prompt message – two or three, with a high probability those skills were recognized and implemented in the model text. Probably due to the small size of the training dataset the model was not capable of working with a bigger number of elements for proper prediction.

The collected learnings gave the model opportunity to ignore by the text generation those prompted skills which are completely uncommon for the specific job profile, so it was not possible to create a cover letter for a Data Scientist with mention of a car driving license as a skill, for example.

Another learning I have collected by the training is the fact that the used transformer model was able to recognize the weight of the specific skills via their count in the specific text. So it was also possible due to duplication of some skills in the prompt to increase their representation in the output text.

Finally, both models (cover letter generation and skill extraction) were set as API service (FastAPI) and containerized (Docker) and deployed into Google Cloud Platform which helped to bring them into production.

[1] https://github.com/minimaxir/gpt-2-simple

[2] https://www.ivanlai.project-ds.net/post/conditional-text-generation-by-fine-tuning-gpt-2

[3] https://github.com/microsoft/SkillsExtractorCognitiveSearch/blob/master/data/skill_patterns.jsonl

About

AI models for automatic job application pipeline (user CV, job description analysis (customized NER/SpaCy) and artificial cover letter generation (trained GPT-2 model) created for Jobzilla project within TechLabs Berlin AI Track programm (03.2021-07.2021).

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published