Software defect prediction based on stacked contractive autoencoder and multi-objective optimization

Nana Zhang, Kun Zhu, Shi Ying*, Xu Wang

*Corresponding author for this work

Research output: Contribution to JournalArticleAcademicpeer-review

Abstract

Software defect prediction plays an important role in software quality assurance. However, the performance of the prediction model is susceptible to the irrelevant and redundant features. In addition, previous studies mostly regard software defect prediction as a single objective optimization problem, and multi-objective software defect prediction has not been thoroughly investigated. For the above two reasons, we propose the following solutions in this paper: (1) we leverage an advanced deep neural network-Stacked Contractive AutoEncoder (SCAE) to extract the robust deep semantic features from the original defect features, which has stronger discrimination capacity for different classes (defective or non-defective). (2) we propose a novel multi-objective defect prediction model named SMONGE that utilizes the Multi-Objective NSGAII algorithm to optimize the advanced neural network-Extreme learning machine (ELM) based on state-of-the-art Pareto optimal solutions according to the features extracted by SCAE. We mainly consider two objectives. One objective is to maximize the performance of ELM, which refers to the benefit of the SMONGE model. Another objective is to minimize the output weight norm of ELM, which is related to the cost of the SMONGE model. We compare the SCAE with six state-of-the-art feature extraction methods and compare the SMONGE model with multiple baseline models that contain four classic defect predictors and the MONGE model without SCAE across 20 open source software projects. The experimental results verify that the superiority of SCAE and SMONGE on seven evaluation metrics.

Original languageEnglish
Pages (from-to)279-308
Number of pages30
JournalComputers, Materials and Continua
Volume65
Issue number1
DOIs
Publication statusPublished - 23 Jul 2020

Funding

Funding Statement: This work is supported in part by the National Science Foundation of China (Grant Nos. 61672392, 61373038), and in part by the National Key Research and Development Program of China (Grant No. 2016YFC1202204).

FundersFunder number
National Natural Science Foundation of China61672392, 61373038
National Natural Science Foundation of China
National Key Research and Development Program of China2016YFC1202204
National Key Research and Development Program of China

    Keywords

    • Deep neural network
    • Extreme learning machine
    • Multi-objective optimization
    • Software defect prediction
    • Stacked contractive autoencoder

    Fingerprint

    Dive into the research topics of 'Software defect prediction based on stacked contractive autoencoder and multi-objective optimization'. Together they form a unique fingerprint.

    Cite this