TCT

Optimization framework

An approach to train and optimize machine learning models in a decentralized setting by convexifying the optimization process

TCT: Convexifying Federated Learning using Bootstrapped Neural Tangent Kernels

GitHub

4 stars
6 watching
3 forks
Language: Python
last commit: over 3 years ago
federated-learningneural-tangent-kerneloptimization

Related projects:

RepositoryDescriptionStars
xidongwu/federated-minimax-and-conditional-stochastic-optimizationThis project presents optimization techniques for federated learning and minimax games in the context of machine learning1
jonfanlab/glonetA software framework for training neural networks to optimize dielectric metasurfaces using physics-driven generative models and global optimization algorithms.101
kai-yue/ntk-fedA framework for federated learning that leverages the neural tangent kernel to address statistical heterogeneity in distributed machine learning.3
jiangoforit/yellowfin_pytorchAn optimizer that automatically tunes momentum and learning rate using local quadratic approximation.287
hui-po-wang/progfedAn approach to efficient federated learning by progressively training models on client devices with reduced communication and computation requirements.20
jungtaekkim/bayesoA framework for optimizing hyperparameters in machine learning models using Bayesian optimization93
alshedivat/fedpaA modular JAX implementation of federated learning via posterior averaging for decentralized optimization50
guopengf/auto-fedrlA reinforcement learning-based framework for optimizing hyperparameters in distributed machine learning environments.15
liboyue/beerA collection of numerical experiments and code for demonstrating the performance of decentralized nonconvex optimization methods8
deng-cy/deep_learning_topology_optOptimization code for topology optimization problems using machine learning and deep learning techniques107
yh-yao/fedgcnAn implementation of a federated learning algorithm for training Graph Convolutional Networks on semi-supervised node classification tasks.59
qinbinli/moonA framework for collaborative machine learning model training that leverages similarity between model representations to correct local training.267
zackzikaixiao/fedgrabA tool for training federated learning models with adaptive gradient balancing to handle class imbalance in multi-client scenarios.14
fpicetti/occamypyA library for solving large-scale optimization problems with flexible and scalable vector and operator definitions55
hongyouc/fed-rodDevelops a framework to balance competing goals in federated learning by decoupling generic and personalized prediction tasks.14