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 条
  • [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] SpotTune: Transfer Learning through Adaptive Fine-tuning
    Guo, Yunhui
    Shi, Honghui
    Kumar, Abhishek
    Grauman, Kristen
    Rosing, Tajana
    Feris, Rogerio
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 4800 - 4809
  • [3] Retraction Note: Improved transfer learning of CNN through fine-tuning and classifier ensemble for scene classification
    S. Thirumaladevi
    K. Veera Swamy
    M. Sailaja
    Soft Computing, 2024, 28 (Suppl 2) : 981 - 981
  • [4] Transfer Learning With Adaptive Fine-Tuning
    Vrbancic, Grega
    Podgorelec, Vili
    IEEE ACCESS, 2020, 8 (08): : 196197 - 196211
  • [5] 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,
  • [6] 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
  • [7] Road-Type Classification through Deep Learning Networks Fine-Tuning
    Saleh, Yaser
    Otoum, Nesreen
    JOURNAL OF INFORMATION & KNOWLEDGE MANAGEMENT, 2020, 19 (01)
  • [8] Adaptive Anomaly Detection Network for Unseen Scene Without Fine-Tuning
    Hu, Yutao
    Huang, Xin
    Luo, Xiaoyan
    PATTERN RECOGNITION AND COMPUTER VISION, PRCV 2021, PT II, 2021, 13020 : 311 - 323
  • [9] Hierarchical BERT with an adaptive fine-tuning strategy for document classification
    Kong, Jun
    Wang, Jin
    Zhang, Xuejie
    KNOWLEDGE-BASED SYSTEMS, 2022, 238
  • [10] Extreme Fine-tuning: A Novel and Fast Fine-tuning Approach for Text Classification
    Jiaramaneepinit, Boonnithi
    Chay-intr, Thodsaporn
    Funakoshi, Kotaro
    Okumura, Manabu
    PROCEEDINGS OF THE 18TH CONFERENCE OF THE EUROPEAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS, VOL 2: SHORT PAPERS, 2024, : 368 - 379