The ant colony algorithm for feature selection in high-dimension gene expression data for disease classification

The use of gene expression data to diagnose complex diseases represents an exciting area of medicine; however, such data sets are often noisy, requiring the selection of feature subsets to obtain maximum classification accuracy. Due to the high dimensions of many expression data sets, filter-based methods are commonly used, but often yield inconsistent results. Optimization algorithms can outperform filter methods, but often require preselection of features to achieve good results. To address the problems of many commonly used feature selection methods, the ant colony algorithm (ACA) is proposed for use on data sets with large numbers of features. The ACA is an optimization algorithm capable of incorporating prior information, allowing it to search the sample space more efficiently than other optimization methods. When applied to several high-dimensional data sets, the ACA was able to identify small subsets of highly predictive and biologically relevant genes without the need for extensive preselection of features. Using the selected genes to train a latent variable model yielded substantial increases in prediction accuracy when compared to several rank-based methods and results obtained in previous studies. The superiority of the ACA algorithm was validated through simulation.