Skip to content
/ TCT Public

TCT: Convexifying Federated Learning using Bootstrapped Neural Tangent Kernels

Notifications You must be signed in to change notification settings

yaodongyu/TCT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 

Repository files navigation

Train-Convexify-Train (TCT)

This is the code for the NeurIPS2022 paper [link]:

TCT: Convexifying Federated Learning using Bootstrapped Neural Tangent Kernels

by Yaodong Yu, Alexander Wei, Sai Praneeth Karimireddy, Yi Ma, Michael I. Jordan from UC Berkeley.

Prerequisites

  • Python
  • Pytorch
  • CUDA
  • numpy

* Example on MNIST

Example code on MNIST can be found under mnist folder.

Reference

For more experimental and technical details, please check our paper. If you find this useful for your work, please consider citing

@article{yu2022tct,
  title={TCT: Convexifying Federated Learning using Bootstrapped Neural Tangent Kernels},
  author={Yu, Yaodong and Wei, Alexander and Karimireddy, Sai Praneeth and Ma, Yi and Jordan, Michael I},
  journal={arXiv preprint arXiv:2207.06343},
  year={2022}
}

About

TCT: Convexifying Federated Learning using Bootstrapped Neural Tangent Kernels

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages