Large-Scale Nonlinear AUC Maximization via Triply Stochastic Gradients.

Learning to improve AUC performance for imbalanced data is an important machine learning research problem. Most methods of AUC maximization assume that the model function is linear in the original feature space. However, this assumption is not suitable for nonlinear separable problems. Although there have been some nonlinear methods of AUC maximization, scaling up nonlinear AUC maximization is still an open question. To address this challenging problem, in this paper, we propose a novel large-scale nonlinear AUC maximization method (named as TSAM) based on the triply stochastic gradient descents. Specifically, we first use the random Fourier feature to approximate the kernel function. After that, we use the triply stochastic gradients w.r.t. the pairwise loss and random feature to iteratively update the solution. Finally, we prove that TSAM converges to the optimal solution with the rate of O(1/t) after t iterations. Experimental results on a variety of benchmark datasets not only confirm the scalability of TSAM, but also show a significant reduction of computational time compared with existing batch learning algorithms, while retaining the similar generalization performance.

Citation

Zhiyuan Dang, Xiang Li, Bin Gu, Cheng Deng, Heng Huang. Large-Scale Nonlinear AUC Maximization via Triply Stochastic Gradients. IEEE transactions on pattern analysis and machine intelligence. 2022 Mar;44(3):1385-1398

PMID: 32946382

search → result