← Search

Jiaye Teng

15 accepted papers

2026

Questioning the Coverage-Length Metric in Conformal Prediction: When Shorter Intervals Are Not Better

ICML 2026poster

Conformal prediction (CP) has become a cornerstone of distribution-free uncertainty quantification, conventionally evaluated by its coverage and interval length. This work critically examines the sufficiency of these standard metrics. We demonstrate that the interval length might be deceptively impr…

Cited by 0SourceScholar
2026

Theoretical Analysis of Sparse Optimization with Reparameterization, Weight Decay, and Adaptive Learning Rate

ICML 2026poster

Sparse optimization is a fundamental challenge in various practical applications. A popular approach to sparse optimization is Lp regularization. However, it may encounter optimization instability due to the unbounded gradients when 0<p<1. In this paper, we introduce a novel approach to sparse optim…

Cited by 0SourceScholar
2026

Theoretical Modeling of Large Language Model Self-Improvement Training Dynamics Through Solver-Verifier Gap

ICLR 2026poster

Self-improvement is a significant techniques within the realm of large language model (LLM), aiming to enhance the LLM performance without relying on external data. Despite its significance, generally how LLM performances evolve during the self-improvement process remains underexplored. In this pape…

Cited by 0SourceScholar
2025

Generalization Lower Bounds for GD and SGD in Smooth Stochastic Convex Optimization

AISTATS 2025poster

This work studies the generalization error of gradient methods. More specifically, we focus on how training steps $T$ and step-size $\eta$ might affect generalization in smooth stochastic convex optimization (SCO) problems. Recent works show that in some cases longer training can hurt generalization…

Cited by 0SourceScholar
2025

Minimax Optimal Two-Stage Algorithm For Moment Estimation Under Covariate Shift

ICLR 2025poster

Covariate shift occurs when the distribution of input features differs between the training and testing phases. In covariate shift, estimating an unknown function's moment is a classical problem that remains under-explored, despite its common occurrence in real-world scenarios. In this paper, we in…

Cited by 0SourcePDFScholar
2023

Benign Overfitting in Classification: Provably Counter Label Noise with Larger Models

ICLR 2023poster

Studies on benign overfitting provide insights for the success of overparameterized deep learning models. In this work, we examine whether overfitting is truly benign in real-world classification tasks. We start with the observation that a ResNet model overfits benignly on Cifar10 but not benignly o…

Cited by 8SourcePDFScholar
2023

Finding Generalization Measures by Contrasting Signal and Noise

ICML 2023poster

Generalization is one of the most fundamental challenges in deep learning, aiming to predict model performances on unseen data. Empirically, such predictions usually rely on a validation set, while recent works showed that an unlabeled validation set also works. Without validation sets, it is extrem…

2023

On Uni-Modal Feature Learning in Supervised Multi-Modal Learning

ICML 2023poster

We abstract the features (i.e. learned representations) of multi-modal data into 1) uni-modal features, which can be learned from uni-modal training, and 2) paired features, which can only be learned from cross-modal interactions. Multi-modal models are expected to benefit from cross-modal interacti…

2023

Predictive Inference with Feature Conformal Prediction

ICLR 2023poster

Conformal prediction is a distribution-free technique for establishing valid prediction intervals. Although conventionally people conduct conformal prediction in the output space, this is not the only possibility. In this paper, we propose feature conformal prediction, which extends the scope of con…

2023

Towards Data-Algorithm Dependent Generalization: a Case Study on Overparameterized Linear Regression

NeurIPS 2023poster

One of the major open problems in machine learning is to characterize generalization in the overparameterized regime, where most traditional generalization bounds become inconsistent even for overparameterized linear regression. In many scenarios, this failure can be attributed to obscuring the cruc…

Cited by 2SourcePDFScholar
2022

Can Pretext-Based Self-Supervised Learning Be Boosted by Downstream Data? A Theoretical Analysis

AISTATS 2022poster

Pretext-based self-supervised learning learns the semantic representation via a handcrafted pretext task over unlabeled data and then uses the learned representation for downstream tasks, which effectively reduces the sample complexity of downstream tasks under Conditional Independence (CI) conditio…

Cited by 13SourcePDFScholar
2022

Fighting Fire with Fire: Avoiding DNN Shortcuts through Priming

ICML 2022spotlight

Across applications spanning supervised classification and sequential control, deep learning has been reported to find “shortcut” solutions that fail catastrophically under minor changes in the data distribution. In this paper, we show empirically that DNNs can be coaxed to avoid poor shortcuts by p…

Cited by 21SourcePDFScholar
2021

T-SCI: A Two-Stage Conformal Inference Algorithm with Guaranteed Coverage for Cox-MLP

ICML 2021spotlight

It is challenging to deal with censored data, where we only have access to the incomplete information of survival time instead of its exact value. Fortunately, under linear predictor assumption, people can obtain guaranteed coverage for the confidence interval of survival time using methods like Cox…