Journal
INFORMATION FUSION
Volume 79, Issue -, Pages 174-187Publisher
ELSEVIER
DOI: 10.1016/j.inffus.2021.09.012
Keywords
Convolutional Neural Networks; Feature extraction; Ensemble learning; Data fusion; Statistical indicators
Funding
- Spanish Ministry of Science and Innovation [TIN2017-85727-C4-3-P, PID2020-117263GB-100, RTI2018-101248-B-I00]
- Comunidad Autonoma de Madrid [S2018/TCS-4566]
- BBVA Foundation
- CHIST-ERA 2017 programme 2017 (BDSI PACMAEL project) [PCI2019-103623]
- Comunidad Autonoma de Madrid
Ask authors/readers for more resources
This paper proposes an ensemble method for accurate image classification, which combines automatically detected features and statistical indicators to achieve better performance. Testing on various datasets shows that including additional indicators and using an ensemble classification approach can improve performance.
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.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available