publications

Publications in reverse chronological order. Equal-contribution information is shown below the relevant entries.

2026

  1. UAI
    uai26_sketch_size_effect.png
    Valid and Efficient Uncertainty Quantification for Federated Joint Shift
    Yuanjie Shi, Peihong Li, Xin Cao, and Yan Yan
    In Proceedings of the Forty-Second Conference on Uncertainty in Artificial Intelligence, 2026
  2. ICML
    ICML26_inference_time.png
    Inference-Time Conformal Reasoning with Valid Factuality Control for Large Language Models
    Tian Wang, Yuanjie Shi, Yan Yan, and Huan Zhang
    In Proceedings of the Forty-Third International Conference on Machine LearningTian Wang and Yuanjie Shi contributed equally. , 2026
  3. ICLR
    ICLR2026.png
    Keep the Best, Forget the Rest: Reliable Alignment with Order-Aware Preference Optimization
    Jiacheng Zhu, Yuanjie Shi, Xuebin Peng, Xiaoran Liu, Yan Yan, and Honghao Wei
    In Proceedings of the Fourteenth International Conference on Learning RepresentationsJiacheng Zhu and Yuanjie Shi contributed equally. , 2026
  4. AISTATS
    AISTATS2026.png
    Conformal Margin Risk Minimization: An Envelope Framework for Robust Learning under Label Noise
    Yuanjie Shi, Peihong Li, Jana Doppa, and Yan Yan
    In Proceedings of the 29th International Conference on Artificial Intelligence and StatisticsYuanjie Shi and Peihong Li contributed equally. , 2026
  5. AAAI
    AAAI26_costsensitive.png
    Cost-Sensitive Conformal Training with Provably Controllable Learning Bounds
    Xuesong Jia, Yuanjie Shi, Ziquan Liu, Yiqing Xu, and Yan Yan
    In Proceedings of the Fortieth AAAI Conference on Artificial IntelligenceXuesong Jia and Yuanjie Shi contributed equally. , 2026
  6. AAAI
    AAAI26_minlength_OCP.png
    Minimum-Length Conformal Prediction Sets for Ordinal Classification
    Zijian Zhang, Xinyu Chen, Yuanjie Shi, Lu Ma, Zhi Xu, and Yan Yan
    In Proceedings of the Fortieth AAAI Conference on Artificial Intelligence, 2026
  7. Under Review
    A Latent-Load Framework for Reliability Analysis and Intervention Design in LLM Pipelines
    Yuanjie Shi and Yan Yan
    2026
    Under review
  8. Under Review
    Conformal Classification on Conformity Spectrum: A Probabilistic Geometric Framework
    Yuanjie Shi, Hamed Shahrokhi, Zijian Zhang, Jana Doppa, and Yan Yan
    2026
    Under review

2025

  1. ICML
    ICML25_DPSM.png
    Direct Prediction Set Minimization via Bilevel Conformal Classifier Training
    Yuanjie Shi, Hamed Shahrokhi, Xuesong Jia, Xinyu Chen, Jana Doppa, and Yan Yan
    In Proceedings of the Forty-Second International Conference on Machine LearningYuanjie Shi and Hamed Shahrokhi contributed equally. , 2025
  2. UAI
    Federated Rényi Fair Inference in Federated Heterogeneous System
    Zicheng Ma, Yuanjie Shi, Yan Yan, and Jie Chen
    In Proceedings of the Forty-First Conference on Uncertainty in Artificial IntelligenceZicheng Ma and Yuanjie Shi contributed equally. , 2025
  3. AAAI
    FedSum: Data-Efficient Federated Learning under Data Scarcity Scenario for Text Summarization
    Zicheng Ma, Zhi Li, Yuanjie Shi, and Jie Chen
    In Proceedings of the Thirty-Ninth AAAI Conference on Artificial Intelligence, 2025
  4. AAAI-DC
    Reliable Uncertainty Quantification in Machine Learning via Conformal Prediction
    Yuanjie Shi
    In AAAI/SIGAI Doctoral Consortium, 2025

2024

  1. NeurIPS
    Conformal Prediction for Class-wise Coverage via Augmented Label Rank Calibration
    Yuanjie Shi, Subhankar Ghosh, Taha Belkhouja, Yan Yan, and Jana Doppa
    In Advances in Neural Information Processing Systems, 2024

2023

  1. UAI
    Probabilistically Robust Conformal Prediction
    Subhankar Ghosh, Yuanjie Shi, Taha Belkhouja, Yan Yan, Jana Doppa, and Brian Jones
    In Proceedings of the Thirty-Ninth Conference on Uncertainty in Artificial IntelligenceSubhankar Ghosh and Yuanjie Shi contributed equally. , 2023