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

被引:28
|
作者
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 条
  • [21] An Application of Transfer Learning: Fine-Tuning BERT for Spam Email Classification
    Bhopale, Amol P.
    Tiwari, Ashish
    MACHINE LEARNING AND BIG DATA ANALYTICS (PROCEEDINGS OF INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND BIG DATA ANALYTICS (ICMLBDA) 2021), 2022, 256 : 67 - 77
  • [22] Classification of Focal Liver Lesions Using Deep Learning with Fine-Tuning
    Wang, Weibin
    Iwamoto, Yutaro
    Han, Xianhua
    Chen, Yen-Wei
    Chen, Qingqing
    Liang, Dong
    Lin, Lanfen
    Hu, Hongjie
    Zhang, Qiaowei
    PROCEEDINGS OF 2018 INTERNATIONAL CONFERENCE ON DIGITAL MEDICINE AND IMAGE PROCESSING (DMIP 2018), 2018, : 56 - 60
  • [23] Fine-tuning of unmanned aerial surveillance for ecological systems
    Baxter, P. W. J.
    Hamilton, G.
    21ST INTERNATIONAL CONGRESS ON MODELLING AND SIMULATION (MODSIM2015), 2015, : 1393 - 1398
  • [24] Model adaptive parameter fine-tuning Based on contribution measure for image classification
    Feng, Le
    Feng, Fujian
    Yang, Yuan
    Tan, Mian
    Wang, Lin
    NEUROCOMPUTING, 2025, 632
  • [25] Neighbor-Based Label Distribution Learning to Model Label Ambiguity for Aerial Scene Classification
    Luo, Jianqiao
    Wang, Yihan
    Ou, Yang
    He, Biao
    Li, Bailin
    REMOTE SENSING, 2021, 13 (04) : 1 - 25
  • [26] Movement Pruning: Adaptive Sparsity by Fine-Tuning
    Sanh, Victor
    Wolf, Thomas
    Rush, Alexander M.
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 33, NEURIPS 2020, 2020, 33
  • [27] Deep learning-based weld defect classification using VGG16 transfer learning adaptive fine-tuning
    Samuel Kumaresan
    K. S. Jai Aultrin
    S. S. Kumar
    M. Dev Anand
    International Journal on Interactive Design and Manufacturing (IJIDeM), 2023, 17 : 2999 - 3010
  • [28] Deep learning-based weld defect classification using VGG16 transfer learning adaptive fine-tuning
    Kumaresan, Samuel
    Aultrin, K. S. Jai
    Kumar, S. S.
    Anand, M. Dev
    INTERNATIONAL JOURNAL OF INTERACTIVE DESIGN AND MANUFACTURING - IJIDEM, 2023, 17 (06): : 2999 - 3010
  • [29] Large Scale Nonlinear Control System Fine-Tuning Through Learning
    Kosmatopoulos, Elias B.
    Kouvelas, Anastasios
    IEEE TRANSACTIONS ON NEURAL NETWORKS, 2009, 20 (06): : 1009 - 1023
  • [30] Towards Robust and Generalized Parameter-Efficient Fine-Tuning for Noisy Label Learning
    Kim, Yeachan
    Kim, Junho
    Lee, SangKeun
    PROCEEDINGS OF THE 62ND ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS, VOL 1: LONG PAPERS, 2024, : 5922 - 5936