A Model Selection Method for Machine Learning by Differential Evolution
| Autor: | Yi-Chuan Chiu, Yung-Tsan Jou, Hsing-Hung Lin |
|---|---|
| Rok vydání: | 2019 |
| Předmět: |
Computer science
business.industry Model selection 05 social sciences Big data 050301 education Inference Regression analysis 02 engineering and technology Machine learning computer.software_genre Ensemble learning Differential evolution Genetic algorithm 0202 electrical engineering electronic engineering information engineering Benchmark (computing) 020201 artificial intelligence & image processing Artificial intelligence business 0503 education computer |
| Zdroj: | Proceedings of the 2019 4th International Conference on Big Data and Computing - ICBDC 2019. |
| DOI: | 10.1145/3335484.3335505 |
| Popis: | While the application of big data attracts more and more attention, machine learning algorithms are developing with each passing day, and the models produced by machine learning are increasingly diversified. The focus of big data applications has gradually shifted from model training to the prediction and inference. Choosing the most suitable model for enterprise application scenarios among many machine learning models has become a topic of research that has attracted much attention. Though ensemble methods have been proposed to discover best model by multiple training phase, studies of finding best combination within multiple modes are still few. Configuring different machine learning models with appropriate parameters and looking for parameters is an NP-hard problem, which requires an optimization algorithm. This study proposes a differential evolution algorithm to integrate multiple trained machine learning models into an appropriate model. In this paper, the regression model is taken as an example and the differential evolution algorithm is compared with the genetic algorithm. Three benchmark datasets are used to examine, and the results show that the differential evolution algorithm outperforms genetic algorithm. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|