Fauzi Muhammad Ali, Yang Bian, Blobel Bernd
Department of Information Security and Communication Technology, Norwegian University of Science and Technology (NTNU), 2815 Gjøvik, Norway.
Medical Faculty, University of Regensburg, 93053 Regensburg, Germany.
J Pers Med. 2022 Sep 26;12(10):1584. doi: 10.3390/jpm12101584.
Machine learning has been proven to provide good performances on stress detection tasks using multi-modal sensor data from a smartwatch. Generally, machine learning techniques need a sufficient amount of data to train a robust model. Thus, we need to collect data from several users and send them to a central server to feed the algorithm. However, the uploaded data may contain sensitive information that can jeopardize the user's privacy. Federated learning can tackle this challenge by enabling the model to be trained using data from all users without the user's data leaving the user's device. In this study, we implement federated learning-based stress detection and provide a comparative analysis between individual, centralized, and federated learning. The experiment was conducted on WESAD dataset by using Logistic Regression as the classifier. The experiment results show that in terms of accuracy, federated learning cannot reach the performance level of both individual and centralized learning. The individual learning strategy performs best with an average accuracy of 0.9998 and an average F-measure of 0.9996.
机器学习已被证明在使用来自智能手表的多模态传感器数据进行压力检测任务时表现良好。一般来说,机器学习技术需要足够的数据量来训练一个强大的模型。因此,我们需要从多个用户那里收集数据并将其发送到中央服务器以输入算法。然而,上传的数据可能包含敏感信息,这可能会危及用户的隐私。联邦学习可以通过使模型能够使用来自所有用户的数据进行训练,而无需将用户数据离开用户设备来应对这一挑战。在本研究中,我们实现了基于联邦学习的压力检测,并对个体学习、集中式学习和联邦学习进行了比较分析。实验在WESAD数据集上使用逻辑回归作为分类器进行。实验结果表明,在准确性方面,联邦学习无法达到个体学习和集中式学习的性能水平。个体学习策略表现最佳,平均准确率为0.9998,平均F值为0.9996。