← Search

Wenxuan Bao

13 accepted papers

2025

Ask, and it shall be given: On the Turing completeness of prompting

ICLR 2025poster

Since the success of GPT, large language models (LLMs) have revolutionized machine learning and have initiated the so-called *LLM prompting* paradigm. In the era of LLMs, people train a single general-purpose LLM and provide the LLM with different *prompts* to perform different tasks. However, such…

2025

Deep Learning with Plausible Deniability

NeurIPS 2025poster

Deep learning models are vulnerable to privacy attacks due to their tendency to memorize individual training examples. Theoretically-sound defenses such as differential privacy can defend against this threat, but model performance often suffers. Empirical defenses may thwart existing attacks while m…

Cited by 0SourceScholar
2025

Latte: Collaborative Test-Time Adaptation of Vision-Language Models in Federated Learning

ICCV 2025poster

Test-time adaptation with pre-trained vision-language models has gained increasing attention for addressing distribution shifts during testing. Among these approaches, memory-based algorithms stand out due to their training-free nature and ability to leverage historical test data. However, existing…

2025

Matcha: Mitigating Graph Structure Shifts with Test-Time Adaptation

ICLR 2025poster

Powerful as they are, graph neural networks (GNNs) are known to be vulnerable to distribution shifts. Recently, test-time adaptation (TTA) has attracted attention due to its ability to adapt a pre-trained model to a target domain, without re-accessing the source domain. However, existing TTA algorit…

2025

Mint: A Simple Test-Time Adaptation of Vision-Language Models against Common Corruptions

NeurIPS 2025poster

Pretrained vision-language models such as CLIP achieve strong zero-shot generalization but remain vulnerable to distribution shifts caused by input corruptions. In this work, we investigate how corruptions affect CLIP’s image embeddings and uncover a consistent phenomenon we term as embedding varian…

Cited by 0SourcecodeScholar
2024

FedSelect: Personalized Federated Learning with Customized Selection of Parameters for Fine-Tuning

CVPR 2024poster

Standard federated learning approaches suffer when client data distributions have sufficient heterogeneity. Recent methods addressed the client data heterogeneity issue via personalized federated learning (PFL) - a class of FL algorithms aiming to personalize learned global knowledge to better suit…

2023

Adaptive Test-Time Personalization for Federated Learning

NeurIPS 2023poster

Personalized federated learning algorithms have shown promising results in adapting models to various distribution shifts. However, most of these methods require labeled data on testing clients for personalization, which is usually unavailable in real-world scenarios. In this paper, we introduce a n…

2023

DP-Mix: Mixup-based Data Augmentation for Differentially Private Learning

NeurIPS 2023poster

Data augmentation techniques, such as image transformations and combinations, are highly effective at improving the generalization of computer vision models, especially when training data is limited. However, such techniques are fundamentally incompatible with differentially private learning approac…

2023

Optimizing the Collaboration Structure in Cross-Silo Federated Learning

ICML 2023poster

In federated learning (FL), multiple clients collaborate to train machine learning models together while keeping their data decentralized. Through utilizing more training data, FL suffers from the potential negative transfer problem: the global FL model may even perform worse than the models trained…