← Search

Yihan Zhang

10 accepted papers

2026

Convergence Analysis of Decentralized Hessian-/Jacobian-Free Algorithm for Nonconvex Stochastic Bilevel Optimization

ICML 2026poster

Decentralized stochastic bi-level optimization has been actively studied in recent years. However, existing studies assume that the lower-level loss function is strongly convex, which limits their applicability to many machine learning models. To address this limitation, in this paper, we propose a …

Cited by 0SourceScholar
2026

On the Convergence of Decentralized Stochastic Minimax Optimization Algorithm with Compressed Communication

ICML 2026poster

The stochastic minimax optimization problem has widespread applications in machine learning. Recently, numerous distributed minimax optimization algorithms have been developed to handle distributed training data. However, most of these algorithms suffer from high communication costs. To address this…

Cited by 0SourceScholar
2025

Enhancing Language Model Hypernetworks with Restart: A Study on Optimization

NAACL 2025long

Hypernetworks are a class of meta-networks that generate weights for main neural networks. Their unique parameter spaces necessitate exploring suitable optimization strategies to enhance performance, especially for language models. However, a comprehensive investigation into optimization strategies…

2025

Federated Stochastic Bilevel Optimization with Fully First-Order Gradients

IJCAI 2025

Federated stochastic bilevel optimization has been actively studied in recent years due to its widespread applications in machine learning. However, most existing federated stochastic bilevel optimization algorithms require the computation of second-order Hessian and Jacobian matrices, which leads t

Cited by 0SourcePDFScholar
2024

Matrix Denoising with Doubly Heteroscedastic Noise: Fundamental Limits and Optimal Spectral Methods

NeurIPS 2024poster

We study the matrix denoising problem of estimating the singular vectors of a rank-$1$ signal corrupted by noise with both column and row correlations. Existing works are either unable to pinpoint the exact asymptotic estimation error or, when they do so, the resulting approaches (e.g., based on whi…

Cited by 3SourcePDFScholar
2023

Communication-Efficient Stochastic Gradient Descent Ascent with Momentum Algorithms

IJCAI 2023poster

Numerous machine learning models can be formulated as a stochastic minimax optimization problem, such as imbalanced data classification with AUC maximization. Developing efficient algorithms to optimize such kinds of problems is of importance and necessity. However, most existing algorithms restri…

Cited by 93SourcePDFScholar
2023

Federated Compositional Deep AUC Maximization

NeurIPS 2023poster

Federated learning has attracted increasing attention due to the promise of balancing privacy and large-scale learning; numerous approaches have been proposed. However, most existing approaches focus on problems with balanced data, and prediction performance is far from satisfactory for many real-wo…

Cited by 12SourcePDFScholar