Privacy-preserving constrained domain generalization via gradient alignment

CX Tian, H Li, Y Wang, S Wang - IEEE Transactions on …, 2023 - ieeexplore.ieee.org
CX Tian, H Li, Y Wang, S Wang
IEEE Transactions on Knowledge and Data Engineering, 2023ieeexplore.ieee.org
Deep neural networks (DNN) have demonstrated unprecedented success for various
applications. However, due to the issue of limited dataset availability and the strict legal and
ethical requirements for data privacy protection, the broad applications of DNN (eg, medical
imaging classification) with large-scale training data have been largely hindered, greatly
constraining the model generalization capability. In this paper, we aim to tackle this problem
by developing the privacy-preserving constrained domain generalization method, aiming to …
Deep neural networks (DNN) have demonstrated unprecedented success for various applications. However, due to the issue of limited dataset availability and the strict legal and ethical requirements for data privacy protection, the broad applications of DNN (e.g., medical imaging classification) with large-scale training data have been largely hindered, greatly constraining the model generalization capability. In this paper, we aim to tackle this problem by developing the privacy-preserving constrained domain generalization method, aiming to improve the generalization capability under the privacy-preserving condition. In particular, we propose to improve the information aggregation process on the centralized server side with a novel gradient alignment loss, expecting that the trained model can be better generalized to the “unseen” but related data. The rationale and effectiveness of our proposed method can be explained by connecting our proposed method with the Maximum Mean Discrepancy (MMD) which has been widely adopted as the distribution distance measure. Experimental results on three domain generalization benchmark datasets indicate that our method can achieve better cross-domain generalization capability compared to the state-of-the-art federated learning methods.
ieeexplore.ieee.org
顯示最佳搜尋結果。 查看所有結果