Non-convex composite federated learning with heterogeneous data