Publications

@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}
}

@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}
}

@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}
}

@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}
}

@inproceedings{qian2024probabilistically,
  title     = {Probabilistically Rewired Message-Passing Neural Networks},
  author    = {Qian, Chendi and Manolache, Andrei and Ahmed, Kareem and Zeng, Zhe and Van den Broeck, Guy and Niepert, Mathias and Morris, Christopher},
  booktitle = {International Conference on Learning Representations (ICLR)},
  year      = {2024}
}

@phdthesis{zeng2024neurosymbolic,
  title  = {Neurosymbolic Learning and Reasoning for Trustworthy AI},
  author = {Zeng, Zhe},
  school = {University of California, Los Angeles},
  year   = {2024}
}

@inproceedings{zeng2023collapsed,
  title     = {Collapsed Inference for Bayesian Deep Learning},
  author    = {Zeng, Zhe and Van den Broeck, Guy},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year      = {2023}
}

@inproceedings{shukla2023unified,
  title     = {A Unified Approach to Count-Based Weakly Supervised Learning},
  author    = {Shukla, Vinay and Zeng, Zhe and Ahmed, Kareem and Van den Broeck, Guy},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year      = {2023}
}

@inproceedings{ahmed2023simple,
  title     = {SIMPLE: A Gradient Estimator for k-Subset Sampling},
  author    = {Ahmed, Kareem and Zeng, Zhe and Niepert, Mathias and Van den Broeck, Guy},
  booktitle = {International Conference on Learning Representations (ICLR)},
  year      = {2023}
}

@inproceedings{qian2023probabilistic,
  title     = {Probabilistic Task-Adaptive Graph Rewiring},
  author    = {Qian, Chendi and Manolache, Andrei and Ahmed, Kareem and Zeng, Zhe and Van den Broeck, Guy and Niepert, Mathias and Morris, Christopher},
  booktitle = {ICML Workshop on Differentiable Almost Everything},
  year      = {2023}
}

@inproceedings{chen2023unified,
  title     = {A Unified Knowledge Distillation Framework for Deep Directed Graphical Models},
  author    = {Chen, Yizhuo and Liang, Kaizhao and Zeng, Zhe and Yao, Shuochao and Shao, Huajie},
  booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  year      = {2023}
}

@inproceedings{li2021tractable,
  title     = {Tractable Computation of Expected Kernels},
  author    = {Li, Wenzhe and Zeng, Zhe and Vergari, Antonio and Van den Broeck, Guy},
  booktitle = {Conference on Uncertainty in Artificial Intelligence (UAI)},
  year      = {2021}
}

@inproceedings{zeng2021parameter,
  title     = {Is Parameter Learning via Weighted Model Integration Tractable?},
  author    = {Zeng, Zhe and Morettin, Paolo and Yan, Fanqi and Passerini, Andrea and Van den Broeck, Guy},
  booktitle = {Workshop on Tractable Probabilistic Modeling (TPM) at UAI},
  year      = {2021}
}

@inproceedings{zeng2020scaling,
  title     = {Scaling up Hybrid Probabilistic Inference with Logical and Arithmetic Constraints via Message Passing},
  author    = {Zeng, Zhe and Morettin, Paolo and Yan, Fanqi and Vergari, Antonio and Van den Broeck, Guy},
  booktitle = {International Conference on Machine Learning (ICML)},
  year      = {2020}
}

@inproceedings{zeng2020probabilistic,
  title     = {Probabilistic Inference with Algebraic Constraints: Theoretical Limits and Practical Approximations},
  author    = {Zeng, Zhe and Morettin, Paolo and Yan, Fanqi and Vergari, Antonio and Van den Broeck, Guy},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year      = {2020}
}

@inproceedings{zeng2020relax,
  title     = {Relax, Compensate and then Integrate},
  author    = {Zeng, Zhe and Morettin, Paolo and Yan, Fanqi and Vergari, Antonio and Van den Broeck, Guy},
  booktitle = {ECML-PKDD Workshop on Deep Continuous-Discrete Machine Learning},
  year      = {2020}
}

@inproceedings{zeng2019efficient,
  title     = {Efficient Search-Based Weighted Model Integration},
  author    = {Zeng, Zhe and Van den Broeck, Guy},
  booktitle = {Conference on Uncertainty in Artificial Intelligence (UAI)},
  year      = {2019}
}

@inproceedings{zeng2019hybrid,
  title     = {Hybrid Probabilistic Inference with Logical Constraints: Tractability and Message Passing},
  author    = {Zeng, Zhe and Yan, Fanqi and Morettin, Paolo and Vergari, Antonio and Van den Broeck, Guy},
  booktitle = {Workshop on Knowledge Representation \& Reasoning Meets Machine Learning (KR2ML), NeurIPS},
  year      = {2019}
}

@inproceedings{wang2018stein,
  title     = {Stein Variational Message Passing for Continuous Graphical Models},
  author    = {Wang, Dilin and Zeng, Zhe and Liu, Qiang},
  booktitle = {International Conference on Machine Learning (ICML)},
  year      = {2018}
}