Download pdf
Federated learning-based YOLOv8 for face detection
- 1 University of Birmingham,
* Author to whom correspondence should be addressed.
Abstract
Recognizing the paramount importance of face detection in the realm of computer vision, there is an urgent need to address the vital concern of protecting individuals' privacy. Face detection inherently involves the handling of extremely sensitive personal information. To tackle this challenge, this study puts forth a proposal to incorporate Federated Learning into the face detection model. The objective is to maintain data localization and enhance security throughout the experiments by harnessing the decentralized nature of collaborative learning. The experimental procedure for Federated learning in face recognition models encompasses several key steps: device selection, global model initialization, model distribution to devices, local training, local model updates, model aggregation, global model updates, and multiple iterations. This methodology enables the collective training of models by dispersed devices, hence enhancing recognition performance, all the while ensuring the preservation of user data privacy. In addition, it is imperative to integrate Federated learning with YOLOv8 in order to establish a distributed target detection system. This method entails numerous devices engaging in local YOLOv8 model training, hence safeguarding data privacy and minimising data transmission. The empirical findings indicate that the use of joint learning in the face detection model leads to successful identification of the face model. In the future, there will be a consideration of novel federated learning algorithms with the aim of enhancing privacy.
Keywords
Federated Learning, Face Detection, Yolov8
[1]. Kumari Sirivarshitha A Sravani K Priya K S and Bhavani V 2023 An approach for Face Detection and Face Recognition using OpenCV and Face Recognition Libraries in Python (India: Coimbatore) pp1274-1278.
[2]. Kim J Park T Kim H and Kim S 2021 Federated Learning for Face Recognition (USA:NV) pp. 1-2.
[3]. Ren J Zhou J Lyu Y and Liu L 2022 Accelerated Federated Learning with Decoupled Adaptive Optimization.
[4]. Lai W and Yan Q 2022 Federated Learning for Detecting COVID-19 in Chest CT Images: A Lightweight Federated Learning Approach ( China:Qingdao) pp. 146-149.
[5]. Kaggle 2023 Human faces object detection https://www.kaggle.com/datasets/sbaghbidi/human-faces-object-detection.
[6]. Zhou Y Zhu W He Y Li Y 2023 YOLOv8-based Spatial Target Part Recognition (China : Chongqing2023) pp. 1684-1687.
[7]. Li Q Ge Z Wang F Luo X Yang Y Yu T 2022 Small Target Repair Parts Detection Algorithm Based on Improved YOLOv5 (China :Guilin) pp 420-433.
[8]. Mahendru M Dubey S K 2021 Real Time Object Detection with Audio Feedback using Yolo vs. Yolo_v3(India: Noida) pp. 734-740,
[9]. Liu Y Huang A and Luo Y 2020 Fedvision: An online visual object detection platform powered by federated learning.
[10]. Niu Y and Deng W 2022 Federated Learning for Face Recognition with Gradient Correction pp. 1999-07.
[11]. Qiu Y Wang J Jin Z Chen H Zhang M and Guo L 2022 Pose-guided matching based on deep learning for assessing quality of action on rehabilitation training Biomedical Signal Processing and Control 72 103323.
[12]. Shao H Li W Cai B Wan J Xiao Y and Yan S 2023 Dual-threshold attention-guided GAN and limited infrared thermal images for rotating machinery fault diagnosis under speed fluctuation IEEE Transactions on Industrial Informatics.
Cite this article
Peng,R. (2024). Federated learning-based YOLOv8 for face detection. Applied and Computational Engineering,54,14-20.
Data availability
The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.
Disclaimer/Publisher's Note
The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.
About volume
Volume title: Proceedings of the 4th International Conference on Signal Processing and Machine Learning
© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license. Authors who
publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons
Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this
series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published
version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial
publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and
during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See
Open access policy for details).