Power of LSTM and SHAP in the Use Case Point Approach for Software Effort and Cost Estimation
Ranković,Nevena Rankovic,Dragica
Ranković,Nevena
Rankovic,Dragica
Abstract
In this paper, we explore the effectiveness of the most popular regression models within Machine Learning (ML), such as XGBoost and MLP (Multilayer Percpetron) Regressor, and compare them with different types of Recurrent Neural Networks: LSTM (Long-Short-Term-Memory) and GRU (Gated Recurrent Unit). These models are later employed to conduct an experimental research in predicting the Real Effort of a software project using the Use Case Points (UCP) approach. Additionally, we applied a data augmentation technique to artificially increase the number of instances in the UCP Benchmark Mendeley dataset. The results have demonstrated that LSTM outperforms all the models, even though GRU is computationally more efficient and easier to train. Furthermore, we examined the SHAP (SHapley Additive exPlanations) feature importance method to improve the interpretability of the best-performing model.
Description
Publisher Copyright: © 2024 IEEE.
Date
2024-02-14
Journal Title
Journal ISSN
Volume Title
Publisher
IEEE
Research Projects
Organizational Units
Journal Issue
Keywords
Machine Learning (ML), Recurrent neural Networks (RNNs), SHAP, Use Case Points (UCP), data augmentation, software estimation
Citation
Ranković, N & Rankovic, D 2024, Power of LSTM and SHAP in the Use Case Point Approach for Software Effort and Cost Estimation. in 2024 IEEE 22nd World Symposium on Applied Machine Intelligence and Informatics (SAMI). 2024 IEEE 22nd World Symposium on Applied Machine Intelligence and Informatics, SAMI 2024 - Proceedings, IEEE, pp. 59-64. https://doi.org/10.1109/SAMI60510.2024.10432878