Publications

Minimum-Length Conformal Prediction Sets for Ordinal Classification.
Zijian Zhang, Xinyu Chen, Yuanjie Shi, Liyuan Lillian Ma, Zifan Xu, Yan Yan
Proceedings of the The Fortieth AAAI Conference on Artificial Intelligence (AAAI 2026).

Cost-Sensitive Conformal Training with Provably Controllable Learning Bounds.
Xuesong Jia*, Yuanjie Shi*, Ziquan Liu, Yi Xu, Yan Yan
Proceedings of the The Fortieth AAAI Conference on Artificial Intelligence (AAAI 2026).

Federated R'enyi Fair Inference in Federated Heterogeneous System.
Zhiyong Ma*, Yuanjie Shi*, Yan Yan, Jian Chen.
Proceedings of the Forty-First Conference on Uncertainty in Artificial Intelligence (UAI 2025).

Direct Prediction Set Minimization via Bilevel Conformal Classifier Training.
Yuanjie Shi*, Hooman Shahrokhi*, Xuesong Jia, Xiongzhi Chen, Jana Doppa, Yan Yan
Proceedings of the Forty-Second International Conference on Machine Learning (ICML 2025).

FedSum: Data-Efficient Federated Learning Under Data Scarcity Scenario for Text Summarization.
Zhiyong Ma, Zhengping Li, Yuanjie Shi, Jian Chen
Proceedings of the Thirty-Ninth AAAI Conference on Artificial Intelligence (AAAI 2025).

Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibration. [arXiv]
Yuanjie Shi, Subhankar Ghosh, Taha Belkhouja, Jana Doppa, Yan Yan
Proceedings of the Thirty-eighth Annual Conference on Neural Information Processing Systems (NeurIPS 2024).

Probabilistically Robust Conformal Prediction. [arXiv]
Subhankar Ghosh*, Yuanjie Shi*, Taha Belkhouja, Yan Yan, Jana Doppa, Brian Jones.
Proceedings of the Thirty-Ninth Conference on Uncertainty in Artificial Intelligence (UAI 2023), PMLR 216:681-690, 2023.