首页|Fusing CNNs and statistical indicators to improve image classification

Fusing CNNs and statistical indicators to improve image classification

扫码查看
Convolutional Neural Networks have dominated the field of computer vision for the last ten years, exhibiting extremely powerful feature extraction capabilities and outstanding classification performance. The main strategy to prolong this trend in the state-of-the-art literature relies on further upscaling networks in size. However, costs increase rapidly while performance improvements may be marginal. Our main hypothesis is that adding additional sources of information can help to increase performance and that this approach is more cost-effective than building bigger networks, which involve higher training time, larger parametrisation space and higher computational resources requirements. In this paper, an ensemble method for accurate image classification is proposed, fusing automatically detected features through a Convolutional Neural Network and a set of manually defined statistical indicators. Through a combination of the predictions of a CNN and a secondary classifier trained on statistical features, a better classification performance can be achieved cheaply. We test five different CNN architectures and multiple learning algorithms in a diverse number of datasets to validate our proposal. According to the results, the inclusion of additional indicators and an ensemble classification approach help to increase the performance in all datasets. Both code and datasets are publicly available via GitHub at: https://github.com/jahuerta92/cnn- prob-ensemble.

Convolutional Neural NetworksFeature extractionEnsemble learningData fusionStatistical indicatorsRANDOM FORESTMULTIPLE CLASSIFIERSFEATURESENSEMBLE

Huertas-Tato, Javier、Martin, Alejandro、Fierrez, Julian、Camacho, David

展开 >

Univ Politecn Madrid

Univ Autonoma Madrid

2022

Information Fusion

Information Fusion

EISCI
ISSN:1566-2535
年,卷(期):2022.79
  • 9
  • 86