├── CODE-OF-CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── alternating_loop.py ├── data_word_prediction.py ├── dp_ftrl ├── BUILD ├── README.md ├── __init__.py ├── dp_fedavg.py ├── dp_fedavg_test.py ├── optimizer_utils.py ├── optimizer_utils_test.py ├── run_emnist.py ├── run_stackoverflow.py ├── training_loop.py └── training_loop_test.py ├── language_models.py ├── metrics.py ├── mime.py ├── mime_loop.py ├── mimelite.py ├── mimelite_loop.py ├── mirror_descent.py ├── mirror_descent_convex.py ├── mirror_descent_convex_loop_v2.py ├── mirror_descent_convex_loop_v3.py ├── mirror_descent_convex_v2.py ├── mirror_descent_convex_v3.py ├── mirror_descent_loop.py ├── requirements.txt ├── run.sh ├── run_dpfl.py ├── scaffold_loop.py ├── scaffold_v2.py ├── utils ├── BUILD ├── README.md ├── datasets │ ├── BUILD │ ├── README.md │ ├── cifar100_dataset.py │ ├── cifar100_dataset_test.py │ ├── cifar10_dataset.py │ ├── cifar10_dataset_test.py │ ├── emnist_dataset.py │ ├── emnist_dataset_test.py │ ├── infinite_emnist.py │ ├── infinite_emnist_test.py │ ├── shakespeare_dataset.py │ ├── shakespeare_dataset_test.py │ ├── stackoverflow_tag_prediction.py │ ├── stackoverflow_tag_prediction_test.py │ ├── stackoverflow_word_prediction.py │ └── stackoverflow_word_prediction_test.py ├── keras_metrics.py ├── keras_metrics_test.py ├── models │ ├── BUILD │ ├── README.md │ ├── emnist_ae_models.py │ ├── emnist_ae_models_test.py │ ├── emnist_models.py │ ├── emnist_models_test.py │ ├── resnet_models.py │ ├── resnet_models_test.py │ ├── shakespeare_models.py │ ├── shakespeare_models_test.py │ ├── stackoverflow_lr_models.py │ ├── stackoverflow_lr_models_test.py │ ├── stackoverflow_models.py │ └── stackoverflow_models_test.py ├── optimizers │ ├── BUILD │ ├── lars.py │ ├── lars_test.py │ ├── optimizer_utils.py │ ├── optimizer_utils_test.py │ ├── shampoo.py │ ├── shampoo_test.py │ ├── yogi.py │ └── yogi_test.py ├── task_utils.py ├── task_utils_test.py ├── tensor_utils.py ├── tensor_utils_test.py ├── training_utils.py ├── training_utils_test.py ├── utils_impl.py └── utils_impl_test.py └── warmstart_loop.py /CODE-OF-CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/CODE-OF-CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/README.md -------------------------------------------------------------------------------- /alternating_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/alternating_loop.py -------------------------------------------------------------------------------- /data_word_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/data_word_prediction.py -------------------------------------------------------------------------------- /dp_ftrl/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/BUILD -------------------------------------------------------------------------------- /dp_ftrl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/README.md -------------------------------------------------------------------------------- /dp_ftrl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dp_ftrl/dp_fedavg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/dp_fedavg.py -------------------------------------------------------------------------------- /dp_ftrl/dp_fedavg_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/dp_fedavg_test.py -------------------------------------------------------------------------------- /dp_ftrl/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/optimizer_utils.py -------------------------------------------------------------------------------- /dp_ftrl/optimizer_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/optimizer_utils_test.py -------------------------------------------------------------------------------- /dp_ftrl/run_emnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/run_emnist.py -------------------------------------------------------------------------------- /dp_ftrl/run_stackoverflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/run_stackoverflow.py -------------------------------------------------------------------------------- /dp_ftrl/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/training_loop.py -------------------------------------------------------------------------------- /dp_ftrl/training_loop_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/dp_ftrl/training_loop_test.py -------------------------------------------------------------------------------- /language_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/language_models.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/metrics.py -------------------------------------------------------------------------------- /mime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mime.py -------------------------------------------------------------------------------- /mime_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mime_loop.py -------------------------------------------------------------------------------- /mimelite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mimelite.py -------------------------------------------------------------------------------- /mimelite_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mimelite_loop.py -------------------------------------------------------------------------------- /mirror_descent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent.py -------------------------------------------------------------------------------- /mirror_descent_convex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_convex.py -------------------------------------------------------------------------------- /mirror_descent_convex_loop_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_convex_loop_v2.py -------------------------------------------------------------------------------- /mirror_descent_convex_loop_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_convex_loop_v3.py -------------------------------------------------------------------------------- /mirror_descent_convex_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_convex_v2.py -------------------------------------------------------------------------------- /mirror_descent_convex_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_convex_v3.py -------------------------------------------------------------------------------- /mirror_descent_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/mirror_descent_loop.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/run.sh -------------------------------------------------------------------------------- /run_dpfl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/run_dpfl.py -------------------------------------------------------------------------------- /scaffold_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/scaffold_loop.py -------------------------------------------------------------------------------- /scaffold_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/scaffold_v2.py -------------------------------------------------------------------------------- /utils/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/BUILD -------------------------------------------------------------------------------- /utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/README.md -------------------------------------------------------------------------------- /utils/datasets/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/BUILD -------------------------------------------------------------------------------- /utils/datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/README.md -------------------------------------------------------------------------------- /utils/datasets/cifar100_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/cifar100_dataset.py -------------------------------------------------------------------------------- /utils/datasets/cifar100_dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/cifar100_dataset_test.py -------------------------------------------------------------------------------- /utils/datasets/cifar10_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/cifar10_dataset.py -------------------------------------------------------------------------------- /utils/datasets/cifar10_dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/cifar10_dataset_test.py -------------------------------------------------------------------------------- /utils/datasets/emnist_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/emnist_dataset.py -------------------------------------------------------------------------------- /utils/datasets/emnist_dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/emnist_dataset_test.py -------------------------------------------------------------------------------- /utils/datasets/infinite_emnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/infinite_emnist.py -------------------------------------------------------------------------------- /utils/datasets/infinite_emnist_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/infinite_emnist_test.py -------------------------------------------------------------------------------- /utils/datasets/shakespeare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/shakespeare_dataset.py -------------------------------------------------------------------------------- /utils/datasets/shakespeare_dataset_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/shakespeare_dataset_test.py -------------------------------------------------------------------------------- /utils/datasets/stackoverflow_tag_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/stackoverflow_tag_prediction.py -------------------------------------------------------------------------------- /utils/datasets/stackoverflow_tag_prediction_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/stackoverflow_tag_prediction_test.py -------------------------------------------------------------------------------- /utils/datasets/stackoverflow_word_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/stackoverflow_word_prediction.py -------------------------------------------------------------------------------- /utils/datasets/stackoverflow_word_prediction_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/datasets/stackoverflow_word_prediction_test.py -------------------------------------------------------------------------------- /utils/keras_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/keras_metrics.py -------------------------------------------------------------------------------- /utils/keras_metrics_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/keras_metrics_test.py -------------------------------------------------------------------------------- /utils/models/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/BUILD -------------------------------------------------------------------------------- /utils/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/README.md -------------------------------------------------------------------------------- /utils/models/emnist_ae_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/emnist_ae_models.py -------------------------------------------------------------------------------- /utils/models/emnist_ae_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/emnist_ae_models_test.py -------------------------------------------------------------------------------- /utils/models/emnist_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/emnist_models.py -------------------------------------------------------------------------------- /utils/models/emnist_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/emnist_models_test.py -------------------------------------------------------------------------------- /utils/models/resnet_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/resnet_models.py -------------------------------------------------------------------------------- /utils/models/resnet_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/resnet_models_test.py -------------------------------------------------------------------------------- /utils/models/shakespeare_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/shakespeare_models.py -------------------------------------------------------------------------------- /utils/models/shakespeare_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/shakespeare_models_test.py -------------------------------------------------------------------------------- /utils/models/stackoverflow_lr_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/stackoverflow_lr_models.py -------------------------------------------------------------------------------- /utils/models/stackoverflow_lr_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/stackoverflow_lr_models_test.py -------------------------------------------------------------------------------- /utils/models/stackoverflow_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/stackoverflow_models.py -------------------------------------------------------------------------------- /utils/models/stackoverflow_models_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/models/stackoverflow_models_test.py -------------------------------------------------------------------------------- /utils/optimizers/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/BUILD -------------------------------------------------------------------------------- /utils/optimizers/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/lars.py -------------------------------------------------------------------------------- /utils/optimizers/lars_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/lars_test.py -------------------------------------------------------------------------------- /utils/optimizers/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/optimizer_utils.py -------------------------------------------------------------------------------- /utils/optimizers/optimizer_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/optimizer_utils_test.py -------------------------------------------------------------------------------- /utils/optimizers/shampoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/shampoo.py -------------------------------------------------------------------------------- /utils/optimizers/shampoo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/shampoo_test.py -------------------------------------------------------------------------------- /utils/optimizers/yogi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/yogi.py -------------------------------------------------------------------------------- /utils/optimizers/yogi_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/optimizers/yogi_test.py -------------------------------------------------------------------------------- /utils/task_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/task_utils.py -------------------------------------------------------------------------------- /utils/task_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/task_utils_test.py -------------------------------------------------------------------------------- /utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/tensor_utils.py -------------------------------------------------------------------------------- /utils/tensor_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/tensor_utils_test.py -------------------------------------------------------------------------------- /utils/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/training_utils.py -------------------------------------------------------------------------------- /utils/training_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/training_utils_test.py -------------------------------------------------------------------------------- /utils/utils_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/utils_impl.py -------------------------------------------------------------------------------- /utils/utils_impl_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/utils/utils_impl_test.py -------------------------------------------------------------------------------- /warmstart_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-research/public-data-in-dpfl/HEAD/warmstart_loop.py --------------------------------------------------------------------------------