arXiv:2411.05173cs.LG2024-11中稿 · SIMBig 2024被引 3

动态加权平均提升联邦学习精度,保护生物数据隐私

DWFL: Enhancing Federated Learning through Dynamic Weighted Averaging

  • 根据本地模型表现动态调整权重,优化全局模型初始化
  • 在真实蛋白序列数据集上准确率显著提升
  • 适合需要隐私保护的生物信息学联合建模场景

联邦学习是一种通过分布式训练保持数据隐私的机器学习方法,尤其适用于生物信息学领域中涉及患者数据时的隐私保护。尽管联邦学习已在生物序列分析中成功应用,但在提高准确性的同时保障隐私仍需深入探索,尤其是在蛋白质序列分析中的最优集成方式尚未充分研究。本文提出一种基于深度前馈神经网络的增强型联邦学习方法(DWFL),用于蛋白质序列分类。该方法引入动态加权联邦学习机制,依据本地模型性能指标对权重进行加权平均,使表现优异的模型获得更高权重,从而构建更优的初始全局模型,提升整体分类准确率。我们在真实世界蛋白质序列数据集上进行了实验,结果表明所提方法在模型准确率方面有显著提升,使联邦学习成为更可靠、更隐私安全的协作机器学习优选方案。

原文摘要 · Abstract (English)

Federated Learning (FL) is a distributed learning technique that maintains data privacy by providing a decentralized training method for machine learning models using distributed big data. This promising Federated Learning approach has also gained popularity in bioinformatics, where the privacy of biomedical data holds immense importance, especially when patient data is involved. Despite the successful implementation of Federated learning in biological sequence analysis, rigorous consideration is still required to improve accuracy in a way that data privacy should not be compromised. Additionally, the optimal integration of federated learning, especially in protein sequence analysis, has not been fully explored. We propose a deep feed-forward neural network-based enhanced federated learning method for protein sequence classification to overcome these challenges. Our method introduces novel enhancements to improve classification accuracy. We introduce dynamic weighted federated learning (DWFL) which is a federated learning-based approach, where local model weights are adjusted using weighted averaging based on their performance metrics. By assigning higher weights to well-performing models, we aim to create a more potent initial global model for the federated learning process, leading to improved accuracy. We conduct experiments using real-world protein sequence datasets to assess the effectiveness of DWFL. The results obtained using our proposed approach demonstrate significant improvements in model accuracy, making federated learning a preferred, more robust, and privacy-preserving approach for collaborative machine-learning tasks.

联邦学习蛋白质分类隐私保护动态加权

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。