← Search

Yuanjie Shi

10 accepted papers

2026

Cost-Sensitive Conformal Training with Provably Controllable Learning Bounds

AAAI 2026technical

Conformal prediction (CP) is a general framework to quantify the predictive uncertainty of machine learning models that uses a set prediction to include the true label with a valid probability. To align the uncertainty measured by CP, conformal training methods minimize the size of the prediction se

Cited by 0SourcePDFScholar
2026

Inference-Time Conformal Reasoning with Valid Factuality Control for Large Language Models

ICML 2026poster

Large language models (LLMs) increasingly perform multi-step reasoning, where intermediate claims form implicit directed acyclic graphs whose node correctness is structurally conditioned on their ancestors. This makes factuality uncertainty structural, rather than a trivial accumulation of node-wise…

Cited by 0SourceScholar
2026

Keep the Best, Forget the Rest: Reliable Alignment with Order-Aware Preference Optimization

ICLR 2026poster

Direct Preference Optimization (DPO) has emerged as a powerful framework for aligning large language models (LLMs) with human preferences via pairwise comparisons. However, its performance is highly sensitive to the quality of training samples: when the reference policy is poorly aligned with human…

Cited by 0SourcecodeScholar
2026

Minimum-Length Conformal Prediction Sets for Ordinal Classification

AAAI 2026technical

Ordinal classification has been widely applied in many high-stakes applications, e.g., medical imaging and diagnosis, where reliable uncertainty quantification (UQ) is essential for decision making. Conformal prediction (CP) is a general UQ framework that provides statistically valid guarantees, whi

Cited by 0SourcePDFScholar
2025

Direct Prediction Set Minimization via Bilevel Conformal Classifier Training

ICML 2025poster

Conformal prediction (CP) is a promising uncertainty quantification framework which works as a wrapper around a black-box classifier to construct prediction sets (i.e., subset of candidate classes) with provable guarantees. However, standard calibration methods for CP tend to produce large predicti…

Cited by 0SourcePDFScholar
2025

FedSum: Data-Efficient Federated Learning Under Data Scarcity Scenario for Text Summarization

AAAI 2025technical

Text summarization task extracts salient information from a large amount of text for productivity enhancement. However, most existing methods heavily rely on training models from ample and centrally stored data which is infeasible to collect in practice, due to privacy concerns and data scarcity nat…

2024

Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibration

NeurIPS 2024poster

Conformal prediction (CP) is an emerging uncertainty quantification framework that allows us to construct a prediction set to cover the true label with a pre-specified marginal or conditional probability. Although the valid coverage guarantee has been extensively studied for classification problems,…

2023

Probabilistically robust conformal prediction

UAI 2023poster

Conformal prediction (CP) is a framework to quantify uncertainty of machine learning classifiers including deep neural networks. Given a testing example and a trained classifier, CP produces a prediction set of candidate labels with a user-specified coverage (i.e., true class label is contained wit…