Publications

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.