Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learning
Zhenhuan Yang, Yunwen Lei, Puyu Wang, Tianbao Yang, Yiming Ying
Abstract
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. * Equal contributions. The corresponding author is Yiming Ying.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0e4650d6-875f-4a91-b8e0-f079be2bc1eeCited by top-tier papers12
- Stability and Generalization for Markov Chain Stochastic Gradient MethodsPuyu Wang, Yunwen Lei, Yiming Ying, Ding-Xuan ZhouNeurIPS 2022 · 26 citations
- FeDXL: Provable Federated Learning for Deep X-Risk OptimizationZhishuai Guo, Rong Jin, Jiebo Luo, Tianbao YangICML 2023 · 11 citations
- Fine-Grained Theoretical Analysis of Federated Zeroth-Order OptimizationJun Chen, Hong Chen, Bin Gu, Hao DengNeurIPS 2023 · 11 citations
- Generalization Analysis of Stochastic Weight Averaging with General SamplingPeng Wang, Li Shen, Zerui Tao, Shuaida He et al.ICML 2024 · 6 citations
- On the Stability and Generalization of Triplet LearningJun Chen, Hong Chen, Xue Jiang, Bin Gu et al.AAAI 2023 · 5 citations
Builds on7
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Stability of Stochastic Gradient Descent on Nonsmooth Convex LossesRaef Bassily, Vitaly Feldman, Cristóbal Guzmán, Kunal TalwarNeurIPS 2020 · 240 citations
- Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient DescentYunwen Lei, Yiming YingICML 2020 · 165 citations
- Stochastic AUC Maximization with Deep Neural NetworksMingrui Liu, Zhuoning Yuan, Yiming Ying, Tianbao YangICLR 2020 · 118 citations
- Sharper Generalization Bounds for Pairwise LearningYunwen Lei, Antoine Ledent, Marius KloftNeurIPS 2020 · 50 citations
Related papers
- Generalization Guarantee of SGD for Pairwise LearningYunwen Lei, Mingrui Liu, Yiming YingNeurIPS 2021 · 37 citations
- Limited Memory Online Gradient Descent for Kernelized Pairwise Learning with Dynamic AveragingHilal AlQuabeh, William de Vazelhes, Bin GuAAAI 2024 · 1 citation
- Pairwise Learning with Differential Privacy GuaranteesMengdi Huai, Di Wang, Chenglin Miao, Jinhui Xu et al.AAAI 2020 · 30 citations
- Stability-based Generalization Analysis of Randomized Coordinate Descent for Pairwise LearningLiang Wu, Ruixi Hu, Yunwen LeiAAAI 2025
- Stability-Based Generalization Analysis for Mixtures of Pointwise and Pairwise LearningJiahuan Wang, Jun Chen, Hong Chen, Bin Gu et al.AAAI 2023 · 2 citations
