CRII: III: Towards Improving the Handling of Heterogeneity and Personalization in Federated Learning

CRII：III：改进联邦学习中异构性和个性化的处理

• 批准号: 2246067
• 负责人: Lichao Sun
• 金额: $ 17.5万
• 依托单位: Lehigh University
• 依托单位国家: 美国
• 项目类别: Standard Grant
• 财政年份: 2023
• 资助国家: 美国
• 起止时间: 2023-08-01 至 2025-07-31
• 项目状态: 未结题
• 来源: https://www.nsf.gov/awardsearch/showAward?AWD_ID=2246067&HistoricalAwards=false
• 关键词: CRII; III; Towards; Improving; Handling

As awareness of the need for privacy preservation continues to grow in society, new legal restrictions, such as the General Data Protection Regulation (GDPR), are emerging. Such laws demand that businesses and organizations not share their clients' raw data for any commercial purposes. Federated Learning (FL) is a distributed machine learning paradigm that works with decentralized data while preserving privacy. FL has gained widespread interest and has been applied in numerous applications, such as healthcare, education, and intelligent manufacturing. However, FL faces some challenges that come from, executing on diverse types of data and devices, such as mobile phones and Internet of Things (IoT) devices. This project aims to address the aforementioned issues in heterogeneous FL by developing mathematical models and efficient algorithms. In addition, the project will integrate trustworthy ML research into new curriculum development and support students from underrepresented groups.Heterogeneous FL faces two significant challenges: (1) each client in FL may generate data according to a distinct distribution; (2) heterogeneous clients, such as mobile phones and IoT devices, are equipped with a wide range of computation and communication capabilities. To address these challenges, this project will dramatically push the boundary of knowledge via the following two integrated research thrusts: (i) The research team aims to tackle data heterogeneity in FL by designing two advanced personalized learning methods. Specifically, the proposed solutions aim to balance the generalization ability from the global model and the personalization ability from the local model, improving both the global model and personalized local models. (ii) The team will study heterogeneous neural network aggregation for FL by providing advanced memory-efficient local training strategies for small devices. In addition, the project will make use of mutual knowledge distillation to improve the generalization ability of the local models. Finally, the team proposes a unified FL framework that integrates data-free knowledge aggregation with advanced memory-efficient solutions to tackle both heterogeneity issues simultaneously.This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.

随着社会对隐私保护需求的意识不断增强，新的法律限制，如《通用数据保护条例》（GDPR）正在出现。此类法律要求企业和组织不得出于任何商业目的共享其客户的原始数据。联邦学习（FL）是一种分布式机器学习范式，可以处理分散的数据，同时保护隐私。FL得到了广泛的关注，并在医疗保健、教育和智能制造等众多应用中得到了应用。然而，FL面临着一些挑战，这些挑战来自于在不同类型的数据和设备上执行，例如移动电话和物联网（IoT）设备。本项目旨在通过开发数学模型和高效算法来解决异构FL中的上述问题。此外，该项目将把值得信赖的机器学习研究整合到新课程开发中，并支持来自代表性不足群体的学生。异构FL面临两个重大挑战：(1)FL中的每个客户端可能根据不同的分布生成数据；(2)异构客户端，如手机和物联网设备，具有广泛的计算和通信能力。为了应对这些挑战，该项目将通过以下两个综合研究重点显著推动知识的边界：(i)研究团队旨在通过设计两种先进的个性化学习方法来解决FL中的数据异构问题。具体而言，所提出的解决方案旨在平衡全局模型的泛化能力和局部模型的个性化能力，同时改进全局模型和个性化的局部模型。（ii）该团队将通过为小型设备提供先进的内存高效本地训练策略，研究FL的异构神经网络聚合。此外，该项目将利用互知识精馏来提高局部模型的泛化能力。最后，该团队提出了一个统一的FL框架，该框架将无数据的知识聚合与先进的内存高效解决方案集成在一起，以同时解决这两个异构问题。该奖项反映了美国国家科学基金会的法定使命，并通过使用基金会的知识价值和更广泛的影响审查标准进行评估，被认为值得支持。