Skip to content

Fusion of CNN Ensemble for Remote Sensing Scene Classification

License

Notifications You must be signed in to change notification settings

alhichri/RS-DeepSuperLearner

Repository files navigation

RS-DeepSuperLearner:

Fusion of CNN Ensemble for Remote Sensing Scene Classification

Scene classification is an important problem in remote sensing (RS) and has attracted a lot of research in the past decade. Nowadays, most proposed methods are based on deep convolutional neural network (CNN) models, and many pretrained CNN models have been investigated. Ensemble techniques are well studied in the machine learning community; however, few works have used them in RS scene classification. In this work, we propose an ensemble approach, called RS-DeepSuperLearner, that fuses the outputs of five advanced CNN models, namely, VGG16, Inception-V3, DenseNet121, InceptionResNet-V2, and EfficientNet-B3. First, we improve the architecture of the five CNN models by attaching an auxiliary branch at specific layer locations. In other words, the models now have two output layers producing predictions each and the final prediction is the average of the two. The RS-DeepSuperLearner method starts by fine-tuning the five CNN models using the training data. Then, it employs a deep neural network (DNN) SuperLearner to learn the best way for fusing the outputs of the five CNN models by training it on the predicted probability outputs and the cross-validation accuracies (per class) of the individual models. The proposed methodology was assessed on six publicly available RS datasets: UC Merced, KSA, RSSCN7, Optimal31, AID, and NWPU-RSC45. The experimental results demonstrate its superior capabilities when compared to state-of-the-art methods in the literature.

H. Alhichri, “RS-DeepSuperLearner: Fusion of CNN Ensemble for Remote Sensing Scene Classification,” Annals of GIS, vol. 29, no. 1, Jan. 2023.

About

Fusion of CNN Ensemble for Remote Sensing Scene Classification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published