Improved Generalization Bounds for Communication Efficient Federated Learning