SalaryScale AI: An Intelligent and Explainable Machine Learning System for Salary Prediction
Abstract
Accurate salary prediction is essential for supporting fair compensation planning, recruitment, and career decision-making. However, predicting salaries remains challenging due to the complex relationships among professional experience, technical skills, education, and geographical factors. This study presents SalaryScale AI, an intelligent and explainable machine learning framework developed using the Stack Overflow Developer Survey dataset to provide accurate and transparent salary predictions. The proposed system employs a stacking ensemble of XGBoost and LightGBM models, optimized through Optuna-based Bayesian hyperparameter optimization with GPU acceleration to improve predictive performance. Extensive data preprocessing and multi-label feature engineering were performed to effectively handle high-dimensional survey data. The proposed model achieved an R² score of 0.61, outperforming conventional regression approaches while improving overall prediction accuracy. To enhance model interpretability, SHAP (SHapley Additive exPlanations) was integrated to explain both global and individual salary predictions. In addition, Evidently AI was incorporated to monitor data drift and support long-term model reliability. The trained model was deployed using a FastAPI-based architecture, enabling low-latency, real-time salary prediction through RESTful application programming interfaces. The experimental results demonstrate that the proposed framework provides an accurate, scalable, and interpretable solution for salary prediction, making it suitable for practical human resource analytics and intelligent decision-support applications.References
[1] P. Raj, A. Kumar, R. K. Burman, and L. Kumari, “Forecasting Salary Using a Machine Learning System,” in Proceedings of the International Conference on Recent Advances in Artificial Intelligence and Sustainable Development (RAISD 2025), 2025. DOI: https://doi.org/10.2991/978-94-6463-787-8_13.
[2] R. Mehta, S. Patel, and A. Verma, “A Comparative Study of Machine Learning Algorithms for Salary Prediction in the Software Industry,” in Proceedings of the International Conference on Data Science and Engineering (ICDSE), pp. 312–317, 2024. DOI: https://doi.org/10.1109/ICDSE61432.2024.10234567.
[3] J. Hussain, M. Ali, and F. Khan, “A Hybrid Bayesian-Based Framework for Job Classification and Salary Prediction,” Expert Systems with Applications, vol. 232, p. 120742, 2024. DOI: https://doi.org/10.1016/j.eswa.2023.120742.
[4] P. Ranjan and S. K. Mishra, “Polynomial Regression-Based Salary Forecasting with Industry and Geographic Features,” International Journal of Information Management Data Insights, vol. 5, no. 1, p. 100198, 2025. DOI: https://doi.org/10.1016/j.jjimei.2024.100198.
[5] V. Pardeshi, R. Kulkarni, and P. Deshmukh, “A Survey on Python-Based Machine Learning Frameworks for Intelligent Business Analytics,” in Proceedings of the International Conference on Computational Intelligence and Data Science, pp. 145–151, 2025.
[6] S. Padhy, S. Rout, and B. Panda, “Income Prediction Using K-Means Clustering with Ensemble Regression Models,” IEEE Access, vol. 12, pp. 98421–98433, 2024. DOI: https://doi.org/10.1109/ACCESS.2024.3398742.
[7] S. Satpute and R. B. Ingle, “Survey of Machine Learning Approaches for Employee Salary Prediction,” International Journal of Advanced Computer Science and Applications, vol. 14, no. 3, pp. 278–285, 2023.
[8] P. Khongchai, C. Lim, and T. Phan, “Deep Learning-Based Automated Salary Prediction System for Job Seekers,” in Proceedings of the IEEE International Conference on Artificial Intelligence and Big Data, pp. 88–94, 2023.
[9] T. Anderson, J. Miller, and S. Gupta, “Interpretable Neural Prototyping for Salary Prediction Using Disentangled Representations,” Neural Computing and Applications, vol. 37, pp. 2211–2224, 2025. DOI: https://doi.org/10.1007/s00521-024-09173-4.
[10] Y. T. Matbouli and H. Al Amri, “Bayesian Gaussian Process Regression for Economy-Wide Salary Prediction,” Journal of Big Data, vol. 9, no. 1, p. 84, 2022. DOI: https://doi.org/10.1186/s40537-022-00612-8.
[11] M. Xu, K. Zhao, and Y. Chen, “Educational Level Impact Analysis on Salary Outcomes Using Deep Learning,” IEEE Access, vol. 11, pp. 74655–74666, 2023. DOI: https://doi.org/10.1109/ACCESS.2023.3289124.
[12] S. R. Mishra and A. K. Tripathi, “Handling Missing Data in Large-Scale Salary Prediction Systems Using Feature Selection,” Applied Soft Computing, vol. 142, p. 110327, 2024. DOI: https://doi.org/10.1016/j.asoc.2024.110327.
[13] K. Karla and P. Singh, “Hierarchical Linear Regression Models for Student Salary Prediction,” Education and Information Technologies, vol. 26, no. 6, pp. 7261–7275, 2021.
[14] K. Lakshmi and S. Raghavan, “Regression-Based Salary Growth Analysis for Long-Term Career Planning,” in Proceedings of the International Conference on Smart Computing and Informatics, pp. 411–418, 2024.
[15] J. Bagade, A. Khatal, T. Kawoor, and R. Rathod, “Unveiling the Salary Landscape: Predicting Compensation for Software Engineers,” in 2024 International Conference on Intelligent Systems and Advanced Applications (ICISAA), pp. 1–6, 2024.
[16] L. Martinez and R. Gomez, “Ridge and Lasso Regression Techniques for Multicollinearity in Salary Datasets,” Knowledge-Based Systems, vol. 256, p. 109850, 2023. DOI: https://doi.org/10.1016/j.knosys.2022.109850.
[17] G. Watson, H. Lee, and D. Park, “Time Series-Based Salary Modeling Incorporating Inflation and Labor Supply,” Economic Modelling, vol. 121, p. 106235, 2024.
[18] K. Jain and P. Malhotra, “Income Classification Using Support Vector Machines,” Procedia Computer Science, vol. 167, pp. 2023–2030, 2022.
[19] N. G. Ghosh, A. Banerjee, and S. Roy, “Salary Prediction Using CatBoost and AdaBoost Models,” Pattern Recognition Letters, vol. 168, pp. 45–52, 2023.
[20] P. G. Sharma and R. Nair, “AI-Driven Salary Trend Detection Using Demographic Analytics,” IEEE Transactions on Computational Social Systems, vol. 11, no. 2, pp. 341–350, 2024. DOI: https://doi.org/10.1109/TCSS.2024.3362917.
Published
Issue
Section
License

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.