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

被引:30
作者
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]   Learning label smoothing for text classification [J].
Ren H. ;
Zhao Y. ;
Zhang Y. ;
Sun W. .
PeerJ Computer Science, 2024, 10
[22]   Adaptive multi-source domain collaborative fine-tuning for transfer learning [J].
Feng, Le ;
Yang, Yuan ;
Tan, Mian ;
Zeng, Taotao ;
Tang, Huachun ;
Li, Zhiling ;
Niu, Zhizhong ;
Feng, Fujian .
PEERJ COMPUTER SCIENCE, 2024, 10
[23]   Compressing BERT for Binary Text Classification via Adaptive Truncation before Fine-Tuning [J].
Zhang, Xin ;
Fan, Jing ;
Hei, Mengzhe .
APPLIED SCIENCES-BASEL, 2022, 12 (23)
[24]   Learning to Identify Illegal Landfills through Scene Classification in Aerial Images [J].
Torres, Rocio Nahime ;
Fraternali, Piero .
REMOTE SENSING, 2021, 13 (22)
[25]   Simple Yet Effective Fine-Tuning of Deep CNNs Using an Auxiliary Classification Loss for Remote Sensing Scene Classification [J].
Bazi, Yakoub ;
Al Rahhal, Mohamad M. ;
Alhichri, Haikel ;
Alajlan, Naif .
REMOTE SENSING, 2019, 11 (24)
[26]   Bagging and Boosting Fine-Tuning for Ensemble Learning [J].
Zhao C. ;
Peng R. ;
Wu D. .
IEEE Transactions on Artificial Intelligence, 2024, 5 (04) :1728-1742
[27]   RAFNet: Interdomain Representation Alignment and Fine-Tuning for Image Series Classification [J].
Gong, Maoguo ;
Qiao, Wenyuan ;
Li, Hao ;
Qin, A. K. ;
Gao, Tianqi ;
Luo, Tianshi ;
Xing, Lining .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
[28]   Fine-Tuning Models for Histopathological Classification of Colorectal Cancer [J].
ALGhafri, Houda Saif ;
Lim, Chia S. .
DIAGNOSTICS, 2025, 15 (15)
[29]   Enhancement of the Facial Recognition Module in the "Safe University" System through Adaptive Fine-Tuning [J].
Denissova, Natalya ;
Dyomina, Irina ;
Tlebaldinova, Aizhan ;
Apayev, Kurmash .
JOURNAL OF ADVANCES IN INFORMATION TECHNOLOGY, 2025, 16 (01) :144-155
[30]   Fine-Tuning of Distil-BERT for Continual Learning in Text Classification: An Experimental Analysis [J].
Shah, Sahar ;
Manzoni, Sara Lucia ;
Zaman, Farooq ;
Es Sabery, Fatima ;
Epifania, Francesco ;
Zoppis, Italo Francesco .
IEEE ACCESS, 2024, 12 :104964-104982