Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2410.13709 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866910889174630400 |
|---|---|
| author | Ahmed, Mustofa Muntakim, Abdul Tabassum, Nawrin Rahim, Mohammad Asifur Shah, Faisal Muhammad |
| author_facet | Ahmed, Mustofa Muntakim, Abdul Tabassum, Nawrin Rahim, Mohammad Asifur Shah, Faisal Muhammad |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_13709 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | On-device Federated Learning in Smartphones for Detecting Depression from Reddit Posts Ahmed, Mustofa Muntakim, Abdul Tabassum, Nawrin Rahim, Mohammad Asifur Shah, Faisal Muhammad Machine Learning 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. |
| title | On-device Federated Learning in Smartphones for Detecting Depression from Reddit Posts |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2410.13709 |