TY - JOUR
T1 - Fruit classification by wavelet-entropy and feedforward neural network trained by fitness-scaled chaotic abc and biogeography-based optimization
AU - Wang, Shuihua
AU - Zhang, Yudong
AU - Ji, Genlin
AU - Yang, Jiquan
AU - Wu, Jianguo
AU - Wei, Ling
N1 - Publisher Copyright:
© 2015 by the authors.
PY - 2015
Y1 - 2015
N2 - Fruit classification is quite difficult because of the various categories and similar shapes and features of fruit. In this work, we proposed two novel machine-learning based classification methods. The developed system consists of wavelet entropy (WE), principal component analysis (PCA), feedforward neural network (FNN) trained by fitness-scaled chaotic artificial bee colony (FSCABC) and biogeography-based optimization (BBO), respectively. The K-fold stratified cross validation (SCV) was utilized for statistical analysis. The classification performance for 1653 fruit images from 18 categories showed that the proposed "WE + PCA + FSCABC-FNN" and "WE + PCA + BBO-FNN" methods achieve the same accuracy of 89.5%, higher than state-of-the-art approaches: "(CH + MP + US) + PCA + GA-FNN " of 84.8%, "(CH + MP + US) + PCA + PSO-FNN" of 87.9%, "(CH + MP + US) + PCA + ABC-FNN" of 85.4%, "(CH + MP + US) + PCA + kSVM" of 88.2%, and "(CH + MP + US) + PCA + FSCABC-FNN" of 89.1%. Besides, our methods used only 12 features, less than the number of features used by other methods. Therefore, the proposed methods are effective for fruit classification.
AB - Fruit classification is quite difficult because of the various categories and similar shapes and features of fruit. In this work, we proposed two novel machine-learning based classification methods. The developed system consists of wavelet entropy (WE), principal component analysis (PCA), feedforward neural network (FNN) trained by fitness-scaled chaotic artificial bee colony (FSCABC) and biogeography-based optimization (BBO), respectively. The K-fold stratified cross validation (SCV) was utilized for statistical analysis. The classification performance for 1653 fruit images from 18 categories showed that the proposed "WE + PCA + FSCABC-FNN" and "WE + PCA + BBO-FNN" methods achieve the same accuracy of 89.5%, higher than state-of-the-art approaches: "(CH + MP + US) + PCA + GA-FNN " of 84.8%, "(CH + MP + US) + PCA + PSO-FNN" of 87.9%, "(CH + MP + US) + PCA + ABC-FNN" of 85.4%, "(CH + MP + US) + PCA + kSVM" of 88.2%, and "(CH + MP + US) + PCA + FSCABC-FNN" of 89.1%. Besides, our methods used only 12 features, less than the number of features used by other methods. Therefore, the proposed methods are effective for fruit classification.
KW - Artificial bee colony
KW - Biogeography-based optimization
KW - Feed-forward neural network
KW - Fruit classification
KW - Machine learning
KW - Shannon entropy
KW - Wavelet transform
UR - http://www.scopus.com/inward/record.url?scp=84940487338&partnerID=8YFLogxK
U2 - 10.3390/e17085711
DO - 10.3390/e17085711
M3 - Article
AN - SCOPUS:84940487338
SN - 1099-4300
VL - 17
SP - 5711
EP - 5728
JO - Entropy
JF - Entropy
IS - 8
ER -