Composite federated learning with heterogeneous data