Federated Learning With Soft Clustering