Academic Journal of Computing & Information Science, 2022, 5(7); doi: 10.25236/AJCIS.2022.050712.
Kun Yao, Xinyue Cao
Hohai University, Changzhou, 213022, China
In the application of unmanned aerial vehicle (UAV), land classification is an important field. Image recognition technology based on deep learning is a new method to solve the problem of land classification in recent years. In this paper, we propose a land classification method based on federated learning (FL) which uses Fedavg-Adam algorithm. This method enables UAVs to learn land models using training data distributed in their local database. By aggregating local computational updates of the land classification model, a shared global model is constructed. UAVs can collectively benefit from global models without sharing datasets and protect sensitive information collected by UAVs. To obtain good classification performance, we further propose an improved CNN network. The experimental results show that the improved CNN network suitable for federated learning has the good performance considering the influence of time and accuracy. In the RSSCN7 Dataset, the FedAvg-Adam algorithm converges in the 43rd rounds with an accuracy rate of 82.71%. Compared with FedAvg and FedAdam, the final accuracy of the three is similar, but the latter two converge at 56 rounds and 114 rounds respectively, and Fedavg-Adam has the fastest speed, which proves the superiority of our method.
UAVs, Land Classification, CNN, Federated Learning
Kun Yao, Xinyue Cao. UAV Land Classification Method Based on Federated Learning. Academic Journal of Computing & Information Science (2022), Vol. 5, Issue 7: 73-78. https://doi.org/10.25236/AJCIS.2022.050712.
[1] M. Bhaskaranand, a. J. (n.d.). Low-complexity video encoding for UAV reconnaissance and surveillance. Proc. IEEE Military Communications Conference (MILCOM), pp. 1633-1638, 2011.
[2] Dong, Q.; Zou, Q. Visual UA V detection method with online feature classifification. In Proceedings of the IEEE 2nd Information Technology, Networking, Electronic and Automation Control Conference (ITNEC), Chengdu, China, 15–17 December 2017; pp. 429–432.
[3] Gong Xi, et al. "A hybrid expert classification model based on transfer learning for scene classification of high-resolution remote sensing images" [J]. Acta Optica Sinica, 2019, 41(23):19-31.
[4] Peng, Yingqiong, et al. "CNN–SVM: a classification method for fruit fly image with the complex background." IET Cyber-Physical Systems: Theory & Applications 5.2 (2020): 181-185.
[5] Tan, Cong, et al. "A Lightweight Underwater Object Detection Model: FL-YOLOV3-TINY." 2021 IEEE 12th Annual Information Technology, Electronics and Mobile Communication Conference (IEMCON). IEEE, 2021.
[6] SHI Chao-jun et al. Automatic Classification Method of Star Spectra Data Based on Convolutional Neural Network[J]. Spectroscopy and Spectral Analysis, 2018, 39(04): 1312-1316.
[7] P. K. Mudrakarta, M. Sandler, A. Zhmoginov, and A. Howard, “K for the price of 1: Parameter efficient multi-task and transfer learning,” in Proc. Int. Conf. Learn. Representations, 2019.
[8] Z. Hao, F. Ren, X. Kang, H. Ni, S. Lv and H. Wang, "Classification of Steel Strip Surface Defects Based on Optimized ResNet18," 2021 IEEE International Conference on Agents (ICA), 2021, pp. 61-62.
[9] Q. Zou, L. Ni, T. Zhang and Q. Wang, "Deep Learning Based Feature Selection for Remote Sensing Scene Classification," in IEEE Geoscience and Remote Sensing Letters, vol. 12, no. 11, pp. 2321-2325, Nov. 2015. Available: https://hyper.ai/datasets/5440/
[10] J. Zhang et al., "Adaptive Federated Learning on Non-IID Data with Resource Constraint," in IEEE Transactions on Computers. 2020.
[11] Q. Wang, H. Wang and Y. Yan, "Reconstruction of CT images based on cross-entropy method," 2010 IEEE Instrumentation & Measurement Technology Conference Proceedings, 2010, pp. 1510-1514.
[12] J. Mills, J. Hu and G. Min, "Multi-Task Federated Learning for Personalised Deep Neural Networks in Edge Computing," in IEEE Transactions on Parallel and Distributed Systems, vol. 33, no. 3, pp. 630-641, 1 March 2022.
[13] D. Leroy, A. Coucke, T. Lavril, T. Gisselbrecht and J. Dureau, "Federated Learning for Keyword Spotting," ICASSP 2019-2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2019, pp. 6341-6345.