publications

2025

  1. mint.png
    Mint: A Simple Test-Time Adaptation of Vision-Language Models against Common Corruptions
    Wenxuan Bao, Ruxi Deng, and Jingrui He
    In NeurIPS, 2025
    Code
  2. latte.png
    Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning
    Wenxuan Bao, Ruxi Deng, Ruizhong Qiu, Tianxin Wei, Hanghang Tong, and Jingrui He
    In ICCV, 2025
    arXiv Code
  3. matcha.png
    Matcha: Mitigating Graph Structure Shifts with Test-Time Adaptation
    Wenxuan Bao, Zhichen Zeng, Zhining Liu, Hanghang Tong, and Jingrui He
    In ICLR, 2025
    arXiv HTML Code
  4. Ask, and it shall be given: Turing completeness of prompting
    Ruizhong Qiu, Zhe Xu, Wenxuan Bao, and Hanghang Tong
    In ICLR, 2025
    arXiv HTML Code
  5. Connecting Domains and Contrasting Samples: A Ladder for Domain Generalization
    Tianxin Wei, Yifan Chen, Xinrui He, Wenxuan Bao, and Jingrui He
    In KDD, 2025
    HTML
  6. Pave Your Own Path: Graph Gradual Domain Adaptation on Fused Gromov-Wasserstein Geodesics
    Zhichen Zeng, Ruizhong Qiu, Wenxuan Bao, Tianxin Wei, Xiao Lin, Yuchen Yan, Tarek F. Abdelzaher, Jiawei Han, and Hanghang Tong
    CoRR, 2025
    arXiv

2024

  1. boba2.png
    BOBA: Byzantine-Robust Federated Learning with Label Skewness
    Wenxuan Bao, Jun Wu, and Jingrui He
    In AISTATS, 2024
    arXiv HTML Code
  2. FedSelect: Personalized Federated Learning with Customized Selection of Parameters for Fine-Tuning
    Rishub Tamirisa, Chulin Xie, Wenxuan Bao, Andy Zhou, Ron Arel, and Aviv Shamsian
    In CVPR, 2024
    arXiv HTML Code

2023

  1. atp.png
    Adaptive Test-Time Personalization for Federated Learning
    Wenxuan Bao, Tianxin Wei, Haohan Wang, and Jingrui He
    In NeurIPS, 2023
    arXiv HTML Code
  2. fedcollab.png
    Optimizing the Collaboration Structure in Cross-Silo Federated Learning
    Wenxuan Bao, Haohan Wang, Jun Wu, and Jingrui He
    In ICML, 2023
    arXiv HTML Code Website
  3. fedora.png
    Personalized Federated Learning with Parameter Propagation
    Jun Wu, Wenxuan Bao, Elizabeth Ainsworth, and Jingrui He
    In KDD, 2023
    HTML PDF Code
  4. Privacy-Preserving Graph Machine Learning from Data to Computation: A Survey
    Dongqi Fu, Wenxuan Bao, Ross Maciejewski, Hanghang Tong, and Jingrui He
    SIGKDD Exploration Newsletter, 2023
    arXiv HTML
  5. Understanding the Effect of Data Augmentation on Knowledge Distillation
    Ziqi Wang, Chi Han, Wenxuan Bao, and Heng Ji
    CoRR, 2023
    arXiv

2020

  1. Deep Grouping Model for Unified Perceptual Parsing
    Zhiheng Li, Wenxuan Bao, Jiayang Zheng, and Chenliang Xu
    In CVPR, 2020
    arXiv