Wang, Bokun

15 publications

ICML 2025 A Near-Optimal Single-Loop Stochastic Algorithm for Convex Finite-Sum Coupled Compositional Optimization Bokun Wang, Tianbao Yang

ICML 2025 Discovering Global False Negatives on the Fly for Self-Supervised Contrastive Learning Vicente Balmaseda, Bokun Wang, Ching-Long Lin, Tianbao Yang

ICLR 2025 On Discriminative Probabilistic Modeling for Self-Supervised Representation Learning Bokun Wang, Yunwen Lei, Yiming Ying, Tianbao Yang

NeurIPS 2025 Stochastic Momentum Methods for Non-Smooth Non-Convex Finite-Sum Coupled Compositional Optimization Xingyu Chen, Bokun Wang, Ming Yang, Qihang Lin, Tianbao Yang

TMLR 2025 Stochastic Primal-Dual Double Block-Coordinate for Two- Way Partial AUC Maximization Linli Zhou, Bokun Wang, My T. Thai, Tianbao Yang

NeurIPSW 2024 On Discriminative Probabilistic Modeling for Self-Supervised Representation Learning Bokun Wang, Yunwen Lei, Yiming Ying, Tianbao Yang

JMLR 2023 Memory-Based Optimization Methods for Model-Agnostic Meta-Learning and Personalized Federated Learning Bokun Wang, Zhuoning Yuan, Yiming Ying, Tianbao Yang

ICML 2023 Provable Multi-Instance Deep AUC Maximization with Stochastic Pooling Dixian Zhu, Bokun Wang, Zhi Chen, Yaxing Wang, Milan Sonka, Xiaodong Wu, Tianbao Yang

ICML 2022 Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applications Bokun Wang, Tianbao Yang

ICML 2022 GraphFM: Improving Large-Scale GNN Training via Feature Momentum Haiyang Yu, Limei Wang, Bokun Wang, Meng Liu, Tianbao Yang, Shuiwang Ji

ICLR 2022 IntSGD: Adaptive Floatless Compression of Stochastic Gradients Konstantin Mishchenko, Bokun Wang, Dmitry Kovalev, Peter Richtárik

ICML 2022 Optimal Algorithms for Stochastic Multi-Level Compositional Optimization Wei Jiang, Bokun Wang, Yibo Wang, Lijun Zhang, Tianbao Yang

JMLR 2022 Riemannian Stochastic Proximal Gradient Methods for Nonsmooth Optimization over the Stiefel Manifold Bokun Wang, Shiqian Ma, Lingzhou Xue

NeurIPS 2022 Theoretically Better and Numerically Faster Distributed Optimization with Smoothness-Aware Quantization Techniques Bokun Wang, Mher Safaryan, Peter Richtarik

ICML 2022 When AUC Meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guarantee Dixian Zhu, Gang Li, Bokun Wang, Xiaodong Wu, Tianbao Yang