Methodology of solving the feature selection problem for the Cox regression model-Reference-Cited by-同舟云学术

Methodology of solving the feature selection problem for the Cox regression model

Published:2024-07-29 Issue:3 Volume:2024 Page:85-94
ISSN:2072-9502
Container-title:Vestnik of Astrakhan State Technical University. Series: Management, computer science and informatics
language:en
Short-container-title:

Author:

Mikulik Il'ya Igorevich¹

Affiliation:

1. Emperor Alexander I St. Petersburg State Transport University

Abstract

The technique based on the use of a hybrid optimization method to solve the feature selection problem for the Cox regression model is proposed. The hybrid optimization method includes two metaheuristic methods: the ant colony optimization and the genetic algorithm. The ant colony optimization used as the basic algorithm that solves the main optimization problem. The genetic algorithm solves the problem of finding the optimal set of parameters for the ant algorithm improving its performance. The method is modified and adapted to solve the problem under consideration. The key feature of adaptation is the deposition of pheromones on the vertices rather than on the edges of the graph, as well as the method for calculating the assessment of heuristic information about each vertex. A fitness target function was constructed that determines the quality of solutions to the feature selection problem and is based on an assessment of the performance of the Cox model with a selected set of parameters. The concordance index (c-index) was used to evaluate the Cox model. The efficiency of the methodology is shown using the implemented program using the example of a database of recidivism. For the database used, the most significant sets of features were obtained that have the greatest impact on the quality of training of the survival analysis model.

Publisher

Astrakhan State Technical University

Reference26 articles.

1. Archetti A., Lomurno E., Lattari F., Martin A., Matteucci M. Heterogeneous Datasets for Federated Survival Analysis Simulation // Companion of the 2023 ACM/SPEC International Conference on Performance Engineering, 2023. DOI: 10.1145/3578245.3584935., Archetti A., Lomurno E., Lattari F., Martin A., Matteucci M. Heterogeneous Datasets for Federated Survival Analysis Simulation. Companion of the 2023 ACM/SPEC International Conference on Performance Engineering, 2023. DOI: 10.1145/3578245.3584935.

2. George B., Seals S., Aban I. Survival analysis and regression models // J. Nucl. Cardiol. 2014. P. 686–694. DOI: 10.1007/s12350-014-9908-2., George B., Seals S., Aban I. Survival analysis and regression models. J. Nucl. Cardiol., 2014, pp. 686-694. DOI: 10.1007/s12350-014-9908-2.

3. Atlam M., Torkey H., El-Fishawy N., Salem H. Coronavirus disease 2019 (COVID-19): survival analysis using deep learning and Cox regression model // Pattern Anal. Applic. 2021. N. 24 (3). P. 993–1005. DOI: 10.1007/s10044-021-00958-0., Atlam M., Torkey H., El-Fishawy N., Salem H. Coronavirus disease 2019 (COVID-19): survival analysis using deep learning and Cox regression model. Pattern Anal. Applic., 2021, no. 24 (3), pp. 993-1005. DOI: 10.1007/s10044-021-00958-0.

4. Chen X., Yuan G., Nie F., Ming Z. Semi-Supervised Feature Selection via Sparse Rescaled Linear Square Regression // IEEE Transaction on Knowledge Discovery and Data Engineering 32, 2018. P. 165–176. DOI: 10.1109/TKDE.2018.2879797., Chen X., Yuan G., Nie F., Ming Z. Semi-Supervised Feature Selection via Sparse Rescaled Linear Square Regression. IEEE Transaction on Knowledge Discovery and Data Engineering 32, 2018. P. 165–176. DOI: 10.1109/TKDE.2018.2879797.

5. Rossi P. H., Berk R. A., Lenihan K. J. Money, work and crime: some experimental results. N. Y.: Academic Press, 1980. 336 p., Rossi P. H., Berk R. A., Lenihan K. J. Money, work and crime: some experimental results. New York, Academic Press, 1980. 336 p.