用手机联邦学习分析Reddit帖子,隐私安全地识别抑郁迹象。
On-device Federated Learning in Smartphones for Detecting Depression from Reddit Posts
- 在手机端用联邦学习训练GRU/RNN/LSTM模型,保护用户隐私。
- 联邦模型性能接近中心化模型,验证了去中心化可行性。
- 适合关注隐私保护与边缘计算的医疗AI研究者。
以往研究广泛探索利用深度学习模型从社交媒体文本中检测抑郁症,因社交平台数据量大且能反映用户心理健康状态。然而,分布式学习在此领域尚未充分应用。本文采用联邦学习(FL)在智能手机上实现去中心化训练,保护用户数据隐私。我们在Reddit帖子上训练GRU、RNN和LSTM三种神经网络架构,评估其在异构联邦环境下的表现。为优化训练过程,所有客户端使用统一分词器以降低计算开销。同时分析手机端资源消耗与通信成本,评估真实场景下的适用性。实验结果表明,联邦模型性能与集中式模型相当。本研究展示了联邦学习在去中心化心理健康预测中的潜力,提供了一种安全高效的边缘设备训练方案。
原文摘要 · Abstract (English)
Depression detection using deep learning models has been widely explored in previous studies, especially due to the large amounts of data available from social media posts. These posts provide valuable information about individuals' mental health conditions and can be leveraged to train models and identify patterns in the data. However, distributed learning approaches have not been extensively explored in this domain. In this study, we adopt Federated Learning (FL) to facilitate decentralized training on smartphones while protecting user data privacy. We train three neural network architectures--GRU, RNN, and LSTM on Reddit posts to detect signs of depression and evaluate their performance under heterogeneous FL settings. To optimize the training process, we leverage a common tokenizer across all client devices, which reduces the computational load. Additionally, we analyze resource consumption and communication costs on smartphones to assess their impact in a real-world FL environment. Our experimental results demonstrate that the federated models achieve comparable performance to the centralized models. This study highlights the potential of FL for decentralized mental health prediction by providing a secure and efficient model training process on edge devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。