Dynamic ensemble selection of convolutional neural networks and its application in flower classification

文献类型： 外文期刊

第一作者： Wang, Zhibin

作者： Wang, Zhibin;Wang, Kaiyi;Wang, Xiaofeng;Pan, Shouhui;Qiao, Xiaojun;Wang, Zhibin;Wang, Kaiyi;Wang, Xiaofeng;Pan, Shouhui;Qiao, Xiaojun

作者机构：

关键词： flowers; classification; convolutional neural network; dynamic ensemble selection

期刊名称：INTERNATIONAL JOURNAL OF AGRICULTURAL AND BIOLOGICAL ENGINEERING （ 影响因子：1.885； 五年影响因子：2.232 ）

ISSN： 1934-6344

年卷期： 2022 年 15 卷 1 期

页码：

收录情况： SCI

摘要： In recent years, convolutional neural networks (CNNs) have achieved great success in image classification. However, CNN models usually have complex network structures that tend to cause some related problems, such as redundancy of network parameters, low training efficiency, overfitting, and weak generalization ability. To solve these problems and improve the accuracy of flower classification, the advantages of CNNs were combined with those of ensemble learning and a method was developed for the dynamic ensemble selection of CNNs. First, MobileNet models pre-trained on a public dataset were transferred to flower datasets to train thirteen different MobileNet classifiers, and a resampling strategy was used to enhance the diversity of individual models. Second, the thirteen classifiers were sorted by a classifier sorting algorithm, before ensemble selection, to avoid an exhaustive search. Finally, with the credibility of recognition results, a classifier subset was dynamically selected and integrated to identify the flower species from their images. To verify the effectiveness, the proposed method was used to classify the images of five flower species. The accuracy of the proposed method was 95.50%, an improvement of 1.62%, 3.94%, 22.04%, 13.77%, and 0.44%, over those of MobileNet, Inception-v1, ResNet-50, Inception-ResNet-v2, and the linear ensemble method, respectively. In addition, the performance of the proposed method was compared with five other methods for flower classification. The experimental results demonstrated the accuracy and robustness of the proposed method.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序