search

FedML-AI / FedML

FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs on any GPU cloud or on-premise cluster. Built on this library, TensorOpera AI (https://TensorOpera.ai) is your generative AI platform at scale.
https://TensorOpera.ai
Apache License 2.0
4.19k stars 787 forks source link

Is there implements of vertical federated learning in FedML #229

Open lepangdan opened 2 years ago

lepangdan commented 2 years ago

hi， I am interested in vertical federated learning, it seems there aren't the implements of vertical federated learning. Will you plan to implement the vertical federated learning in distributed way.

chaoyanghe commented 2 years ago

@lepangdan VFL is always there:

https://github.com/FedML-AI/FedML/tree/master/python/fedml/simulation/mpi/classical_vertical_fl https://github.com/FedML-AI/FedML/tree/master/python/fedml/simulation/sp/classical_vertical_fl

for both SP and Distributed training with MPI.

AlexGrig commented 2 years ago

Hello, @chaoyanghe, thanks for your work,

The MPI example seems to be still incomplete https://github.com/FedML-AI/FedML/blob/d5718c031b04ab8c91003174c4fffccc166f3a44/python/fedml/simulation/simulator.py#L157

Please correct me if I am wrong, regards.