Publications

Counterfactual probabilistic diffusion with expert models

Published in Arxiv, 2026

Wenhao Mu*, Zhi Cao, Mehmed Uludag, Alexander Rodríguez

Diffusion models as constrained samplers for optimization with unknown constraints

Published in Artificial Intelligence and Statistics (AISTATS), 2025

Lingkai Kong*, Yuanqi Du*, Wenhao Mu*, Kirill Neklyudov, Valentin De Bortoli, Dongxia Wu, Haorui Wang, Aaron Ferber, Yi-An Ma, Carla P Gomes, Chao Zhang

Aligning Large Language Models with Representation Editing: A Control Perspective

Published in Advances in Neural Information Processing Systems (NeurIPS), 2024

Lingkai Kong*, Haorui Wang*, Wenhao Mu*, Yuanqi Du, Yuchen Zhuang, Yifei Zhou, Yue Song, Rongzhi Zhang, Kai Wang, Chao Zhang

Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process

Published in Artificial Intelligence and Statistics (AISTATS), 2024

Lingkai Kong, Haotian Sun, Yuchen Zhuang, Haorui Wang, Wenhao Mu, Chao Zhang

MUBen: benchmarking the uncertainty of pre-trained models for molecular property prediction

Published in Transactions on Machine Learning Research (TMLR), NeurIPS Workshop on AI4Science, 2023

Yinghao Li, Lingkai Kong, Yuanqi Du, Yue Yu, Yuchen Zhuang, Wenhao Mu, Chao Zhang

DF2: Distribution-Free Decision-Focused Learning

Published in preprint, 2023

Lingkai Kong, Wenhao Mu, Jiaming Cui, Yuchen Zhuang, B Aditya Prakash, Bo Dai, Chao Zhang