← Search

Ke Wei

6 accepted papers

2026

Stability and Generalization of Nonconvex Optimization with Heavy-Tailed Noise

ICML 2026poster

The empirical evidence indicates that stochastic optimization with heavy-tailed gradient noise is more appropriate to characterize the training of machine learning models than that with standard bounded gradient variance noise. Most existing works on this phenomenon focus on the convergence of optim…

Cited by 0SourceScholar
2025

A Near-Optimal Algorithm for Decentralized Convex-Concave Finite-Sum Minimax Optimization

NeurIPS 2025spotlight

In this paper, we study the distributed convex-concave finite-sum minimax optimization over the network, and a decentralized variance-reduced optimistic gradient method with stochastic mini-batch sizes (DIVERSE) is proposed. For the strongly-convex-strongly-concave objective, it is shown that DIVERS…

Cited by 0SourceScholar
2024

Investigating Stability Outcomes Across Diverse Gait Patterns in Quadruped Robots: A Comparative Analysis

RA-L 2024

Quadruped robots have gained attention for their potential to navigate various terrains. However, the stability of these robots in different gait sequences remains an open question. This study investigates the relationship between different gait sequences and the motion stability of quadruped robots

Cited by 4SourceScholar
2021

Is Attention Better Than Matrix Decomposition?

ICLR 2021poster

As an essential ingredient of modern deep learning, attention mechanism, especially self-attention, plays a vital role in the global correlation discovery. However, is hand-crafted attention irreplaceable when modeling the global context? Our intriguing finding is that self-attention is not better t…