Probabilistically robust conformal prediction. - Subhankar Ghosh; Yuanjie Shi; Taha Belkhouja; Yan Yan 006; Jana Doppa; Brian Jones (UAI, 2023)

Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibration. - Yuanjie Shi; Subhankar Ghosh; Taha Belkhouja; Jana Doppa; Yan Yan 006 (NeurIPS, 2024)

FedSum: Data-Efficient Federated Learning Under Data Scarcity Scenario for Text Summarization. - Zhiyong Ma; Zhengping Li; Yuanjie Shi; Jian Chen (AAAI, 2025)

Reliable Uncertainty Quantification in Machine Learning via Conformal Prediction. - Yuanjie Shi (AAAI, 2025)

Federated Rényi Fair Inference in Federated Heterogeneous System. - Zhiyong Ma; Yuanjie Shi; Yan Yan 006; Jian Chen 011 (UAI, 2025)

Direct Prediction Set Minimization via Bilevel Conformal Classifier Training. - Yuanjie Shi; Hooman Shahrokhi; Xuesong Jia; Xiongzhi Chen; Janardhan Rao Doppa; Yan Yan 006 (CoRR, 2025)