Compare Where It Matters: Using Layer-Wise Regularization To Improve Federated Learning on Heterogeneous Data.