Jun Wu

Jun Wu

Assistant Professor

Publications

You can also find my articles on my Google Scholar profile.

Preprints

  1. Single-Cell Data Integration and Cell Type Annotation through Contrastive Adversarial Open-set Domain Adaptation
    Fatemeh Aminzadeh*, Jun Wu*, Jingrui He, Morteza Saberi, Fatemeh Vafaee

  2. Plan and Budget: Effective and Efficient Test-Time Scaling on Large Language Model Reasoning
    Junhong Lin, Xinyue Zeng, Jie Zhu, Song Wang, Julian Shun, Jun Wu, Dawei Zhou

Selected Publications (Full List)

Heterogeneous Machine Learning

  1. LensLLM: Unveiling Fine-Tuning Dynamics for LLM Selection [Paper]
    Xinyue Zeng, Haohui Wang, Junhong Lin, Jun Wu, Tyler Cody, Dawei Zhou
    Forty-second International Conference on Machine Learning (ICML-2025)

  2. Distributional Network of Networks for Modeling Data Heterogeneity [Paper]
    Jun Wu, Jingrui He, Hanghang Tong
    30th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (KDD-2024)

  3. Graph-Structured Gaussian Processes for Transferable Graph Learning [Paper]
    Jun Wu, Elizabeth Ainsworth, Andrew Leakey, Haixun Wang, Jingrui He
    Thirty-seventh Conference on Neural Information Processing Systems (NeurIPS-2023)

  4. Distribution-Informed Neural Networks for Domain Adaptation Regression [Paper]
    Jun Wu, Jingrui He, Sheng Wang, Kaiyu Guan, Elizabeth Ainsworth
    36th Conference on Neural Information Processing Systems (NeurIPS-2022)

  5. Domain Adaption with Dynamic Open-Set Targets [Paper]
    Jun Wu, Jingrui He
    28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD-2022)

Trustworthy Machine Learning

  1. Trustworthy Transfer Learning: A Survey [Paper]
    Jun Wu, Jingrui He
    Journal of Artificial Intelligence Research (JAIR-2025)

  2. BOBA: Byzantine-Robust Federated Learning with Label Skewness [Paper]
    Wenxuan Bao, Jun Wu, Jingrui He
    27th International Conference on Artificial Intelligence and Statistics (AISTATS-2024)

  3. Personalized Federated Learning with Parameter Propagation [Paper]
    Jun Wu, Wenxuan Bao, Elizabeth Ainsworth, Jingrui He
    29th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD-2023)

  4. Optimizing the Collaboration Structure in Cross-silo Federated Learning [Paper]
    Wenxuan Bao, Haohan Wang, Jun Wu, Jingrui He
    Fortieth International Conference on Machine Learning (ICML-2023)

  5. Indirect Invisible Poisoning Attacks on Domain Adaptation [Paper]
    Jun Wu, Jingrui He
    27th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD-2021)

(* indicates equal contribution)