11 citations · 46 across the 24 of their papers we have counts for
4 papers · 1 filter
GraphFM: Improving Large-Scale GNN Training via Feature Momentum
Haiyang Yu, Limei Wang, Bokun Wang +3
Training of graph neural networks (GNNs) for large-scale node classification is challenging. A key difficulty lies in obtaining accurate hidden node representations while avoiding…
When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guarantee
Dixian Zhu, Gang Li, Bokun Wang +2
In this paper, we propose systematic and efficient gradient-based methods for both one-way and two-way partial AUC (pAUC) maximization that are applicable to deep learning. We prop…
Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applications
Bokun Wang, Tianbao Yang
This paper studies stochastic optimization for a sum of compositional functions, where the inner-level function of each summand is coupled with the corresponding summation index. W…
Optimal Algorithms for Stochastic Multi-Level Compositional Optimization
Wei Jiang, Bokun Wang, Yibo Wang +2
In this paper, we investigate the problem of stochastic multi-level compositional optimization, where the objective function is a composition of multiple smooth but possibly non-co…