Huang, Xinmeng

18 publications

JMLR 2025 Decentralized Bilevel Optimization: A Perspective from Transient Iteration Complexity Boao Kong, Shuchen Zhu, Songtao Lu, Xinmeng Huang, Kun Yuan
ICML 2024 Distributed Bilevel Optimization with Communication Compression Yutong He, Jie Hu, Xinmeng Huang, Songtao Lu, Bin Wang, Kun Yuan
ICLR 2024 Momentum Benefits Non-Iid Federated Learning Simply and Provably Ziheng Cheng, Xinmeng Huang, Pengfei Wu, Kun Yuan
NeurIPS 2024 One-Shot Safety Alignment for Large Language Models via Optimal Dualization Xinmeng Huang, Shuo Li, Edgar Dobriban, Osbert Bastani, Hamed Hassani, Dongsheng Ding
ICMLW 2024 One-Shot Safety Alignment for Large Language Models via Optimal Dualization Xinmeng Huang, Shuo Li, Edgar Dobriban, Osbert Bastani, Hamed Hassani, Dongsheng Ding
ICMLW 2024 One-Shot Safety Alignment for Large Language Models via Optimal Dualization Xinmeng Huang, Shuo Li, Edgar Dobriban, Osbert Bastani, Hamed Hassani, Dongsheng Ding
NeurIPS 2024 SPARKLE: A Unified Single-Loop Primal-Dual Framework for Decentralized Bilevel Optimization Shuchen Zhu, Boao Kong, Songtao Lu, Xinmeng Huang, Kun Yuan
ICLR 2024 Stochastic Controlled Averaging for Federated Learning with Communication Compression Xinmeng Huang, Ping Li, Xiaoyun Li
ICML 2023 Demystifying Disagreement-on-the-Line in High Dimensions Donghwan Lee, Behrad Moniri, Xinmeng Huang, Edgar Dobriban, Hamed Hassani
JMLR 2023 Removing Data Heterogeneity Influence Enhances Network Topology Dependence of Decentralized SGD Kun Yuan, Sulaiman A. Alghunaim, Xinmeng Huang
JMLR 2023 T-Cal: An Optimal Test for the Calibration of Predictive Models Donghwan Lee, Xinmeng Huang, Hamed Hassani, Edgar Dobriban
NeurIPS 2023 Unbiased Compression Saves Communication in Distributed Optimization: When and How Much? Yutong He, Xinmeng Huang, Kun Yuan
NeurIPS 2022 Collaborative Learning of Discrete Distributions Under Heterogeneity and Communication Constraints Xinmeng Huang, Donghwan Lee, Edgar Dobriban, Hamed Hassani
NeurIPS 2022 Lower Bounds and Nearly Optimal Algorithms in Distributed Learning with Communication Compression Xinmeng Huang, Yiming Chen, Wotao Yin, Kun Yuan
NeurIPSW 2022 Optimal Complexity in Non-Convex Decentralized Learning over Time-Varying Networks Xinmeng Huang, Kun Yuan
NeurIPS 2022 Revisiting Optimal Convergence Rate for Smooth and Non-Convex Stochastic Decentralized Optimization Kun Yuan, Xinmeng Huang, Yiming Chen, Xiaohan Zhang, Yingya Zhang, Pan Pan
NeurIPS 2021 An Improved Analysis and Rates for Variance Reduction Under Without-Replacement Sampling Orders Xinmeng Huang, Kun Yuan, Xianghui Mao, Wotao Yin
ICCV 2021 DecentLaM: Decentralized Momentum SGD for Large-Batch Deep Training Kun Yuan, Yiming Chen, Xinmeng Huang, Yingya Zhang, Pan Pan, Yinghui Xu, Wotao Yin