联邦指令微调(FedIT):让大模型在私有数据上协作学习

联邦指令微调(FedIT):让大模型在私有数据上协作学习

核心概念: 联邦指令微调(Federated Instruction Tuning)、LoRA 参数高效微调、FedAvg 聚合、数据异构性、GPT-4 自动评估

项目地址

https://gitcode.com/u012133341/fedit-project/tree/main

实战内容:

  • 完整解析 FedIT 论文的架构设计与核心发现
  • 从零实现 Shepherd 框架:联邦 LoRA 微调 + FedAvg 聚合
  • 复现论文实验:异构数据分区 → 多轮联邦训练 → GPT-4 评估
  • 可视化:客户端标签分布、通信轮次-评估得分曲线

技术栈: PyTorch + HuggingFace PEFT + LoRA + bitsandbytes
关键算法: FedIT(Zhang et al., ICASSP 2024)、LoRA、FedAvg
数据集: Databricks-Dolly-15k(8 个指令类别)
基座模型: LLaMA-7B / Alpaca-7B
硬件需求: 单卡 24GB 显存(TITAN RTX / RTX 30

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

VectorShift

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值