Mathematical Framework for Custom Reward Functions in Job Application Evaluation using Reinforcement Learning

TL;DR

This paper introduces a two-stage fine-tuning approach using supervised learning and reinforcement learning with a custom reward function to improve resume assessment accuracy in applicant tracking systems, surpassing traditional methods.

Contribution

It presents a novel two-step fine-tuning pipeline with a custom reward function and hyperparameter tuning to enhance small language models for candidate evaluation.

Findings

Achieved 91% accuracy on unseen data

High recall of 0.85 with perfect precision of 1.0

Demonstrated stability and reliability in candidate assessment

Abstract

Most of the traditional Applicant Tracking Systems (ATS) depend on strict matching using keywords, where candidates that are highly qualified are many times disqualified because of minor semantic differences. In this article, the two-stage process of developing a more comprehensive resume assessment system based on a small language model that is trained with fewer than 600M parameters is introduced and fine-tuned by using GRPO with a uniquely designed reward function. The initial stage is Supervised Fine-Tuning (SFT), which is used to create a strong base model with the ability to perceive resumes beyond superficial overlap of keywords. This SFT model is further optimized in the second step with Reinforcement Learning (RL) via GRPO with the help of multi-component-based rewarding, which will not be considered as a commission of tokens matching. In the initial RL experiments, we found a severe difficulty in the shape of reward hacking: overly aggressive penalty terms resulted in unstable training dynamics and prohibitively negative model behavior. This was solved by trial-and-error refinement of the reward and careful training hyperparameter tuning, which led to a stable and controlled process of gentle polishing. The GRPO-refined model shows high real-life performance, as it shows an accuracy of 91% on unseen data used for testing. It has a high recall of 0.85 on the SELECTED class with a perfect precision of 1.0, which highlights its high reliability for identifying qualified applicants. These findings demonstrate that an appropriately structured two-step fine-tuning pipeline can effectively be used to transfer a small language model into human-like candidate evaluation, surpassing the shortcomings of both traditional ATS systems and unrefined uses of reinforcement learning.