The Study of Transferable Learning in Satellite Images Classification with Different Convolution Neural Networks

Konferenz: CAIBDA 2022 - 2nd International Conference on Artificial Intelligence, Big Data and Algorithms
17.06.2022 - 19.06.2022 in Nanjing, China

Tagungsband: CAIBDA 2022

Seiten: 6Sprache: EnglischTyp: PDF

Autoren:
Guo, Yutong (Department of Electronics and Communication Engineering, East China University of Science and Technology, Shanghai, Xuhui District, China)
Xiang, Huiting (Department of English, Nankai University, Tianjin, Nankai District, China)
Zhang, Yanan (Electrical and Computer Engineering, University of Massachusetts Amherst, USA)

Inhalt:
In the past, the detection of damaged buildings based on satellite images after hurricane is labor-intensive. The recent development of Convolutional Neural Network (CNN) makes possible a more effective method for the classification task. This paper seeks to compare the accuracy of different convolutional neural network architectures such as ResNet, VGG, MobileNet, and InceptionV3 with and without transfer learning. With Hyperparameter Configuration, a set of hyperparameters are adjusted and utilized for each architecture to achieve a highest accuracy. After the model construction, optimization, and architecture comparison for both with and without transfer learning, ResNet model with transfer learning demonstrates the best performance, with an accuracy of 99.25% on the test dataset. Moreover, this paper aims to investigate into the nature of transfer learning on different architectures when implementing classification task with satellite images and explore how the model structure influence the training of classifying satellite images with and without Transfer Learning. By exploring the model structures and visualizing the feature maps of the representative models such as ResNet and VGG both with and without transfer learning, this paper finds that the suitability of transfer learning in satellite images may be affected by the characteristics of the model architecture such as the residual block of ResNet model and that the brightness of the feature map may influence the accuracy of the model, which is related to whether transfer learning is implemented.