Optimizing Model and Feature Selection using Evolutionary Swarm Intelligence for Image Classification

Sharma, Chethan and Kapoor, Raloue and Shenoy, Ajitha K B and Poornalatha, G (2020) Optimizing Model and Feature Selection using Evolutionary Swarm Intelligence for Image Classification. Journal of Digital Imaging, 29 (5). pp. 2160-2165. ISSN 0897-1889

[img] PDF
8617.pdf - Published Version
Restricted to Registered users only

Download (343kB) | Request a copy


Model selection is the way towards choosing one machine learning model among an assortment of competitor models for a training dataset. The correct model selection is vital in academic machine learning research as well as in many industrial settings. Figuring out the relevant features from a set of features and removing the irrelevant or less vital features which no longer make contributions to the target variable so that we can gain higher accuracy for our model is a common problem. Feature Selection is one of the main things which highly influences the performance of the model. The features that are used to train the models have a big effect on the performance one could gain. Irrelevant features can negatively impact model performance. Evolutionary algorithms are known for optimizing the process. In this paper we use an evolutionary algorithm called Ant Colony Optimization (ACO) to select an optimized model for given data and then use another algorithm called Particle Swarm Optimization (PSO) to select best features so that we get best of both model and features for the give data. We found that the combination of an optimal architecture selected by ACO and optimal features selected by PSO gives higher accuracy.

Item Type: Article
Uncontrolled Keywords: Evolutionary algorithm, Ant Colony Optimization, Particle Swarm Optimization, Model selection, machine learning, Feature selection
Subjects: Engineering > MIT Manipal > Information and Communication Technology
Depositing User: MIT Library
Date Deposited: 25 Jun 2020 09:54
Last Modified: 25 Jun 2020 09:54
URI: http://eprints.manipal.edu/id/eprint/155352

Actions (login required)

View Item View Item