This directory contains source code for reproducting the results in the ICML publication Eichner, Koren, McMahan, Srebro, Talwar. Semi-Cyclic Stochastic Gradient Descent. While these experiments are highly relevant to the federated learning setting, we experiment with standard SGD (without partitioning the data among users), and so these experiments use vanilla TF rather than TFF.
Running the experiments
The experiments can be run with
run.sh, which will
- download the Sentiment140 dataset
- preprocess it
- train + evaluate models for the parameter settings used in the paper.
- This step will take several hours on a single machine.
- Each run persists its configuration and results to a separate log file