A proximal forward-backward splitting based algorithmic framework for Wasserstein logistic regression using heavy ball strategy

In this paper, a forward-backward splitting based algorithmic framework incorporating the heavy ball strategy is proposed so as to efficiently solve the Wasserstein logistic regression problem. The proposed algorithmic framework consists two phases: the first phase involves a gradient descent step extension method, whilst the second phase involves a problem of instantaneous optimisation which balances the minimisation of a regularisation term while maintaining close proximity to the interim state given in the first phase. Then, it proves that the proposed algorithmic framework converges to the optimal solution of the Wasserstein logistic regression problem. Finally, numerical experiments are conducted, which illustrate the efficient implementation for high-dimensional sparsity data. The numerical results demonstrate that the proposed algorithmic framework outperforms not only the off-the-shelf solvers, but also some existing first-order algorithms.