MUBen: Benchmarking the Uncertainty of Pre-Trained Models for Molecular Property Prediction. - Yinghao Li; Lingkai Kong; Yuanqi Du; Yue Yu 001; Yuchen Zhuang; Wenhao Mu; Chao Zhang 014 (CoRR, 2023)

DF2: Distribution-Free Decision-Focused Learning. - Lingkai Kong; Wenhao Mu; Jiaming Cui; Yuchen Zhuang; B. Aditya Prakash; Bo Dai 001; Chao Zhang 014 (CoRR, 2023)

Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process. - Lingkai Kong; Haotian Sun; Yuchen Zhuang; Haorui Wang; Wenhao Mu; Chao Zhang 014 (AISTATS, 2024)

Aligning Large Language Models with Representation Editing: A Control Perspective. - Lingkai Kong; Haorui Wang; Wenhao Mu; Yuanqi Du; Yuchen Zhuang; Yifei Zhou; Yue Song; Rongzhi Zhang; Kai Wang; Chao Zhang 014 (NeurIPS, 2024)

MUBen: Benchmarking the Uncertainty of Molecular Representation Models. - Yinghao Li; Lingkai Kong; Yuanqi Du; Yue Yu 001; Yuchen Zhuang; Wenhao Mu; Chao Zhang 014 (Trans. Mach. Learn. Res., 2024)

Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints. - Lingkai Kong; Yuanqi Du; Wenhao Mu; Kirill Neklyudov; Valentin De Bortoli; Dongxia Wu; Haorui Wang; Aaron M. Ferber; Yian Ma; Carla P. Gomes; Chao Zhang 014 (AISTATS, 2025)

DF2: Distribution-Free Decision-Focused Learning. - Lingkai Kong; Wenhao Mu; Jiaming Cui; Yuchen Zhuang; B. Aditya Prakash; Bo Dai 001; Chao Zhang 014 (UAI, 2025)