Predicting AI Job Salary Classes Through a Comparative Study of Machine Learning Algorithms
DOI:
https://doi.org/10.30865/jurikom.v12i6.8979Keywords:
Salary Prediction, Artificial Intelligence, Machine Learning, Classification, Feature ImportanceAbstract
The rapid growth of Artificial Intelligence (AI) has brought significant transformation to the global job market, particularly in salary structures across various AI-related professions. This study aims to classify AI job salaries into three categories—Low, Medium, and High—using supervised machine learning algorithms. The dataset, sourced from Kaggle, combines two real-world datasets featuring key attributes such as experience level, job type, education level, technical skills, remote work ratio, and salary in USD. Preprocessing techniques include One-Hot Encoding for categorical data, StandardScaler for normalization, and MultiLabelBinarizer to handle multi-skill entries. Four machine learning models—Logistic Regression, Random Forest, Gradient Boosting, and XGBoost—were trained and evaluated using consistent pipelines, with evaluation metrics including accuracy, precision, recall, and F1-score, applying macro-averaging to address class imbalance. Logistic Regression achieved the highest performance with 85.4% accuracy and 77.6% F1-score, followed by Gradient Boosting with 84.8% accuracy and 76.3% F1-score. High-salary classes were predicted with higher precision and recall than low-salary classes, indicating skewness in class distribution. Feature importance analysis shows that experience, remote work ratio, and key skills such as Python and SQL significantly affect prediction accuracy. This study demonstrates that traditional machine learning methods, when applied with appropriate preprocessing, can effectively support salary classification and labor market analysis in the AI domain.
References
[1] Y. T. Matbouli and S. M. Alghamdi, “Statistical Machine Learning Regression Models For Salary Prediction Featuring Economy-Wide Activities And Occupations,” Information, vol. 13, no. 10, p. 495, 2022.
[2] W. Jiang, “The Investigation And Prediction For Salary Trends In The Data Science Industry,” 2023.
[3] E. A. Bagyam, “Analysis of Data Science Job Salaries from 2020 to 2024: Trends and Influencing Factors,” in AI and the Multidisciplinary Landscape, 2024.
[4] A. A. G., N. S. D., and R. R., “Salary Estimator Using ML Algorithms,” Int. J. Res. Publ. Rev., vol. 4, no. 10, pp. 2834–2839, 2023.
[5] S. Aufiero, G. De Marzo, A. Sbardella, and A. Zaccaria, “Mapping Job Complexity And Skills Into Wages.” 2023.
[6] A. Das and P. Saha, “AI-Based Salary Predictor For Data Science Jobs,” in 2023 International Conference on Computer Communication and Informatics (ICCCI), 2023, pp. 1–6.
[7] J. Zhu, “Unveiling salary trends: Exploring Machine Learning Models For Predicting Data Science Job Salaries,” in Proc. ICIAAI, Atlantis Press, 2024, pp. 173–182.
[8] Q. Bao, “Enhancing Salary Prediction Accuracy with Advanced Machine Learning Models,” Appl. Comput. Eng., vol. 96, no. 1, pp. 149–154, 2024.
[9] T. Chen and C. Guestrin, “XGBoost: A Scalable Tree Boosting System,” in Proc. 22nd ACM SIGKDD Int. Conf. Knowledge Discovery and Data Mining (KDD), 2016, pp. 785–794.
[10] G. James, D. Witten, T. Hastie, and R. Tibshirani, An Introduction to Statistical Learning, 2nd ed. New York, NY: Springer, 2021.
[11] A. Singh and P. Mehta, “Classification of Salary Categories Using Ensemble Learning,” in Proceedings of the 2023 International Conference on Data Analytics, 2023, pp. 110–117.
[12] M. Khan and S. Ali, “Salary Prediction Framework Based on Employment Attributes,” J. AI Data Sci., vol. 5, no. 1, pp. 55–62, 2022.
[13] I. Goodfellow, Y. Bengio, and A. Courville, Deep Learning. MIT Press, 2016.
[14] T. Roy and S. Kundu, “Deep Learning for Job Salary Classification in the IT Sector,” Appl. Artif. Intell., vol. 38, no. 2, pp. 140–150, 2024.
[15] N. Sharma and V. Patel, “Predictive Modeling for IT Job Salaries Using Machine Learning,” Int. J. Comput. Appl., vol. 182, no. 30, pp. 25–30, 2024.
[16] H. Rahman and M. Uddin, “Evaluating Machine Learning Algorithms for Predicting AI Job Salaries,” AI Res. Rev., vol. 9, no. 3, pp. 90–98, 2023.
[17] C. Lee and H. Choi, “Analysis of Job Market Trends Using Supervised Learning,” in Proceedings of the 2022 IEEE Conference on Big Data, 2022, pp. 299–306.
[18] D. Bose and A. Chakraborty, “Hybrid Classification Methods for Job Role-Based Salary Estimation,” Expert Syst. Appl., vol. 225, p. 120097, 2023.
[19] S. Russell and P. Norvig, Artificial Intelligence: A Modern Approach, 4th ed. Pearson, 2021.
[20] Indeed.com, “AI Job Trends and Salary Reports.” 2024.
[21] T. Nguyen and Q. Pham, “A Survey on AI-Driven Labor Market Analysis,” Comput. Econ., vol. 64, pp. 89–104, 2023.
