PPaperPicks

Wei Huang

RIKEN Center for Advanced Intelligence Projec, Tokyo, Japan

27 papers at tracked venues · 26 at CORE A* · active 2024–2026

DBLP profile ↗ORCID 0000-0001-5583-1774 ↗Homepage ↗

Venues

NeurIPS ×11ICML ×6ICLR ×4AAAI ×2CVPR ×2AISTATS ×1SIGKDD ×1

Frequent coauthors

Taiji Suzuki
×13
Andi Han
×10
Ye Shi
×4
Zhanpeng Zhou
×3
Yilan Chen
×3
Yuan Cao
×3
Jun Zhu
×3
Dake Bu
×3
Qingfu Zhang
×3
Hau-San Wong
×3
Liqiang Nie
×3
Yuxuan Liang
×3

Papers

  1. On the Learning Dynamics of Two-layer Linear Networks with Label Noise SGD
    AAAI 2026 ·
    Tongcheng Zhang
    ,
    Zhanpeng Zhou
    ,
    Mingze Wang
    ,
    Andi Han
    , Wei Huang,
    Taiji Suzuki
    ,
    Junchi Yan
  2. GRU: Mitigating the Trade-off between Unlearning and Retention for LLMs
    ICML 2025 ·
    Yue Wang
    ,
    Qizhou Wang
    ,
    Feng Liu
    , Wei Huang,
    Yali Du
    ,
    Xiaojiang Du
    ,
    Bo Han
  3. Generalization Bound of Gradient Flow through Training Trajectory and Data-dependent Kernel
    NeurIPS 2025 ·
    Yilan Chen
    ,
    Zhichao Wang
    , Wei Huang,
    Andi Han
    ,
    Taiji Suzuki
    ,
    Arya Mazumdar
  4. How Does Label Noise Gradient Descent Improve Generalization in the Low SNR Regime?
    NeurIPS 2025 · Wei Huang,
    Andi Han
    ,
    Yujin Song
    ,
    Yilan Chen
    ,
    Denny Wu
    ,
    Difan Zou
    ,
    Taiji Suzuki
  5. NLPrompt: Noise-Label Prompt Learning for Vision-Language Models
    CVPR 2025 ·
    Bikang Pan
    ,
    Qun Li
    ,
    Xiaoying Tang
    , Wei Huang,
    Zhen Fang
    ,
    Feng Liu
    ,
    Jingya Wang
    ,
    Jingyi Yu
    ,
    Ye Shi
  6. On the Feature Learning in Diffusion Models
    ICLR 2025 ·
    Andi Han
    , Wei Huang,
    Yuan Cao
    ,
    Difan Zou
  7. On the Optimization and Generalization of Two-layer Transformers with Sign Gradient Descent
    ICLR 2025 ·
    Bingrui Li
    , Wei Huang,
    Andi Han
    ,
    Zhanpeng Zhou
    ,
    Taiji Suzuki
    ,
    Jun Zhu
    ,
    Jianfei Chen
  8. On the Role of Label Noise in the Feature Learning Process
    ICML 2025 ·
    Andi Han
    , Wei Huang,
    Zhanpeng Zhou
    ,
    Gang Niu
    ,
    Wuyang Chen
    ,
    Junchi Yan
    ,
    Akiko Takeda
    ,
    Taiji Suzuki
  9. Provable In-Context Vector Arithmetic via Retrieving Task Concepts
    ICML 2025 ·
    Dake Bu
    , Wei Huang,
    Andi Han
    ,
    Atsushi Nitanda
    ,
    Qingfu Zhang
    ,
    Hau-San Wong
    ,
    Taiji Suzuki
  10. Quantifying the Optimization and Generalization Advantages of Graph Neural Networks Over Multilayer Perceptrons
    AISTATS 2025 · Wei Huang,
    Yuan Cao
    ,
    Haonan Wang
    ,
    Xin Cao
    ,
    Taiji Suzuki
  11. Scaling Diffusion Transformers Efficiently via μP
    NeurIPS 2025 ·
    Chenyu Zheng
    ,
    Xinyu Zhang
    ,
    Rongzhen Wang
    , Wei Huang,
    Zhi Tian
    ,
    Weilin Huang
    ,
    Jun Zhu
    ,
    Chongxuan Li
  12. Test-Time Graph Neural Dataset Search With Generative Projection
    ICML 2025 ·
    Xin Zheng
    , Wei Huang,
    Chuan Zhou
    ,
    Ming Li
    ,
    Shirui Pan
  13. Trained Mamba Emulates Online Gradient Descent in In-Context Linear Regression
    NeurIPS 2025 ·
    Jiarui Jiang
    , Wei Huang,
    Miao Zhang
    ,
    Taiji Suzuki
    ,
    Liqiang Nie
  14. Understanding the Forgetting of (Replay-based) Continual Learning via Feature Learning: Angle Matters
    ICML 2025 ·
    Hongyi Wang
    ,
    Shiyuan Ren
    , Wei Huang,
    Miao Zhang
    ,
    Xiang Deng
    ,
    Yixin Bao
    ,
    Liqiang Nie
  15. Earthfarsser: Versatile Spatio-Temporal Dynamical Systems Modeling in One Model
    AAAI 2024 ·
    Hao Wu
    ,
    Yuxuan Liang
    ,
    Wei Xiong
    ,
    Zhengyang Zhou
    , Wei Huang,
    Shilong Wang
    ,
    Kun Wang
  16. Federated Learning from Vision-Language Foundation Models: Theoretical Analysis and Method
    NeurIPS 2024 ·
    Bikang Pan
    , Wei Huang,
    Ye Shi
  17. Global and Local Prompts Cooperation via Optimal Transport for Federated Learning
    CVPR 2024 ·
    Hongxia Li
    , Wei Huang,
    Jingya Wang
    ,
    Ye Shi
  18. Graph Lottery Ticket Automated
    ICLR 2024 ·
    Guibin Zhang
    ,
    Kun Wang
    , Wei Huang,
    Yanwei Yue
    ,
    Yang Wang
    ,
    Roger Zimmermann
    ,
    Aojun Zhou
    ,
    Dawei Cheng
    ,
    Jin Zeng
    ,
    Yuxuan Liang
  19. On Mesa-Optimization in Autoregressively Trained Transformers: Emergence and Capability
    NeurIPS 2024 ·
    Chenyu Zheng
    , Wei Huang,
    Rongzhen Wang
    ,
    Guoqiang Wu
    ,
    Jun Zhu
    ,
    Chongxuan Li
  20. On the Comparison between Multi-modal and Single-modal Contrastive Learning
    NeurIPS 2024 · Wei Huang,
    Andi Han
    ,
    Yongqiang Chen
    ,
    Yuan Cao
    ,
    Zhiqiang Xu
    ,
    Taiji Suzuki
  21. Provable and Efficient Dataset Distillation for Kernel Ridge Regression
    NeurIPS 2024 ·
    Yilan Chen
    , Wei Huang,
    Lily Weng
  22. Provably Neural Active Learning Succeeds via Prioritizing Perplexing Samples
    ICML 2024 ·
    Dake Bu
    , Wei Huang,
    Taiji Suzuki
    ,
    Ji Cheng
    ,
    Qingfu Zhang
    ,
    Zhiqiang Xu
    ,
    Hau-San Wong
  23. Provably Transformers Harness Multi-Concept Word Semantics for Efficient In-Context Learning
    NeurIPS 2024 ·
    Dake Bu
    , Wei Huang,
    Andi Han
    ,
    Atsushi Nitanda
    ,
    Taiji Suzuki
    ,
    Qingfu Zhang
    ,
    Hau-San Wong
  24. SLTrain: a sparse plus low rank approach for parameter and memory efficient pretraining
    NeurIPS 2024 ·
    Andi Han
    ,
    Jiaxiang Li
    , Wei Huang,
    Mingyi Hong
    ,
    Akiko Takeda
    ,
    Pratik Kumar Jawanpuria
    ,
    Bamdev Mishra
  25. The Heterophilic Snowflake Hypothesis: Training and Empowering GNNs for Heterophilic Graphs
    SIGKDD 2024 ·
    Kun Wang
    ,
    Guibin Zhang
    ,
    Xinnan Zhang
    ,
    Junfeng Fang
    ,
    Xun Wu
    ,
    Guohao Li
    ,
    Shirui Pan
    , Wei Huang,
    Yuxuan Liang
  26. Understanding Convergence and Generalization in Federated Learning through Feature Learning Theory
    ICLR 2024 · Wei Huang,
    Ye Shi
    ,
    Zhongyi Cai
    ,
    Taiji Suzuki
  27. Unveil Benign Overfitting for Transformer in Vision: Training Dynamics, Convergence, and Generalization
    NeurIPS 2024 ·
    Jiarui Jiang
    , Wei Huang,
    Miao Zhang
    ,
    Taiji Suzuki
    ,
    Liqiang Nie