A Wasserstein Minimax Framework for Mixed Linear Regression

1 code implementation • 14 Jun 2021 • Theo Diamandis, Yonina C. Eldar, Alireza Fallah, Farzan Farnia, Asuman Ozdaglar

We propose an optimal transport-based framework for MLR problems, Wasserstein Mixed Linear Regression (WMLR), which minimizes the Wasserstein distance between the learned and target mixture regression models.

Federated Learning regression