Может ли Large Language Model предсказать текучесть кадров?
Аннотация.
Увольнение сотрудников является критической проблемой, с которой сталкиваются организации, со значительными издержками, связанными с текучестью кадров и потерей ценных талантов. Традиционные методы прогнозирования текучести кадров часто полагаются на статистические методы, которые, хотя и полезны, с трудом отражают сложность современной рабочей силы. Недавние достижения в области machine learning предоставили более точные, масштабируемые решения, позволяющие организациям анализировать различные точки данных и прогнозировать текучесть кадров с большей точностью. Однако появление LLMs открыло новые возможности в управлении человеческими ресурсами, предложив способность интерпретировать контекстную информацию из коммуникаций сотрудников и обнаруживать тонкие сигналы, связанные с текучестью кадров.
В данной статье мы используем набор данных IBM HR Analytics Employee Attrition для оценки эффективности дообученной модели GPT-3.5 по сравнению с традиционными классификаторами machine learning, включая Logistic Regression, k-Nearest Neighbors (KNN), Support Vector Machine (SVM), Decision Tree, Random Forest, AdaBoost и XGBoost. Наше исследование сосредоточено на оценке предсказательной силы, интерпретируемости и применимости в реальных условиях каждой модели. В то время как традиционные модели предлагают простоту использования и прозрачность, LLM обладают потенциалом для выявления более тонких закономерностей в поведении сотрудников. Посредством нашего анализа мы стремимся предоставить практические выводы для организаций, стремящихся улучшить свои стратегии удержания персонала с помощью продвинутых инструментов predictive modeling.
Наши результаты показывают, что дообученная большая языковая модель (LLM) GPT-3.5 превосходит традиционные подходы machine learning по точности прогнозирования, демонстрируя впечатляющую точность (precision) 0.91, полноту (recall) 0.94 и F1-меру 0.92. В то же время лучшая традиционная модель, метод опорных векторов (SVM), достигла F1-меры 0.82, в то время как ансамблевые методы, такие как Random Forest и XGBoost, показали F1-меру 0.80. Эти результаты подчеркивают способность GPT-3.5 выявлять сложные закономерности в поведении сотрудников и рисках оттока, обеспечивая повышенную интерпретируемость за счет обнаружения тонких лингвистических маркеров и повторяющихся тем. Это демонстрирует потенциал интеграции LLM в стратегии HR для значительного улучшения прогнозирования и принятия решений в области удержания персонала.
Ключевые слова:
Business, LLM, Machine learning1. Введение
Сотрудники являются бесценными активами и инвестициями для компаний любого размера, от малого бизнеса до крупных транснациональных корпораций. Например, средняя стоимость одного собеседования в Швейцарии составляет 400 швейцарских франков, а средняя стоимость найма квалифицированного сотрудника может достигать 9 738 швейцарских франков (Blatter et al., 2016). Хотя процессы подбора персонала предназначены для выбора наиболее подходящих кандидатов, они не всегда гарантируют успешные результаты. Текучесть кадров уже давно является проблемой для менеджеров по управлению человеческими ресурсами и работодателей, особенно в быстро растущих экономиках (Iqbal, 2010; Alao and Adeyemo, 2013).
Увольнение сотрудников — это многоаспектная проблема, на которую влияют различные факторы, отражающие эффективность работы организации. К частым причинам относятся отсутствие возможностей для карьерного роста, неадекватная компенсация и плохой баланс между работой и личной жизнью, что в совокупности ведет к неудовлетворенности и выгоранию (Latha, 2013; Alao and Adeyemo, 2013). Кроме того, ошибки управления, несоответствие между должностными обязанностями и навыками сотрудника, а также культурные разрывы внутри организации могут дополнительно усуграблять текучесть кадров (Pallathadka et al., 2022). Сотрудники также могут уходить из-за более выгодных предложений о работе, отсутствия признания, неразрешенных конфликтов на рабочем месте или ощущения ненадежности занимаемой должности (Raza et al., 2022; Burgard et al., 2009). Понимание этих факторов имеет решающее значение для организаций, стремящихся улучшить стратегии удержания персонала.
Традиционные подходы к прогнозированию оттока сотрудников обычно включают анализ исторических данных с использованием статистических методов для выявления закономерностей и тенденций. Эти методы часто требуют ручного анализа данных, такого как изучение выходных интервью, опросов сотрудников и демографической информации, чтобы определить общие причины увольнения. HR-команды склонны фокусироваться на ключевых показателях, таких как стаж работы, результаты аттестаций и рост заработной платы, для оценки риска текучести кадров. Однако эти методы могут быть ограничены их опорой на ретроспективные данные и могут испытывать трудности с отражением сложностей динамики оттока в более крупных и разнообразных коллективах. Они также могут сталкиваться с проблемами, связанными с масштабируемостью и точностью. Несмотря на эти ограничения, традиционные методы служили фундаментальными инструментами для понимания тенденций в рабочей силе и формирования стратегий удержания до появления более продвинутых методов machine learning.
Для решения этих задач машинное обучение (ML) приобрело популярность как более мощный метод прогнозирования оттока сотрудников. Модели ML могут анализировать огромные объемы данных, выявлять сложные закономерности и делать более точные прогнозы по сравнению с традиционными подходами. Включая разнообразные точки данных, такие как уровень вовлеченности сотрудников, показатели работы, демографические характеристики и внешние факторы вроде рыночных тенденций, модели ML предлагают более полное представление о рисках оттока. Эти модели, включая алгоритмы классификации, такие как логистическая регрессия, деревья решений, случайные леса, и передовые методы вроде XGBoost и нейронных сетей, способны непрерывно обучаться на новых данных, со временем повышая эффективность прогнозирования.
Кроме того, ML-based системы способны обрабатывать крупномасштабные данные в реальном времени, что делает их altamente адаптируемыми к динамичным условиям рабочей среды. Эти системы также могут обнаруживать ранние предупреждающие признаки attrition, позволяя организациям вмешиваться проактивно. Используя эти predictive capabilities, компании могут предпринимать целевые действия, такие как персонализация программ развития сотрудников, корректировка структур компенсаций или решение проблем снижения вовлеченности. Таким образом, ML предлагает ориентированное на будущее решение на основе данных, которое позволяет организациям снижать employee turnover и сохранять конкурентное преимущество в talent retention.
В последнее время large language models (LLMs) стали мощными инструментами в различных областях, от обработки естественного языка (Achiam et al., 2023; Team et al., 2023; Touvron et al., 2023) до научных исследований (Chen et al., 2023b; Ding et al., 2023; Chen et al., 2023a), и их применения теперь распространяются на управление человеческими ресурсами. LLMs позволяют организациям выйти за рамки числовых данных и использовать контекстуальное богатство, присутствующее в коммуникациях сотрудников. Эти модели могут анализировать тонкие лингвистические паттерны, изменения тона или повторяющиеся темы в коммуникациях сотрудников для выявления потенциальных рисков увольнения. Более глубокое понимание настроений сотрудников позволяет специалистам по HR принимать более обоснованные решения относительно стратегий удержания. По мере того как LLMs продолжают развиваться, их интеграция в модели прогнозирования оттока сотрудников открывает потенциал для значительного повышения точности, масштабируемости и интерпретируемости, тем самым преобразуя практики удержания персонала.
Несмотря на преимущества LLM, их эффективность в прогнозировании текучести кадров все еще находится под вопросом. В данной работе мы используем широко известный набор данных IBM HR Analytics Employee Attrition для сравнения производительности дообученной модели GPT-3.5 с классическими моделями machine learning, включая Logistic Regression, k-Nearest Neighbors (KNN), Support Vector Machine (SVM), Decision Tree, Random Forest, AdaBoost и XGBoost.
Наша цель — оценить предсказательные возможности LLM по сравнению с этими традиционными классификаторами, уделяя особое внимание точности, интерпретируемости и применимости в реальных условиях. Хотя традиционные модели долгое время пользовались предпочтением для прогнозирования текучести кадров из-за их простости и интерпретируемости, LLM обладают потенциалом для выявления более сложных паттернов в поведении сотрудников. Данное исследование направлено на определение того, могут ли дообученные LLM превзойти традиционные методы или дополнить их, а также предоставить уникальное понимание факторов, влияющих на текучесть кадров.
Оценивая сильные и слабые стороны каждой модели, мы стремимся предложить практические рекомендации для организаций, стремящихся интегрировать передовые инструменты прогнозирования в свои стратегии управления человеческими ресурсами. Кроме того, мы исследуем, как интерпретируемость моделей, имеющая решающее значение в HR-решениях, различается в зависимости от подхода, особенно с точки зрения прозрачности и принятия решений между LLM и более традиционными методами machine learning..
Остальная часть этой статьи организована следующим образом. В разделе 2 представлен обзор существующих исследований по прогнозированию оттока сотрудников с акцентом как на традиционные модели machine learning, так и на недавние применения LLM. Мы обсуждаем сильные стороны и ограничения различных подходов и выделяем пробел в исследованиях, который данное исследование призвано восполнить. В разделе 3 описывается набор данных IBM HR Analytics Employee Attrition, использованный в нашем анализе. Мы обрисовываем ключевые атрибуты набора данных, включая демографические данные сотрудников, удовлетворенность работой и показатели производительности. Мы также обсуждаем этапы предварительной обработки, такие как очистка данных, обработка пропущенных значений, кодирование категориальных признаков и нормализация данных для подготовки их к обучению моделей. В разделе 4 подробно описана методология, использованная для обучения и оценки как традиционных моделей machine learning, так и дообученной модели GPT-3.5. Мы объясняем процесс настройки параметров, выбор признаков и стратегии обучения моделей. Кроме того, мы описываем метрики оценки, используемые для сравнения производительности моделей, такие как accuracy, precision, recall и F1-score, а также соображения относительно интерпретируемости моделей. В разделе 5 представлены результаты наших экспериментов, в которых сравнивается производительность каждой модели в задаче прогнозирования оттока сотрудников. Мы приводим подробный анализ сильных и слабых сторон как традиционных моделей machine learning, так и LLM с точки зрения точности прогнозирования, интерпретируемости и практической применимости при принятии решений в сфере HR.
Основные результаты данной работы заключаются в следующем:
- ⟦• подробные этапы обработки набора данных IBM HR Analytics Employee Attrition для обучения ML и LLM.
- ⟦• новаторская работа по тонкой настройке модели GPT для задачи оттока персонала.
- ⟦• практические выводы для подхода ML в HR-менеджменте.
2. Предпосылки
В этом разделе представлена справочная информация об оттоке персонала и LLM.
2.1. Текучесть кадров
Текучесть кадров относится к добровольному или принудительному уходу сотрудников из организации и является серьезной проблемой для бизнеса из-за влияния на операционную эффективность и затраты. Высокие показатели текучести могут нарушить производительность, увеличить финансовое бремя найма и обучения новых сотрудников, а также негативно повлиять на общий моральный дух коллектива. Текучесть можно разделить на два основных типа: добровольная текучесть, при которой сотрудники уходят по собственному желанию (например, из-за лучших возможностей или неудовлетворенности текущей ролью), и принудительная текучесть, при которой работодатель расторгает трудовой договор с сотрудником из-за таких факторов, как низкая эффективность работы или организационная реструктуризация..
Несколько факторов способствуют оттоку кадров. К ним могут относиться личные причины, такие как отсутствие возможностей для карьерного роста, неадекватная компенсация, плохой баланс между работой и личной жизнью и ограниченное признание заслуг. Организационные факторы, такие как неудовлетворительное руководство, несоответствие между должностными обязанностями и навыками сотрудников, токсичная рабочая культура и воспринимаемая неуверенность в завтрашнем дне, также являются основными драйверами текучести кадров (Pallathadka et al., 2022). Внешние факторы, включая изменения на рынке труда или более выгодные предложения от конкурирующих фирм, еще больше усугубляют эту проблему (Raza et al., 2022).
Для организаций последствия текучести кадров выходят за рамки затрат на замену сотрудников. Сам процесс найма обходится дорого и отнимает много времени, так как включает в себя рекрутинг, собеседования, адаптацию и обучение. Кроме того, потеря опытных сотрудников может привести к временному снижению организационных знаний и производительности, особенно если уходящие сотрудники занимают критически важные должности. Согласно отчету Общества по управлению человеческими ресурсами за 2016 год, средняя стоимость найма нового сотрудника превышает 4000 долларов, а средний годовой показатель текучести кадров в США оценивается в 19% (Blatter et al., 2016). Следовательно, прогнозирование и предотвращение оттока персонала стали стратегическим приоритетом для HR-отделов, что стимулирует разработку более совершенных аналитических методов для выявления сотрудников из группы риска и реализации целевых стратегий удержания.
2.2. Подход к прогнозированию текучести кадров
Традиционные методы прогнозирования текучести кадров включают ручной анализ исторических данных, таких как выходные интервью и аттестации сотрудников. Однако эти подходы ограничены по охвату, точности и масштабируемости, особенно для крупных организаций со сложной динамикой. Machine learning (ML) стало более продвинутым решением, предлагающим более высокую точность и адаптивность. Алгоритмы ML могут анализировать большие объемы данных, выявляя сложные, нелинейные взаимосвязи между характеристиками сотрудников и рисками увольнения. Распространенные методы ML для прогнозирования текучести включают логистическую регрессию, деревья решений, случайные леса и более продвинутые методы, такие как машины опорных векторов и XGBoost. Эти модели обрабатывают различные структурированные точки данных, включая демографические показатели сотрудников, стаж работы и оценки эффективности, а также учитывают внешние факторы, такие как рыночные тенденции и экономические условия. Такой комплексный подход позволяет организациям создавать более надежные предиктивные модели. Ряд исследований (Raza et al., 2022; Alduayj and Rajpoot, 2018; Jain et al., 2020) продемонстрировали эффективность различных подходов ML при прогнозировании текучести кадров..
2.3. Большие языковые модели
LLM произвели революцию в области обработки естественного языка с момента появления моделей на основе transformers, представленных Васвани и соавторами (Vaswani, 2017) в 2017 году. Эти модели используют механизмы self-attention для эффективного учета долгосрочных зависимостей в тексте, превосходя традиционные рекуррентные нейронные сети. Архитектура multi-head attention позволяет параллельно обрабатывать несколько паттернов внимания, решая проблемы затухающих градиентов и повышая масштабируемость. LLM, построенные на принципах transformers, обучаются на огромных наборах данных для моделирования сложных языковых распределений. Их исключительная эффективность в NLP (Achiam et al., 2023; Chen et al., 2024a) и научных приложениях (Chen et al., 2024b; Chen et al., 2022) привела к расширению исследований их потенциала в управлении персоналом, включая применение в управленческом консалтинге (Mohan, 2024), HR чат-ботах (Singh, 2023) и развитии сотрудников (Budhwar et al., 2023)⟧.
3. Подготовка набора данных
В этом разделе представлен набор данных, используемый в данной работе. Мы провели различные анализы, чтобы представить данные с разных точек зрения, и подробно описали наши методы обработки данных для обучения и тестирования моделей машинного обучения..
3.1. Введение в набор данных
Набор данных IBM HR Analytics Employee Attrition & Performance представляет собой хорошо структурированную коллекцию данных о сотрудниках, направленную на выявление факторов, способствующих текучести кадров. Он содержит подробную информацию о 1470 сотрудниках с 35 различными атрибутами, которые предлагают всесторонний обзор различных аспектов, связанных с эффективностью работы сотрудников, демографическими данными и удовлетворенностью работой. Этот набор данных обеспечивает ценную основу для HR analytics и predictive modeling, что делает его подходящим для задач machine learning, сфокусированных на прогнозировании текучести кадров. Набор данных обладает следующими базовыми характеристиками:
- ⟦• Размер: Набор данных состоит из 1 470 строк, каждая из которых представляет отдельного сотрудника, и 35 столбцов, фиксирующих конкретные атрибуты, связанные с демографическими данными сотрудников, должностными обязанностями и показателями эффективности..
- ⟦• Формат: Набор данных представлен в формате CSV, что делает его легкодоступным и совместимым с большинством инструментов обработки данных и библиотек машинного обучения..
- ⟦• Пропущенные значения: Набор данных не содержит пропущенных значений, при этом во всех столбцах насчитывается 1470 ненулевых записей. Это обеспечивает целостность данных и устраняет необходимость в обширной очистке данных, что позволяет использовать более простой конвейер предварительной обработки..
3.2. Основные характеристики
| Category | Features |
|---|---|
| Demographic Information | Age, Gender, Marital Status, Education, Education Field |
| Job-related Attributes | Department, Job Role, Job Level, Job Involvement, Job Satisfaction, Years at Company, Years in Current Role, Years Since Last Promotion, Years with Current Manager |
| Compensation and Benefits | Daily Rate, Hourly Rate, Monthly Income, Monthly Rate, Percent Salary Hike, Stock Option Level |
| Performance Metrics | Performance Rating, Training Times Last Year |
| Work Environment Factors | Business Travel, Environment Satisfaction, Work-Life Balance, Overtime |
| Other Relevant Information | Distance from Home, Relationship Satisfaction, Total Working Years, Number of Companies Worked |
| Target Variable | Attrition (Yes/No) |
Набор данных IBM HR Analytics Employee Attrition & Performance, представленный в таблице 1, содержит исчерпывающий перечень признаков для анализа факторов, способствующих текучести кадров. Набор данных охватывает широкий спектр информации: от базовых демографических данных (таких как возраст, пол и семейное положение) до детальных характеристик, связанных с работой (включая отдел, должностную роль и показатели стажа). Кроме того, он содержит данные о компенсациях, льготах, показателях эффективности и факторах рабочей среды. Примечательно, что признаки, связанные с удовлетворенностью сотрудников, такие как удовлетворенность работой, баланс между работой и личной жизнью и удовлетворенность отношениями, позволяют получить представление о нематериальных аспектах опыта сотрудников. Другие переменные, такие как расстояние от дома и количество компаний, в которых сотрудник работал ранее, могут указывать на менее очевидные факторы влияния на текучесть кадров. Поскольку текучесть является бинарной целевой переменной, данный набор признаков позволяет проводить надежный анализ и предиктивное моделирование, помогая HR-специалистам лучше понимать и контролировать текучесть персонала..
Таблица 2 представляет описательную статистику для ключевых числовых признаков в наборе данных, которые могут влиять на отток сотрудников. Каждый признак описывается его количеством, средним значением, стандартным отклонением и перцентилями (минимальное, 25-е, 50-е, 75-е и максимальное значения).).
Например, средний возраст сотрудника в наборе данных составляет примерно 36,92 года при стандартном отклонении 9,14 года, в то время как месячный доход имеет среднее значение 6 502,93 доллара США с диапазоном от 1 009 до 19 999 долларов США. Эти статистические данные дают представление о распределении набора данных и послужат основой для нашего прогнозного моделирования закономерностей оттока кадров.
| Feature | Count | Mean | Std | Min | 25% | 50% | 75% | Max |
|---|---|---|---|---|---|---|---|---|
| Age | 1470 | 36.92 | 9.14 | 18 | 30 | 36 | 43 | 60 |
| Daily Rate | 1470 | 802.49 | 403.51 | 102 | 465 | 802 | 1157 | 1499 |
| Distance from Home | 1470 | 9.19 | 8.11 | 1 | 2 | 7 | 14 | 29 |
| Education | 1470 | 2.91 | 1.02 | 1 | 2 | 3 | 4 | 5 |
| Monthly Income | 1470 | 6502.93 | 4707.96 | 1009 | 2911 | 4919 | 8379 | 19999 |
| Number of Companies Worked | 1470 | 2.69 | 2.50 | 0 | 1 | 2 | 4 | 9 |
| Percent Salary Hike | 1470 | 15.21 | 3.66 | 11 | 12 | 14 | 18 | 25 |
| Total Working Years | 1470 | 11.28 | 7.78 | 0 | 6 | 10 | 15 | 40 |
| Training Times Last Year | 1470 | 2.80 | 1.29 | 0 | 2 | 3 | 3 | 6 |
| Years at Company | 1470 | 7.01 | 6.13 | 0 | 3 | 5 | 9 | 40 |
| Years in Current Role | 1470 | 4.23 | 3.62 | 0 | 2 | 3 | 7 | 18 |
| Years Since Last Promotion | 1470 | 2.19 | 3.22 | 0 | 0 | 1 | 3 | 15 |
| Years with Current Manager | 1470 | 4.12 | 3.57 | 0 | 2 | 3 | 7 | 17 |
3.3. Очистка и сокращение данных
Чтобы убедиться, что набор данных является чистым и релевантным для нашего анализа, мы выполнили несколько этапов предварительной обработки. Сначала был загружен набор данных, содержащий различные атрибуты сотрудников. Некоторые из этих атрибутов были избыточными или бесполезными для прогнозирования оттока кадров.
Мы уменьшили набор данных, удалив следующие признаки:
- ⟦• EmployeeCount: Этот признак имеет постоянное значение для всех записей и, следовательно, не предоставляет полезной информации для predictive modeling.
- ⟦• StandardHours: Аналогично EmployeeCount, этот столбец содержит одинаковое значение для каждого сотрудника и не вносит вклад в вариативность, необходимую для задач классификации.
- ⟦• Over18: Этот признак содержит только одно значение («Да») во всех экземплярах, что не обеспечивает никакого полезного различия между сотрудниками.
- ⟦• EmployeeNumber: Это просто идентификатор для каждого сотрудника и он не влияет на результат прогнозирования оттока.
Удаляя эти признаки, мы устранили ненужные столбцы и снизили размерность данных, что может улучшить производительность модели за счет концентрации внимания только на релевантных атрибутах. Этот шаг имеет решающее значение для предотвращения переобучения модели на нерелевантных или избыточных данных.
3.4. Анализ данных
Баланс данных. На рисунке 1 показано распределение оттока персонала в наборе данных после очистки и сокращения. Большинство сотрудников, а именно 83,9%, не покинули компанию, в то время как уволились лишь 16,1%. Такой существенный дисбаланс в классах оттока подчеркивает необходимость учета дисбаланса классов при построении прогнозных моделей.
В частности, методы вродеoversampling миноритарного класса или применение специализированных алгоритмов для несбалансированных данных могут потребоваться для того, чтобы модель не развила смещение в сторону прогнозирования мажоритарного класса. Более того, дисбаланс предполагает, чтоemployee attrition является относительно редким явлением, что может указывать либо на успешные стратегии workforce retention, либо на естественные тенденции внутри коллектива..
Частотное распределение. Чтобы лучше понять распределение числовых признаков в наборе данных, мы построили гистограмму, которая дает представление о распределении данных. Гистограмма визуализирует частотное распределение признака, разделяя данные на интервалы. Высота каждого интервала представляет количество наблюдений, попадающих в соответствующий диапазон значений. Это помогает выявить асимметрию и общий разброс данных. На рисунке 2 перечислили признаки, имеющие проблему асимметрии.
3.5. Обработка несбалансированных данных
На Figure 2 illustrated features that have skewed issues. In our dataset, several numerical features exhibited right-skewed distributions, where the majority of the data points are concentrated on the lower end of the scale, with a long tail extending to the right. Right-skewed distributions are common when certain features, such as income or years of experience, have a small number of very large values that disproportionately influence the distribution.
Скошенные данные могут создавать серьезные проблемы при прогностическом моделировании. Многие алгоритмы machine learning, особенно линейные модели и алгоритмы, предполагающие нормальное распределение данных, могут быть чувствительны к асимметрии. Высоко асимметричные признаки могут приводить к тому, что модели становятся менее точными или смещенными в сторону скошенных данных. Устраняя асимметрию, мы обеспечиваем более эффективную работу наших моделей за счет обучения на лучше распределенных данных, что приводит к более точным прогнозам.
Для устранения правосторонней асимметрии мы применили логарифмическое преобразование к затронутым признакам. Для каждого числового признака с асимметрией больше 0,5 мы выполнили логарифмическое преобразование с использованием функции . Этот метод помогает уменьшить асимметрию путем сжатия больших значений и распределения меньших, приближая распределение к нормальному. После применения преобразования признаки демонстрируют более симметричное распределение, что улучшает способность модели к генерализации и точным предсказаниям.
3.6. Инженерия признаков
Для повышения предсказательной способности модели и упрощения датасета было выполнено feature engineering путем создания двух новых составных признаков: WorkExperience и OverallSatisfaction. Эти признаки инкапсулируют важные аспекты опыта работы и удовлетворенности сотрудников, уменьшая избыточность в датасете при сохранении необходимой информации для моделирования.
Признак WorkExperience был получен путем усреднения нескольких ключевых переменных, связанных с опытом, включая TotalWorkingYears, YearsAtCompany, YearsInCurrentRole, YearsSinceLastPromotion и YearsWithCurrManager. Путем объединения этих столбцов была сформирована единая метрика, представляющая общий опыт работы сотрудника, что позволило упростить данные без потери критически важной информации.
Аналогично, признак ⟦BROKEN_K0⟧ был создан путем усреднения столбцов ⟦BROKEN_K1⟧, ⟦BROKEN_K2⟧, ⟦BROKEN_K3⟧ и ⟦BROKEN_K4⟧. Этот новый признак обеспечивает комплексную меру удовлетворенности сотрудника как своей работой, так и рабочей средой.
После генерации этих новых признаков исходные столбцы, использованные для их создания, были удалены из набора данных. Этот шаг помог уменьшить избыточность и повысить эффективность набора данных, гарантируя, что ключевая информация, необходимая для predictive modeling, сохранена в более упорядоченном виде.
3.7. Категориальное кодирование
Для подготовки категориальных признаков к моделированию мы применили кодирование меток (label encoding) ко всем категориальным переменным. Кодирование меток преобразует категориальные значения в числовые представления, которые требуются большинству алгоритмов machine learning. В частности, мы определили категориальные столбцы в наборе данных и использовали класс LabelEncoder из библиотеки sklearn для кодирования каждого столбца. Этот процесс гарантировал, что такие категориальные признаки, как Department и JobRole, были преобразованы в числовые значения, что позволило модели machine learning эффективно их обрабатывать..
3.8. Разделение на обучающую и тестовую выборки
После подготовки данных мы разделили их на обучающий и тестовый наборы для оценки производительности модели. Используя функцию train_test_split из sklearn, мы разделили датасет, выделив 80% данных для обучения и 20% для тестирования. Целевая переменная Attrition была стратифицирована, чтобы гарантировать сохранение одинакового распределения классов в обучающем и тестовом наборах.
Полученное разделение содержит 1176 примеров для обучения и 294 примера для тестирования, как показано в следующем выводе:
- ⟦• X_train shape: (1176, 24)
- ⟦• X_test shape: (294, 24)
- ⟦• y_train shape: (1176,)
- ⟦• y_test shape: (294,)
Такое разделение гарантирует, что модель может быть обучена на большей части данных и в то же время оценена на отдельном, ранее не видимом тестовом наборе, что обеспечивает точную оценку способности модели к генерализации.
3.9. Масштабирование признаков
Перед обучением нашей модели машинного обучения было необходимо стандартизировать признаки, чтобы убедиться, что они находятся в одном масштабе. Стандартизация преобразует признаки так, чтобы их среднее значение было равно нулю, а стандартное отклонение — единице. Этот процесс особенно важен для моделей, чувствительных к величине входных признаков, таких как логистическая регрессия и машины опорных векторов. Стандартизируя признаки, мы гарантируем, что модель не будет уделять чрезмерного внимания признакам с большими диапазонами или единицами измерения, что могло бы негативно повлиять на производительность модели.
Данные для обучения использовались для вычисления коэффициентов масштабирования (среднее значение и стандартное отклонение), и эти коэффициенты последовательно применялись как к обучающему, так и к тестовому наборам данных. Это гарантирует, что модель хорошо обобщает ранее не виденные данные, сохраняя при этом согласованность между наборами данных..
3.10. Обработка несбалансированных данных
Наш набор данных продемонстрировал значительный дисбаланс классов, при котором количество случаев отсутствия текучести кадров значительно превышало количество случаев увольнений. Несбалансированные наборы данных могут приводить к появлению смещенных моделей, которые показывают хорошие результаты на мажоритарном классе, но плохо справляются с миноритарным классом, которым в нашем случае является текучесть кадров. Для устранения этого дисбаланса мы применили метод синтетической избыточной выборки меньшинства (SMOTE).
SMOTE генерирует синтетические выборки миноритарного класса путем интерполяции между существующими образцами, эффективно увеличивая количество экземпляров миноритарного класса и балансируя распределение классов. До применения SMOTE обучающий набор данных содержал 986 образцов без увольнений и 190 образцов с увольнениями. После оверсэмплинга набор данных был сбалансирован: по 986 образцов для каждого класса..
Этот процесс балансировки гарантирует, что модель имеет равный доступ к обоим классам, что повышает ее способность точно обнаруживать случаи увольнения сотрудников и снижает предвзятость по отношению к мажоритарному классу. В результате модель лучше подготовлена к обобщению в реальных сценариях, где оба класса имеют важное значение..
4. Эксперимент
В этом разделе подробно описаны эксперименты, включая выбор моделей и сравнение результатов.
4.1. Классические модели, используемые для прогнозирования текучести кадров
Для создания надежной и точной прогностической модели текучести кадров мы использовали разнообразный набор алгоритмов машинного обучения. Каждая из этих моделей обладает уникальными преимуществами и подходами к классификации, что позволяет нам сравнивать их эффективность и определять наиболее подходящий метод для нашего набора данных. В нашем анализе были использованы следующие модели:
Логистическая регрессия оценивает вероятности классов с использованием логистической функции для входных признаков. Данная реализация sklearn использует параметры по умолчанию: L2 регуляризацию с C=1.0, решатель ’lbfgs’ и отсутствие штрафа. Она обеспечивает интерпретируемые прогнозы за счет выявления вклада признаков. Модель обучается на X_train и y_train, а затем используется для прогнозирования результатов как на обучающих, так и на тестовых данных..
K-Nearest Neighbors (KNN) — это непараметрический алгоритм, классифицирующий экземпляры на основе близости к соседям в пространстве признаков. Мы использовали 10 соседей, равномерные веса и расстояние Минковского (p=2, эквивалентно евклидову). Выбор алгоритма осуществляется автоматически, с размером листа 50 для оптимизированного поиска по дереву. KNN хорошо работает для нелинейных границ в небольших наборах данных, но чувствителен к метрикам расстояния и значениям k..
Метод опорных векторов (Support Vector Machine, SVM) находит оптимальную гиперплоскость для разделения классов в пространстве признаков. В данной реализации используется RBF ядро с C=200, gamma=’scale’ и степенью 5. В нем применяется решающая функция ’один против всех’, допускается погрешность 0.1 и отсутствуют оценки вероятностей. SVM превосходно работает в пространствах высокой размерности и способна выполнять нелинейную классификацию с помощью ядерных функций. Она особенно эффективна для наборов данных с четким разделением классов.
Классификатор на основе дерева решений создает древовидную структуру путем разделения данных на основе значения признака. Деревья решений просты для интерпретации и визуализации, но могут быть склонны к переобучению, особенно на небольших наборах данных. Мы ограничили глубину дерева 5 уровнями, предотвратив переобучение.
Random Forest объединяет множество деревьев решений для повышения точности и уменьшения переобучения. Наша реализация использует 200 деревьев, каждое с максимальной глуиной 5. В ней применяется параметр max_features равный ’sqrt’, выбирающий квадратный корень от общего числа признаков для каждого разделения. Усредняя предсказания различных деревьев, Random Forest справляется со сложными взаимосвязями признаков и превосходит отдельные деревья решений.
AdaBoost Classifier объединяет слабые классификаторы для создания сильного, уделяя особое внимание неверно классифицированным объектам. Наша реализация использует 150 оценщиков со скоростью обучения 0.01. В ней применяется алгоритм ’SAMME’ для многоклассовых задач. AdaBoost итеративно корректирует веса объектов, повышая общую точность и уменьшая смещение. Он эффективен для умеренно сложных наборов данных, обеспечивая баланс между слабыми классификаторами для формирования надежного ансамбля.
XGBoost (Extreme Gradient Boosting) — это продвинутая реализация градиентного бустинга. Наша конфигурация использует бинарную логистическую регрессию для классификации со скоростью обучения 0.01 и 350 оценщиками. Глубина дерева ограничена 3, а также используется параметр max_features со значением 'sqrt'. XGBoost сочетает в себе регуляризацию и оптимизированные вычисления, превосходя другие методы по скорости и точности при работе с большими несбалансированными наборами данных. Его гибкость позволяет выполнять fine-tuning для решения различных задач машинного обучения..
4.2. Модель GPT для прогнозирования текучести кадров
GPT-3.5 Turbo — это large language model, которая использует transformer архитектуру для выполнения задач прогнозирования. Хотя она преимущественно используется в задачах обработки естественного языка, GPT-3.5 может подвергаться fine-tuning для различных задач прогнозирования и классификации, включая прогнозирование оттока персонала. Мы следовали процессу, описанному в (Liga and Robaldo, 2023), для fine-tuning GPT-3.5 turbo с использованием пар промпт-ответ, созданных на основе обучающего набора..
Подготовка данных. Мы сформировали данные в формате JSONL, где каждая строка представляет собой пару промпт-ответ, на которой модель обучается.
{
"prompt": "Analyze the employee information and
predict employee turnover: {Employee_Info}",
"completion": "{Attrition_result}"
}
Тонкая настройка GPT-3.5 Turbo. Мы использовали библиотеку и API OpenAI для точной настройки модели GPT на наших данных. Мы установили равным 50 для сокращения расходов, поскольку наши прогнозируемые результаты представляют собой или .
Инференс. Для генерации предсказаний мы передали тестовый набор в fine-tuned GPT модель, используя ее fine-tuned ID.
4.3. Метрики оценки
Для оценки эффективности прогностических моделей мы использовали такие метрики классификации, как precision, recall и F1-score, как определено в уравнениях 1, 2 и 3⟧:
| (1) |
| (2) |
| (3) |
Эти метрики позволяют оценить соотношение между точностью и полнотой и помогают определить модель с наилучшей производительностью для прогнозирования оттока персонала.
4.4. Результаты
Сравнительный анализ различных моделей machine learning для данной задачи классификации выявил интересные закономерности, приведенные в Таблице 3. Доработанная модель GPT-3.5 продемонстрировала наилучшие результаты, достигнув исключительных метрик: precision 0.91, recall 0.94 и F1-score 0.92. Эта производительность значительно превзошла традиционные алгоритмы machine learning, подчеркивая потенциал LLM в задачах классификации при правильном fine-tuning. Среди традиционных моделей Support Vector Machine (SVM) продемонстрировала самую высокую эффективность с F1-score 0.82, за ней с небольшим отрывом следуют ансамблевые методы, такие как Random Forest и XGBoost, обе достигнувшие F1-score 0.80. Logistic Regression и AdaBoost показали сопоставимую эффективность с F1-score 0.78 и 0.79 соответственно, в то время как K-Nearest Neighbors (KNN) показала наихудший результат с F1-score 0.71. Эти результаты подчеркивают надежность ансамблевых методов и особую пригодность SVM для данного датасета. Заметное превосходство доработанной модели GPT-3.5 открывает перспективные возможности для использования продвинутых языковых моделей в задачах классификации, что потенциально может произвести революцию в подходах к аналогичным проблемам в этой области.
| Model | Precision | Recall | F1-score |
|---|---|---|---|
| Logistic Regression | 0.84 | 0.76 | 0.78 |
| KNN | 0.80 | 0.66 | 0.71 |
| SVM | 0.81 | 0.83 | 0.82 |
| Decision Tree | 0.83 | 0.78 | 0.80 |
| Random Forest | 0.82 | 0.80 | 0.80 |
| AdaBoost | 0.82 | 0.77 | 0.79 |
| XGBoost | 0.80 | 0.80 | 0.80 |
| GPT 3.5 Fine-tuned | 0.91 | 0.94 | 0.92 |
5. Заключение
В этом исследовании мы изучили производительность различных machine learning моделей для задачи предсказания employee attrition, сравнив традиционные алгоритмы с дообученной моделью GPT-3.5. Наши эксперименты показали, что дообученная модель GPT-3.5 значительно превзошла традиционные machine learning методы, достигнув наибольших показателей precision, recall и F1-score. Этот результат подчеркивает потенциал large language models в задачах классификации, особенно при дообучении на специфичных для предметной области данных..
Среди традиционных моделей метод опорных векторов (SVM) продемонстрировал наилучшую производительность, за которым с небольшим отрывом следуют ансамблевые методы, такие как Random Forest и XGBoost. Эти модели также показали конкурентоспособные результаты, подтвердив свою эффективность для данного типа задач классификации. Однако более простые модели, такие как Logistic Regression и K-Nearest Neighbors (KNN), продемонстрировали более низкую прогнозирующую способность, особенно в учете сложности данных.
Превосходные показатели GPT-3.5 подчеркивают возрастающую роль, которую передовые языковые модели могут играть в решении задач классификации за пределами обработки естественного языка. По мере развития методов machine learning интеграция large language models в конвейеры predictive analytics обладает огромным потенциалом для улучшения процессов принятия решений в таких областях, как управление человеческими ресурсами.
В будущих исследованиях можно дополнительно изучить влияние различных подходов к отбору признаков на производительность моделей для комплексного понимания факторов, влияющих на эффективность предсказаний. Кроме того, исследование интерпретируемости этих моделей в реальных приложениях будет иметь решающее значение для их более широкого внедрения в бизнес- и операционных средах. В целом, данное исследование демонстрирует преимущества использования LLM в задачах классификации и указывает на перспективные направления для будущих исследований в этой области.
Список литературы
- ⟦Achiam et al. (2023) Josh Achiam, Steven Adler, Sandhini Agarwal, Lama Ahmad, Ilge Akkaya, Florencia Leoni Aleman, Diogo Almeida, Janko Altenschmidt, Sam Altman, Shyamal Anadkat, et al. 2023. Gpt-4 technical report. arXiv preprint arXiv:2303.08774 (2023).
- ⟦Alao and Adeyemo (2013) DABA Alao and AB Adeyemo. 2013. Analyzing employee attrition using decision tree algorithms. Computing, Information Systems, Development Informatics and Allied Research Journal 4, 1 (2013), 17–28.
- ⟦Alduayj and Rajpoot (2018) Сара С Алдуайж и Кашиф Раджпут. 2018. Прогнозирование текучести кадров с использованием machine learning. В 2018 international conference on innovations in information technology (iit). IEEE, 93–98.
- ⟦Blatter et al. (2016) Марк Блаттер, Самуэль Мюлеманн, Самуэль Шенкер и Стефан К. Вольтер. 2016. Затраты на найм квалифицированных работников и предложение корпоративного обучения. Оксфордские экономические бумаги 68, 1 (2016), 238–257.
- ⟦Budhwar et al. (2023) Паван Будхвар, Соумьядеб Чоудхури, Джеффри Вуд, Херман Агинис, Грег Д. Бамбер, Хосе Р. Бельтран, Пол Бозели, Фанг Ли Кук, Стефани Декер, Анджело ДеНиси и др. 2023. Управление человеческими ресурсами в эпоху генеративного искусственного интеллекта: перспективы и направления исследований ChatGPT. Журнал управления человеческими ресурсами 33, 3 (2023), 606–659.
- ⟦Burgard et al. (2009) Сара А Бюргард, Дженни Е Бранд и Джеймс С Хаус. 2009. Воспринимаемая ненадежность работы и здоровье работников в Соединенных Штатах. Социальная наука и медицина 69, 5 (2009), 777–785.
- ⟦Chen et al. (2024a) Le Chen, Nesreen K Ahmed, Akash Dutta, Arijit Bhattacharjee, Sixing Yu, Quazi Ishtiaque Mahmud, Waqwoya Abebe, Hung Phan, Aishwarya Sarkar, Branden Butler, et al. 2024a. The Landscape and Challenges of HPC Research and LLMs. CoRR (2024).
- ⟦Chen et al. (2024b) Le Chen, Arijit Bhattacharjee, Nesreen Ahmed, Niranjan Hasabnis, Gal Oren, Vy Vo, and Ali Jannesari. 2024b. Ompgpt: A generative pre-trained transformer model for openmp. In European Conference on Parallel Processing. Springer, 121–134.
- ⟦Chen et al. (2023a) Ле Чен, Сяньчжун Дин, Мурали Эмани, Тристан Вандербрюгген, Пей-Хун Линь и Чуньхуа Ляо. 2023a. Обнаружение состояний гонки данных с использованием LLM. В материалах семинаров SC’23 Международной конференции по высокопроизводительным вычислениям, сетям, хранению данных и анализу. 215–223.
- ⟦Chen et al. (2023b) Le Chen, Pei-Hung Lin, Tristan Vanderbruggen, Chunhua Liao, Murali Emani, and Bronis De Supinski. 2023b. Lm4hpc: Towards effective language model application in high-performance computing. In International Workshop on OpenMP. Springer, 18–33.
- ⟦Chen et al. (2022) Ле Чен, Куази Иштиак Махмуд и Али Джаннесари. 2022. Многопредставленческое обучение для обнаружения параллелизма последовательных программ. В сборнике: 2022 IEEE International Parallel and Distributed Processing Symposium Workshops (IPDPSW). IEEE, 295–303.
- Ding et al. (2023) Сианчжун Дин, Ле Чен, Мурали Эмани, Чуньхуа Ляо, Пэй-Хун Линь, Тристан Вандербрюгген, Чжэнь Се, Альберто Серпа и Ван Ду. 2023. Hpc-gpt: Integrating large language model for high-performance computing. In Proceedings of the SC’23 Workshops of The International Conference on High Performance Computing, Network, Storage, and Analysis. 951–960.
- ⟦Iqbal (2010) Adnan Iqbal. 2010. Employee turnover: Causes, consequences and retention strategies in the Saudi organizations. The Business Review, Cambridge 16, 2 (2010), 275–281.
- ⟦Jain et al. (2020) Прапхула Кумар Джейн (Praphula Kumar Jain), Мадхур Джейн (Madhur Jain) и Раджендра Памула (Rajendra Pamula). 2020. Объяснение и прогнозирование текучести кадров: подход на основе машинного обучения. SN Applied Sciences 2, 4 (2020), 757.
- ⟦Latha (2013) K Lavanya Latha. 2013. A study on employee attrition and retention in manufacturing industries. BVIMSR’s Journal of Management Research (BJMR) 5, 1 (2013), 1–23.
- ⟦Liga and Robaldo (2023) Давиде Лига и Ливио Робальдо. 2023. Fine-tuning GPT-3 для классификации правовых норм. Computer Law & Security Review 51 (2023), 105864.
- ⟦Mohan (2024) Сай Кришнан Мохан. 2024. Управленческий консалтинг в эпоху искусственного интеллекта и LLM. Management Consulting Journal 7, 1 (2024), 9–24.
- ⟦Pallathadka et al. (2022) Харикумар Паллатхадка, В Хари Лила, Сушант Патил, БХ Рашми, Випин Джаин и Самрат Рэй. 2022. Отток кадров в программных компаниях: причины и меры. Materials Today: Proceedings 51 (2022), 528–531.
- ⟦Raza et al. (2022) Али Раза, Кашиф Мунир, Мубарак Альмутайри, Файзан Юнас и Миан Мухаммад Садик Фарид. 2022. Прогнозирование текучести кадров с использованием подходов machine learning. Applied Sciences 12, 13 (2022), 6424.
- ⟦Singh (2023) Васудха Сингх. 2023. Изучение роли чат-ботов на основе large language model (llm) в сфере человеческих ресурсов. Диссертация на соискание ученой степени Ph. D..
- ⟦Team et al. (2023) Gemini Team, Rohan Anil, Sebastian Borgeaud, Yonghui Wu, Jean-Baptiste Alayrac, Jiahui Yu, Radu Soricut, Johan Schalkwyk, Andrew M Dai, Anja Hauth, et al. 2023. Gemini: a family of highly capable multimodal models. arXiv preprint arXiv:2312.11805 (2023).
- Touvron et al. (2023) Hugo Touvron, Thibaut Lavril, Gautier Izacard, Xavier Martinet, Marie-Anne Lachaux, Timothée Lacroix, Baptiste Rozière, Naman Goyal, Eric Hambro, Faisal Azhar, et al. 2023. Llama: Open and efficient foundation language models. arXiv preprint arXiv:2302.13971 (2023).
- ⟦Vaswani (2017) А. Васвани. 2017. Attention is all you need. Достижения в системах нейронной обработки информации (2017)).