← Search

Shaojie Li

18 accepted papers

2026

High Probability Bounds for Non-Convex Stochastic Optimization with Momentum

ICLR 2026poster

Stochastic gradient descent with momentum (SGDM) is widely used in machine learning, yet high-probability learning bounds for SGDM in non-convex settings remain scarce. In this paper, we provide high-probability convergence bounds and generalization bounds for SGDM. First, we establish such bounds f…

Cited by 0SourceScholar
2025

Interactive and Balanced Multimodal Learning via Cross Attention and Gradient Modulation for Compressed Video Action Recognition

ICASSP 2025accepted

Compressed video action recognition is a crucial task in video processing. Compared with traditional methods, it directly processes RGB (I-frames) and motion (motion vectors and residuals) modalities, which effectively alleviates computational burdens. However, this task suffers from dynamic noise a…

Cited by 0SourceScholar
2025

Stability and Sharper Risk Bounds with Convergence Rate $\tilde{O}(1/n^2)$

NeurIPS 2025poster

Prior work (Klochkov \& Zhivotovskiy, 2021) establishes at most $O\left(\log (n)/n\right)$ excess risk bounds via algorithmic stability for strongly-convex learners with high probability. We show that under the similar common assumptions — Polyak-Lojasiewicz condition, smoothness, and Lipschitz cont…

Cited by 0SourceScholar
2023

Generalization Bounds for Federated Learning: Fast Rates, Unparticipating Clients and Unbounded Losses

ICLR 2023poster

In {federated learning}, the underlying data distributions may be different across clients. This paper provides a theoretical analysis of generalization error of {federated learning}, which captures both heterogeneity and relatedness of the distributions. In particular, we assume that the heterogene…

Cited by 18SourcePDFScholar
2023

Knowledge Distillation with Active Exploration and Self-Attention Based Inter-Class Variation Transfer for Image Segmentation

ICASSP 2023accepted

Knowledge distillation (KD) aims to distill the knowledge from a more extensive deep neural network into a small net-work without losing validity. This paper proposes a novel approach with active exploration and passive transfer (AEPT) and self-attention-based inter-class feature variation (AIFV) di…

Cited by 0SourceScholar
2023

MTFD: Multi-Teacher Fusion Distillation for Compressed Video Action Recognition

ICASSP 2023accepted

As an important work in computer vision, some recent representative works such as Two-stream networks, 3D ConvNets, and Transformer-based networks have achieved outstanding performance. However, due to the high computational cost, the explosion of computation time and parameters, they cannot meet th…

Cited by 0SourceScholar
2023

Understanding the Generalization Performance of Spectral Clustering Algorithms

AAAI 2023technical

The theoretical analysis of spectral clustering is mainly devoted to consistency, while there is little research on its generalization performance. In this paper, we study the excess risk bounds of the popular spectral clustering algorithms: relaxed RatioCut and relaxed NCut. Our analysis follows th…

Cited by 4SourcePDFScholar
2021

Revisiting Discriminator in GAN Compression: A Generator-discriminator Cooperative Compression Scheme

NeurIPS 2021poster

Recently, a series of algorithms have been explored for GAN compression, which aims to reduce tremendous computational overhead and memory usages when deploying GANs on resource-constrained edge devices. However, most of the existing GAN compression work only focuses on how to compress the generator…