Abstract
Fine-grained vehicle recognition has been widely used in intelligent transportation systems, such as urban traffic monitoring, traffic flow statistics and public security criminal investigation. Nevertheless, state-of-the-art fine-grained vehicle recognition systems usually employ a centralized deep learning model, which leads to costly computing resource consumption. In addition, amounts of vehicle image data are collected by a central server, where the involved sensitive information may be leaked. To address the concerns, we propose a privacy-preserving and lightweight federated learning-based framework for fine-grained vehicle recognition in edge-assisted intelligent traffic system, termed as FedLVR. Technically, we introduce a lightweight fine-grained vehicle recognition model that can be efficiently deployed on edge nodes, and develop its extensions based on federated learning framework. In particular, our model improves ShuffleNet v2 by designing parallel dual attention mechanism and introduces improved contrastive center loss, which highly enhances the feature discriminative ability of the model. Moreover, we propose a new weighted federated averaging algorithm to improve the generalization ability of the model. To clarify performance of our FedLVR, we conduct extensive experiments on the Stanford Cars dataset and Compcars dataset. The results show that our proposed FedLVR not only achieves high accuracy, but also greatly reduces the complexity of the network model. Compared to state-of-the-art research, our FedLVR effectively reduces the risk of privacy leakage and addresses the problem of data islands.










Similar content being viewed by others
References
Badura P, Skotnicka M (2014) Automatic car make recognition in low-quality images. In: Information technologies in biomedicine. Springer, vol 3, pp 235–246
Baran R, Glowacz A, Matiolanski A (2015) The efficient real- and non-real-time make and model recognition of cars. Multimed Tools Appl 74(12):4269–4288
Bhatti UA, Yu Z, Hasnain A, Nawaz SA, Yuan L, Wen L, Bhatti MA (2022) Evaluating the impact of roads on the diversity pattern and density of trees to improve the conservation of species. Environ Sci Pollut Res:1–11
Bhatti UA, Yu Z, Li J, Nawaz SA, Mehmood A, Zhang K, Yuan L (2020) Hybrid watermarking algorithm using clifford algebra with arnold scrambling and chaotic encryption. IEEE Access 8:76386–76398
Bhatti UA, Yu Z, Yuan L, Nawaz SA, Aamir M, Bhatti MA (2022) A robust remote sensing image watermarking algorithm based on region-specific surf. In: Ullah A, Anwar S, Rocha Á, Gill S (eds) Proceedings of international conference on information technology and applications. Springer, pp 75–85
Bhatti UA, Yu Z, Yuan L, Zeeshan Z, Nawaz SA, Bhatti M, Mehmood A, Ain QU, Wen L (2020) Geometric algebra applications in geospatial artificial intelligence and remote sensing image processing. IEEE Access 8:155783–155796
Bhatti UA, Yuan L, Yu Z, Nawaz SA, Mehmood A, Bhatti MA, Nizamani MM, Xiao S et al (2021) Predictive data modeling using sp-knn for risk factor evaluation in urban demographical healthcare data. J Med Imaging Health Inf 11(1):7–14
Cinbis RG, Verbeek J, Schmid C (2013) Segmentation driven object detection with fisher vectors 2013. IEEE international conference on computer vision
Dorko S (2003) Selection of scale-invariant parts for object class recognition. IEEE Int Conf Comput Vis:634–6391
Fang J, Zhou Y, Yu Y, Du S (2017) Fine-grained vehicle model recognition using a coarse-to-fine convolutional neural network architecture. IEEE Trans Intell Transp Syst 18(7):1782–1792
Fraz M, Edirisinghe EA, Sarfraz MS (2014) Mid-level-representation based lexicon for vehicle make and model recognition. In: 2014 22nd International conference on pattern recognition, pp 393–398
Gholamhosseinian A, Seitz J (2021) Vehicle classification in intelligent transport systems: An overview, methods and software perspective. IEEE Open J Intell Trans Syst
Goddard M (2017) The eu general data protection regulation (gdpr): European regulation that has a global impact. Int J Mark Res 59(6):703–705
Gururaj H, Janhavi V, Tanuja U, Flamini F, Soundarya B, Ravi VK (2022) Predicting traffic accidents and their injury severities using machine learning techniques. Int J Trans Dev Integr 6(4):363–377
Hammouche R, Attia A, Akhrouf S, Akhtar Z (2022) Gabor filter bank with deep autoencoder based face recognition system. Expert Syst Appl:116743
Han K, Guo J, Zhang C, Zhu M (2018) Attribute-aware attention model for fine-grained representation learning. In: Proceedings of the 26th ACM international conference on multimedia, pp 2040–2048
He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: 2016 IEEE conference on computer vision and pattern recognition (CVPR). IEEE, pp 770–778
Hernandez-Olivan C, Beltran JR, Diaz-Guerra D (2020) Music boundary detection using convolutional neural networks: a comparative analysis of combined input features. arXiv:2008.07527
Howard AG, Zhu M, Chen B, Kalenichenko D, Wang W, Weyand T, Andreetto M, Adam H (2017) Mobilenets: efficient convolutional neural networks for mobile vision applications. arXiv:1704.04861
Hu W, Cheng F (2021) Application research of urban subway traffic mode based on behavior entropy in the background of big data. J High Speed Netw 27(3):291–304
Hu Q, Wang H, Li T, Shen C (2017) Deep cnns with spatially weighted pooling for fine-grained car recognition. IEEE Trans Intell Transp Syst 18 (11):3147–3156
Huang C, Huang Y (2021) Urban rail transit signal and control based on internet of things. J High Speed Netw 27(3):237–250
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ (2016) Densely connected convolutional networks. arXiv e-prints
Iandola FN, Han S, Moskewicz MW, Ashraf K, Dally WJ, Keutzer K (2016) Squeezenet: alexnet-level accuracy with 50x fewer parameters and 0.5 mb model size. arXiv:1602.07360
Kang J, Li X, Nie J, Liu Y, Xu M, Xiong Z, Niyato D, Yan Q (2022) Communication-efficient and cross-chain empowered federated learning for artificial intelligence of things. IEEE Trans Netw Sci Eng 9(5):2966–2977
Khattak MI, Al-Hasan M, Jan A, Saleem N, Verdu E, Khurshid N (2021) Automated detection of covid-19 using chest x-ray images and ct scans through multilayer-spatial convolutional neural networks
Kim J, Oh J, Heo T-Y (2021) Acoustic classification of mosquitoes using convolutional neural networks combined with activity circadian rhythm information
Kong X, Gao H, Shen G, Duan G, Das SK (2022) Fedvcp: a federated-learning-based cooperative positioning scheme for social internet of vehicles. IEEE Trans Computat Social Syst 9(1):197–206
Kong X, Wang K, Hou M, Hao X, Shen G, Chen X, Xia F (2021) A federated learning-based license plate recognition scheme for 5g-enabled internet of vehicles. IEEE Trans Industr Inf 17(12):8523–8530
Krause J, Stark M, Deng J, Fei-fei L (2013) 3d object representations for fine-grained categorization. In: Proceedings of the IEEE international conference on computer vision (ICCV) workshops
Krizhevsky A, Sutskever I, Hinton GE (2012) Imagenet classification with deep convolutional neural networks. In: Advances in neural information processing systems. Curran Associates, Inc, vol 25
Laishram A, Thongam K (2022) Automatic classification of oral pathologies using orthopantomogram radiography images based on convolutional neural network
Li Y (2022) Research and application of deep learning in image recognition. In: 2022 IEEE 2nd International conference on power, electronics and computer applications (ICPECA). IEEE, pp 994–999
Li Y, Deng J, Wu Q, Wang Y (2021) Eye-tracking signals based affective classification employing deep gradient convolutional neural networks
Li X, Yu L, Chang D, Ma Z, Cao J (2019) Dual cross-entropy loss for small-sample fine-grained vehicle classification. IEEE Trans Veh Technol 68(5):4204–4212
Liang W, Li Y, Xie K, Zhang D, Li K-C, Souri A, Li K (2022) Spatial-temporal aware inductive graph neural network for c-its data recovery. IEEE Trans Intell Transp Syst:1–12
Liu Y, Yu JJQ, Kang J, Niyato D, Zhang S (2020) Privacy-preserving traffic flow prediction: a federated learning approach. IEEE Internet Things J 7(8):7751–7763
Llorca DF, Arroyo R, Sotelo MA (2013) Vehicle logo recognition in traffic images using hog features and svm. In: 16th International IEEE conference on intelligent transportation systems (ITSC 2013). IEEE, pp 2229–2234
Lu L, Wang P, Cao Y (2022) A novel part-level feature extraction method for fine-grained vehicle recognition. Pattern Recogn 131:108869
Ma Z, Chang D, Xie J, Ding Y, Wen S, Li X, Si Z, Guo J (2019) Fine-grained vehicle classification with channel max pooling modified cnns. IEEE Trans Veh Technol 68(4):3224–3233
Ma N, Zhang X, Sun J (2020) Funnel activation for visual recognition. In: European conference on computer vision. Springer, pp 351–368
Ma N, Zhang X, Zheng H-T, Sun J (2018) Shufflenet v2: practical guidelines for efficient cnn architecture design. In: Proceedings of the european conference on computer vision (ECCV), pp 116–131
McMahan B, Moore E, Ramage D, Hampson S, Y Arcas BA (2017) Communication-efficient learning of deep networks from decentralized data. In: Artificial intelligence and statistics. PMLR, pp 1273–1282
Mohammadi R, Javidan R, Rikhtegar N, Keshtgari M (2020) An intelligent multicast traffic engineering method over software defined networks. J High Speed Netw 26(1):77–88
Peng Y, Yan Y, Zhu W, Zhao J (2014) Vehicle classification using sparse coding and spatial pyramid matching. In: 17th International IEEE conference on intelligent transportation systems (ITSC). IEEE, pp 259–263
Qi C, Su F (2017) Contrastive-center loss for deep neural networks. In: 2017 IEEE international conference on image processing (ICIP). IEEE, pp 2851–2855
Sandler M, Howard A, Zhu M, Zhmoginov A, Chen L-C (2018) Mobilenetv2: inverted residuals and linear bottlenecks. arXiv e-prints
Siddiqui AJ, Mammeri A, Boukerche A (2016) Real-time vehicle make and model recognition based on a bag of surf features. IEEE Trans Intell Trans Syst 17(11):3205–3219. IEEE
Silva B, Barbosa-Anda FR, Batista J (2021) Multi-view fine-grained vehicle classification with multi-loss learning. In: 2021 IEEE international conference on autonomous robot systems and competitions (ICARSC), pp 209–214
Simonyan K, Zisserman A (2014) Very deep convolutional networks for large-scale image recognition. arXiv e-prints
Souri A, Hussien A, Hoseyninezhad M, Norouzi M (2022) A systematic review of iot communication strategies for an efficient smart environment. Trans Emerging Telecommun Technol 33(3):3736
Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Erhan D, Vanhoucke V, Rabinovich A (2015) Going deeper with convolutions. In: 2015 IEEE conference on computer vision and pattern recognition (CVPR). IEEE, pp 1–9
Wang H, Peng J, Chen D, Jiang G, Zhao T, Fu X (2020) Attribute-guided feature learning network for vehicle reidentification. IEEE MultiMed 27 (4):112–121
Wang H, Peng J, Jiang G, Xu F, Fu X (2021) Discriminative feature and dictionary learning with part-aware model for vehicle re-identification. Neurocomputing 438:55–62
Wen Y, Zhang K, Li Z, Qiao Y (2016) A discriminative feature learning approach for deep face recognition. In: European conference on computer vision. Springer, pp 499–515
Woo S, Park J, Lee J-Y, Kweon IS (2018) CBAM: convolutional block attention module. arXiv E-prints
Xiang Y, Fu Y, Huang H (2020) Global topology constraint network for fine-grained vehicle recognition. IEEE Trans Intell Transp Syst 21 (7):2918–2929
Yang L, Luo P, Loy CC, Tang X (2015) A large-scale car dataset for fine-grained categorization and verification. In: 2015 IEEE conference on computer vision and pattern recognition (CVPR). IEEE, pp 3973–3981
Zhang C, Zhang S, Yu JJQ, Yu S (2021) Fastgnn: a topological information protected federated learning approach for traffic speed forecasting. IEEE Trans Industr Inf 17(12):8464–8474
Zhang B, Zhao C (2011) Classification of vehicle make by combined features and random subspace ensemble. In: 2011 Sixth international conference on image and graphics. IEEE, pp 920–925
Zhang X, Zhou X, Lin M, Sun J (2018) Shufflenet: an extremely efficient convolutional neural network for mobile devices. In: Proceedings of the IEEE conference on computer vision and pattern recognition, pp 6848–6856
Zhang Q, Zhuo L, Zhang S, Li J, Zhang H, Li X (2018) Fine-grained vehicle recognition using lightweight convolutional neural network with combined learning strategy. In: 2018 IEEE fourth international conference on multimedia big data (BigMM). IEEE, pp 1–5
Zhu W, Yu S, Zheng X, Wu Y (2019) Fine-grained vehicle classification technology based on fusion of multi-convolutional neural networks. Sensors Mater 31(2):569–578
Zou Z, Shi Z, Guo Y, Ye J (2019) Object detection in 20 years: a survey. arXiv:1905.05055
Funding
This work was supported by National Natural Science Foundation of China (U1936213), Shanghai Rising-Star Program (No. 22QA1403800), Program of Shanghai Academic Research Leader (No.21XD1421500), and the Open Research Fund of Key Laboratory of Cryptography of Zhejiang Province.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of Interests
The authors declare that they do not have any commercial or associative interest that represents a conflict of interest in connection with the work submitted.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Zeng, J., Zhang, K., Wang, L. et al. FedLVR: a federated learning-based fine-grained vehicle recognition scheme in intelligent traffic system. Multimed Tools Appl 82, 37431–37452 (2023). https://doi.org/10.1007/s11042-023-15004-w
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11042-023-15004-w