← Search

Runtian Zhai

9 accepted papers

2025

Contextures: Representations from Contexts

ICML 2025poster

Despite the empirical success of foundation models, we do not have a systematic characterization of the representations that these models learn. In this paper, we establish the contexture theory. It shows that a large class of representation learning methods can be characterized as learning from th…

Cited by 0SourcePDFScholar
2024

Spectrally Transformed Kernel Regression

ICLR 2024spotlight

Unlabeled data is a key component of modern machine learning. In general, the role of unlabeled data is to impose a form of smoothness, usually from the similarity information encoded in a base kernel, such as the ϵ-neighbor kernel or the adjacency matrix of a graph. This work revisits the classical…

Cited by 3SourcePDFScholar
2024

Understanding Augmentation-based Self-Supervised Representation Learning via RKHS Approximation and Regression

ICLR 2024spotlight

Data augmentation is critical to the empirical success of modern self-supervised representation learning, such as contrastive learning and masked language modeling. However, a theoretical understanding of the exact role of the augmentation remains limited. Recent work has built the connection betwee…

Cited by 16SourcePDFScholar
2023

Characterizing Out-of-Distribution Error via Optimal Transport

NeurIPS 2023poster

Out-of-distribution (OOD) data poses serious challenges in deployed machine learning models, so methods of predicting a model's performance on OOD data without labels are important for machine learning safety. While a number of methods have been proposed by prior work, they often underestimate the a…

Cited by 20SourcePDFScholar
2023

Responsible AI (RAI) Games and Ensembles

NeurIPS 2023poster

Several recent works have studied the societal effects of AI; these include issues such as fairness, robustness, and safety. In many of these objectives, a learner seeks to minimize its worst-case loss over a set of predefined distributions (known as uncertainty sets), with usual examples being per…

2023

Understanding Why Generalized Reweighting Does Not Improve Over ERM

ICLR 2023poster

Empirical risk minimization (ERM) is known to be non-robust in practice to distributional shift where the training and the test distributions are different. A suite of approaches, such as importance weighting, and variants of distributionally robust optimization (DRO), have been proposed to solve th…

2021

Boosted CVaR Classification

NeurIPS 2021poster

Many modern machine learning tasks require models with high tail performance, i.e. high performance over the worst-off samples in the dataset. This problem has been widely studied in fields such as algorithmic fairness, class imbalance, and risk-sensitive decision making. A popular approach to maxim…

2021

DORO: Distributional and Outlier Robust Optimization

ICML 2021spotlight

Many machine learning tasks involve subpopulation shift where the testing data distribution is a subpopulation of the training distribution. For such settings, a line of recent work has proposed the use of a variant of empirical risk minimization(ERM) known as distributionally robust optimization (D…

2020

MACER: Attack-free and Scalable Robust Training via Maximizing Certified Radius

ICLR 2020poster

Adversarial training is one of the most popular ways to learn robust models but is usually attack-dependent and time costly. In this paper, we propose the MACER algorithm, which learns robust models without using adversarial training but performs better than all existing provable l2-defenses. Recent…

Cited by 205SourcecodeScholar