Aerial Scene Classification through Fine-Tuning with Adaptive Learning Rates and Label Smoothing

被引:29
|
作者
Petrovska, Biserka [1 ]
Atanasova-Pacemska, Tatjana [2 ]
Corizzo, Roberto [3 ]
Mignone, Paolo [4 ]
Lameski, Petre [5 ]
Zdravevski, Eftim [5 ]
机构
[1] Minist Def, Skopje 1000, North Macedonia
[2] Univ Goce Delcev, Fac Comp Sci, Stip 2000, North Macedonia
[3] Amer Univ, Dept Comp Sci, 4400 Massachusetts Ave NW, Washington, DC 20016 USA
[4] Univ Bari Aldo Moro, Dept Comp Sci, Via E Orabona 4, I-70125 Bari, Italy
[5] Ss Cyril & Methodius Univ Skopje, Fac Comp Sci & Engn, Rugjer Boshkovik 16, Skopje 1000, North Macedonia
来源
APPLIED SCIENCES-BASEL | 2020年 / 10卷 / 17期
关键词
remote sensing; convolutional neural network; fine-tuning; learning rate scheduler; cyclical learning rates; label smoothing; classification accuracy; CONVOLUTIONAL NEURAL-NETWORKS; RECOGNITION;
D O I
10.3390/app10175792
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Remote Sensing (RS) image classification has recently attracted great attention for its application in different tasks, including environmental monitoring, battlefield surveillance, and geospatial object detection. The best practices for these tasks often involve transfer learning from pre-trained Convolutional Neural Networks (CNNs). A common approach in the literature is employing CNNs for feature extraction, and subsequently train classifiers exploiting such features. In this paper, we propose the adoption of transfer learning by fine-tuning pre-trained CNNs for end-to-end aerial image classification. Our approach performs feature extraction from the fine-tuned neural networks and remote sensing image classification with a Support Vector Machine (SVM) model with linear and Radial Basis Function (RBF) kernels. To tune the learning rate hyperparameter, we employ a linear decay learning rate scheduler as well as cyclical learning rates. Moreover, in order to mitigate the overfitting problem of pre-trained models, we apply label smoothing regularization. For the fine-tuning and feature extraction process, we adopt the Inception-v3 and Xception inception-based CNNs, as well the residual-based networks ResNet50 and DenseNet121. We present extensive experiments on two real-world remote sensing image datasets: AID and NWPU-RESISC45. The results show that the proposed method exhibits classification accuracy of up to 98%, outperforming other state-of-the-art methods.
引用
收藏
页数:25
相关论文
共 50 条
  • [1] Improved transfer learning of CNN through fine-tuning and classifier ensemble for scene classification
    Thirumaladevi, S.
    Swamy, K. Veera
    Sailaja, M.
    SOFT COMPUTING, 2022, 26 (12) : 5617 - 5636
  • [2] Transfer Learning With Adaptive Fine-Tuning
    Vrbancic, Grega
    Podgorelec, Vili
    IEEE ACCESS, 2020, 8 (08): : 196197 - 196211
  • [3] Road-Type Classification through Deep Learning Networks Fine-Tuning
    Saleh, Yaser
    Otoum, Nesreen
    JOURNAL OF INFORMATION & KNOWLEDGE MANAGEMENT, 2020, 19 (01)
  • [4] Adaptive Fine-tuning for Deep Transfer Learning Based Traffic Signs Classification
    Nasri, Ismail
    Messaoudi, Abdelhafid
    Kassmi, Kamal
    Karrouchi, Mohammed
    Snoussi, Hajar
    2021 4TH INTERNATIONAL SYMPOSIUM ON ADVANCED ELECTRICAL AND COMMUNICATION TECHNOLOGIES (ISAECT), 2021,
  • [5] Improving unbalanced image classification through fine-tuning method of reinforcement learning
    Wang, Jin-Qiang
    Guo, Lan
    Jiang, Yuanbo
    Zhang, Shengjie
    Zhou, Qingguo
    APPLIED SOFT COMPUTING, 2024, 163
  • [6] Comparison of fine-tuning strategies for transfer learning in medical image classification
    Davila, Ana
    Colan, Jacinto
    Hasegawa, Yasuhisa
    IMAGE AND VISION COMPUTING, 2024, 146
  • [7] Evaluation of Parameter Fine-Tuning with Transfer Learning for Osteoporosis Classification in Knee Radiograph
    Abubakar, Usman Bello
    Boukar, Moussa Mahamat
    Adeshina, Steve
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2022, 13 (08) : 246 - 252
  • [8] Adaptive fine-tuning strategy for few-shot learning
    Zhuang, Xinkai
    Shao, Mingwen
    Gao, Wei
    Yang, Jianxin
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (06)
  • [9] Gemstone classification using ConvNet with transfer learning and fine-tuning
    Freire, Willian M.
    Amaral, Aline M. M. M.
    Costa, Yandre M. G.
    2022 29TH INTERNATIONAL CONFERENCE ON SYSTEMS, SIGNALS AND IMAGE PROCESSING (IWSSIP), 2022,
  • [10] Forest Image Classification Based on Fine-Tuning CaffeNet
    Zhang G.
    Li Y.
    Wang H.
    Zhou H.
    Linye Kexue/Scientia Silvae Sinicae, 2020, 56 (10): : 121 - 128