Trustworthy AI through Knowledge and Optimization Lab

News

• Apr 2026 paper
  Two papers accepted to ICML 2026: ProbMoE: Differentiable Probabilistic Routing for Mixture-of-Experts and Automatic Layer Selection for Hallucination Detection 🎉
• Apr 2026 paper
  One paper accepted to NeSy 2026: Learning Deep Generative Models under Hard Linear Equality Constraints 🎉
• Feb 2026 funding
  We received the DAC award. Thank you UVA RC!
• Aug 2025 team
  Welcome Zhizhen, Heng and Xinpeng to the group 🥳

• ProbMoE: Differentiable Probabilistic Routing for Mixture-of-Experts

  Heng Zhao, Zilei Shao, Guy Van den Broeck, Zhe Zeng

  International Conference on Machine Learning (ICML) , 2026

  PDF Code BibTeX
  Copy

  @article{zhao2026probmoe,
    title   = {ProbMoE: Differentiable Probabilistic Routing for Mixture-of-Experts},
    author  = {Zhao, Heng and Shao, Zilei and Van den Broeck, Guy and Zeng, Zhe},
    journal = {arXiv preprint arXiv:2606.01509},
    year    = {2026}
  }

• Automatic Layer Selection for Hallucination Detection

  Xinpeng Wang, William X. Cao, Andrew Gordon Wilson, Zhe Zeng

  International Conference on Machine Learning (ICML) , 2026

  PDF Code BibTeX
  Copy

  @article{wang2026automatic,
    title   = {Automatic Layer Selection for Hallucination Detection},
    author  = {Wang, Xinpeng and Cao, William X. and Wilson, Andrew Gordon and Zeng, Zhe},
    journal = {arXiv preprint arXiv:2605.26366},
    year    = {2026}
  }

• Learning Deep Generative Models under Hard Linear Equality Constraints

  Ruoyan Li, Dipti Ranjan Sahu, Guy Van den Broeck, Zhe Zeng

  Proceedings of Machine Learning Research (PMLR) , 2026

  PDF BibTeX
  Copy

  @inproceedings{li2026learning,
    title     = {Learning Deep Generative Models under Hard Linear Equality Constraints},
    author    = {Li, Ruoyan and Sahu, Dipti Ranjan and Van den Broeck, Guy and Zeng, Zhe},
    booktitle = {Proceedings of Machine Learning Research (PMLR)},
    year      = {2026}
  }

• The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learning

  Wenqian Ye, Luyang Jiang, Eric Xie, Guangtao Zheng, Yunsheng Ma, Xu Cao, Dongliang Guo, Daiqing Qi, Zeyu He, Yijun Tian, Megan Coffee, Zhe Zeng, Sheng Li, Ting-Hao Kenneth Huang, Ziran Wang, James M. Rehg, Henry Kautz, Andrew Gordon Wilson, Aidong Zhang

  Transactions on Machine Learning Research , 2026

  PDF BibTeX
  Copy

  @article{ye2024clever,
    title   = {The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learning},
    author  = {Ye, Wenqian and Jiang, Luyang and Xie, Eric and Zheng, Guangtao and Ma, Yunsheng and Cao, Xu and Guo, Dongliang and Qi, Daiqing and He, Zeyu and Tian, Yijun and Coffee, Megan and Zeng, Zhe and Li, Sheng and Huang, Ting-Hao Kenneth and Wang, Ziran and Rehg, James M. and Kautz, Henry and Wilson, Andrew Gordon and Zhang, Aidong},
    journal = {arXiv preprint arXiv:2402.12715},
    year    = {2024}
  }