Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Dopamine: Differentially Private Federated Learning on Medical Data
(Submitted on 27 Jan 2021 (v1), last revised 29 Jan 2021 (this version, v2))
Abstract: While rich medical datasets are hosted in hospitals distributed across the world, concerns on patients' privacy is a barrier against using such data to train deep neural networks (DNNs) for medical diagnostics. We propose Dopamine, a system to train DNNs on distributed datasets, which employs federated learning (FL) with differentially-private stochastic gradient descent (DPSGD), and, in combination with secure aggregation, can establish a better trade-off between differential privacy (DP) guarantee and DNN's accuracy than other approaches. Results on a diabetic retinopathy~(DR) task show that Dopamine provides a DP guarantee close to the centralized training counterpart, while achieving a better classification accuracy than FL with parallel DP where DPSGD is applied without coordination. Code is available at this https URL
Submission history
From: Mohammad Malekzadeh [view email][v1] Wed, 27 Jan 2021 21:27:23 GMT (6502kb,D)
[v2] Fri, 29 Jan 2021 16:40:17 GMT (5434kb,D)
Link back to: arXiv, form interface, contact.