← Search

Yihao Xue

8 accepted papers

2025

Representations Shape Weak-to-Strong Generalization: Theoretical Insights and Empirical Predictions

ICML 2025poster

Weak-to-Strong Generalization (W2SG), where a weak model supervises a stronger one, serves as an important analogy for understanding how humans might guide superhuman intelligence in the future. Promising empirical results revealed that a strong model can surpass its weak supervisor. While recent w…

Cited by 0SourcePDFScholar
2024

Few-shot Adaptation to Distribution Shifts By Mixing Source and Target Embeddings

ICML 2024poster

Pretrained machine learning models need to be adapted to distribution shifts when deployed in new target environments. When obtaining labeled data from the target distribution is expensive, few-shot adaptation with only a few examples from the target distribution becomes essential. In this work, we…

Cited by 1SourcePDFScholar
2024

Investigating the Benefits of Projection Head for Representation Learning

ICLR 2024poster

An effective technique for obtaining high-quality representations is adding a projection head on top of the encoder during training, then discarding it and using the pre-projection representations. Despite its proven practical effectiveness, the reason behind the success of this technique is poorly…

Cited by 9SourcePDFScholar
2024

Investigating the Impact of Model Width and Density on Generalization in Presence of Label Noise

UAI 2024poster

Increasing the size of overparameterized neural networks has been a key in achieving state-of-the-art performance. This is captured by the double descent phenomenon, where the test loss follows a decreasing-increasing-decreasing pattern (or sometimes monotonically decreasing) as model width increase…

Cited by 3SourcePDFScholar
2024

Understanding the Robustness of Multi-modal Contrastive Learning to Distribution Shift

ICLR 2024poster

Recently, multimodal contrastive learning (MMCL) approaches, such as CLIP, have achieved a remarkable success in learning representations that are robust against distribution shift and generalize to new domains. Despite the empirical success, the mechanism behind learning such generalizable represen…

Cited by 9SourcePDFScholar
2023

Which Features are Learnt by Contrastive Learning? On the Role of Simplicity Bias in Class Collapse and Feature Suppression

ICML 2023oral

Contrastive learning (CL) has emerged as a powerful technique for representation learning, with or without label supervision. However, supervised CL is prone to collapsing representations of subclasses within a class by not capturing all their features, and unsupervised CL may suppress harder class-…

Cited by 25SourcePDFScholar
2022

Investigating Why Contrastive Learning Benefits Robustness against Label Noise

ICML 2022spotlight

Self-supervised Contrastive Learning (CL) has been recently shown to be very effective in preventing deep networks from overfitting noisy labels. Despite its empirical success, the theoretical understanding of the effect of contrastive learning on boosting robustness is very limited. In this work, w…

2021

Toward Understanding the Influence of Individual Clients in Federated Learning

AAAI 2021technical

Federated learning allows mobile clients to jointly train a global model without sending their private data to a central server. Extensive works have studied the performance guarantee of the global model, however, it is still unclear how each individual client influences the collaborative training p…

Cited by 51SourcePDFScholar