This repository contains the codes for the paper
Communication-Efficient Distributed SVD via Local Power Iterations
If you find this code useful in your research, please consider citing:
@inproceedings{li2021communication,
title={Communication-efficient distributed SVD via local power iterations},
author={Li, Xiang and Wang, Shusen and Chen, Kun and Zhang, Zhihua},
booktitle={International Conference on Machine Learning},
pages={6504--6514},
year={2021},
organization={PMLR}
}
File name | Description |
---|---|
pm.py | Main functions for various algorithms |
one-shot.py | Code for Table 1 |
eta.py | Code for Table 3, 5 |
plot_p.py | Code for Figure 2 and 5-7 |
plot_decay.py | Code for Table 7 and Figure 8-10 |
plot_m.py | Code for Figure 11-12 |
plot_sin.py | Code for Figure 13 |