Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learning

About

Pairwise learning refers to learning tasks where the loss function depends on a pair of instances. It instantiates many important machine learning tasks such as bipartite ranking and metric learning. A popular approach to handle streaming data in pairwise learning is an online gradient descent (OGD) algorithm, where one needs to pair the current instance with a buffering set of previous instances with a sufficiently large size and therefore suffers from a scalability issue. In this paper, we propose simple stochastic and online gradient descent methods for pairwise learning. A notable difference from the existing studies is that we only pair the current instance with the previous one in building a gradient direction, which is efficient in both the storage and computational complexity. We develop novel stability results, optimization, and generalization error bounds for both convex and nonconvex as well as both smooth and nonsmooth problems. We introduce novel techniques to decouple the dependency of models and the previous instance in both the optimization and generalization analysis. Our study resolves an open question on developing meaningful generalization bounds for OGD using a buffering set with a very small fixed size. We also extend our algorithms and stability analysis to develop differentially private SGD algorithms for pairwise learning which significantly improves the existing results.

Zhenhuan Yang, Yunwen Lei, Puyu Wang, Tianbao Yang, Yiming Ying• 2021

Related benchmarks

TaskDatasetResultRank
AUC MaximizationGerman
AUC0.763
19
AUC MaximizationDiabetes (test)
AUC81.3
13
Binary Classificationdiabetes LIBSVM (test)
AUC0.831
7
Binary Classificationijcnn1 LIBSVM (test)
Average AUC0.934
7
Binary Classificationmnist LIBSVM (test)
Average AUC0.932
7
Binary Classificationusps LIBSVM (test)
Average AUC0.926
7
Binary Classificationletter LIBSVM (test)
Average AUC0.81
7
Binary Classificationgerman LIBSVM (test)
AUC0.793
7
AUC MaximizationDiabetes
AUC0.79
6
Pairwise LearningDiabetes (test)
Average AUC83.1
5
Showing 10 of 15 rows

Other info

Code

Follow for update