3 papers
cs.LG2026
MoLS: SNR-Guided Module-Wise Learning-Rate Calibration for Large Language Models
Ziqing Wen, Zhouyang Liu, Jiahuan Wang +5
Large language models (LLMs) contain functionally distinct modules with heterogeneous gradient statistics. Although Adam(W) adapts updates at the parameter level, selecting relativ…
cs.LG2024
Exploring the Generalization Capabilities of AID-based Bi-level Optimization
Congliang Chen, Li Shen, Zhiqiang Xu +3
Bi-level optimization has achieved considerable success in contemporary machine learning applications, especially for given proper hyperparameters. However, due to the two-level op…
cs.LG2023
Rethinking SIGN Training: Provable Nonconvex Acceleration without First- and Second-Order Gradient Lipschitz
Tao Sun, Congliang Chen, Peng Qiao +3
Sign-based stochastic methods have gained attention due to their ability to achieve robust performance despite using only the sign information for parameter updates. However, the c…