Zijun Yu PhD · McGill

Publications

† denotes equal contribution.

    ICML · 2026 Accepted

    Beyond Procedure: Substantive Fairness in Conformal Prediction

    Pengqi Liu†, Zijun Yu†, Mouloud Belbahri, Arthur Charpentier, Masoud Asgharian, Jesse C. Cresswell

    • Conformal Prediction
    • Fairness
    arXiv ↗
    In submission · 2026 Under review

    G-CARB: Graph-Localized Conformal Agent Risk Budgets for Compositional Harm

    Zijun Yu†, Yu Gu†, Vahid Partovi Nia, Masoud Asgharian

    • Conformal Prediction
    • Agent Safety
    Preprint coming soon
    In submission · 2026 Under review

    Pause and Reflect: Conformal Aggregation for Chain-of-Thought Reasoning

    Yu Gu†, Zijun Yu†, Vahid Partovi Nia, Masoud Asgharian

    • Conformal Prediction
    • LLM Reasoning
    arXiv ↗
    In submission · 2026 Under review

    Severity-Controlled Prediction Sets for Medication Recommendation

    Yu Gu†, Zijun Yu†, Chi-Kuang Yeh, Xinyu Wang, Ziyang Song

    • Conformal Prediction
    • Healthcare
    Preprint coming soon