Publications

Bohan Lyu, Yucheng Yang, Siqiao Huang, Jiaru Zhang, Qixin Xu, Xinghan Li, Xinyang Han, Yicheng Zhang, Huaqing Zhang, Runhan Huang, Kaicheng Yang, Zitao Chen, Wentao Guo, Junlin Yang, Xinyue Ai, Wenhao Chai, Yadi Cao, Ziran Yang, Kun Wang, Dapeng Jiang, Huan-ang Gao, Shange Tang, Chengshuai Shi, Simon S. Du, Max Simchowitz, Jiantao Jiao, Dawn Song, Chi Jin (2026). MLS-Bench: A Holistic and Rigorous Assessment of AI Systems on Building Better AI. In NeurIPS (oral).

Cite Code Dataset arXiv Project

Jiaru Zhang, Zeyun Deng, Juanwu Lu, Ziran Wang, Ruqi Zhang (2026). Analytical Correction for Subsampling Bias in Drifting Models. In NeurIPS.

Cite Code arXiv

Can Cui, Jiaru Zhang, Yupeng Zhou, Juntong Peng, Sung-Yeon Park, Zichong Yang, Prashanth Sankaranarayanan, Ruqi Zhang, Ziran Wang (2026). ViLaD: A Large Vision Language Diffusion Framework for End-to-End Autonomous Driving. In IROS.

PDF Video

Jiaru Zhang, Manav Gagvani, Can Cui, Juntong Peng, Ruqi Zhang, Ziran Wang (2026). Efficient and Explainable End-to-End Autonomous Driving via Masked Vision-Language-Action Diffusion. In IROS.

PDF Code Full Text 中文全文

Xiaoyu Wu, Jiaru Zhang, Yang Hua, Bohan Lyu, Hao Wang, Tao Song, Haibing Guan (2026). Exploring Diffusion Models' Corruption Stage in Few-Shot Fine-tuning and Mitigating with Bayesian Neural Networks. In KDD.

PDF Poster arXiv Full Text 中文全文

Jiaru Zhang, Juanwu Lu, Xiaoyu Wu, Ziran Wang, Ruqi Zhang (2026). Accelerating Inference of Discrete Autoregressive Normalizing Flows by Selective Jacobi Decoding. In TMLR (Featured Certification).

PDF Code OpenReview Full Text 中文全文

Pascal Jutras-Dubé, Jiaru Zhang, Ziran Wang, Ruqi Zhang (2026). One-Step Diffusion Samplers via Self-Distillation and Deterministic Flow. In AISTATS.

PDF Code AISTATS OpenReview arXiv

Can Cui, Yunsheng Ma, Sung-Yeon Park, Zichong Yang, Yupeng Zhou, Peiran Liu, Juanwu Lu, Juntong Peng, Jiaru Zhang, Ruqi Zhang, Lingxi Li, Yaobin Chen, Jitesh H. Panchal, Amr Abdelraouf, Rohit Gupta, Kyungtae Han, Ziran Wang (2026). LLM4AD: Large Language Models for Autonomous Driving—Concept, Review, Benchmark, Experiments, and Future Trends. In Proceedings of the IEEE.

PDF Cite DOI IEEE Xplore arXiv Project

Jiaru Zhang, Zesong Wang, Hao Wang, Tao Song, Huai-an Su, Rui Chen, Yang Hua, Xiangwei Zhou, Ruhui Ma, Miao Pan, Haibing Guan (2025). AMPERE: A Generic Energy Estimation Approach for On-Device Training. In ACM SIGMETRICS Performance Evaluation Review.

PDF Cite DOI ACM Digital Library arXiv (THOR preprint)

Qingqian Yang, Peishen Yan, Xiaoyu Wu, Jiaru Zhang, Tao Song, Yang Hua, Hao Wang, Liangliang Wang, Haibing Guan (2025). Stealthy Backdoor Attack in Federated Learning via Adaptive Layer-wise Gradient Alignment. In ICCV.

Xiaoyu Wu, Jiaru Zhang, Steven Wu (2025). Leveraging Model Guidance to Extract Training Data from Personalized Diffusion Models. In ICML.

PDF Code Poster

Jiaru Zhang, Rui Ding, Qiang Fu, Bojun Huang, Zizhen Deng, Yang Hua, Haibing Guan, Shi Han, Dongmei Zhang (2025). Learning Identifiable Structures Helps Avoid Bias in DNN-based Supervised Causal Learning. In AISTATS.

PDF Cite Code Poster arXiv Full Text 中文全文

Xiaoyu Wu, Yang Hua, Chumeng Liang, Jiaru Zhang, Hao Wang, Tao Song, Haibing Guan (2024). CGI-DM: Digital Copyright Authentication for Diffusion Models via Contrasting Gradient Inversion. In CVPR.

Pingchuan Ma, Rui Ding, Qiang Fu, Jiaru Zhang, Shuai Wang, Shi Han, Dongmei Zhang (2024). SPOT: Harnessing Differentiable Causal Discovery in the Presence of Latent Confounders with Skeleton Posterior. In KDD.

Jiaru Zhang, Yang Hua, Tao Song, Hao Wang, Zhengui Xue, Ruhui Ma, Haibing Guan (2023). Information Bound and its Applications in Bayesian Neural Networks. In ECAI.

PDF Cite Code Poster

Chumeng Liang, Xiaoyu Wu, Yang Hua, Jiaru Zhang, Yiming Xue, Tao Song, Zhengui Xue, Ruhui Ma, Haibing Guan (2023). Adversarial Example Does Good: Preventing Painting Imitation from Diffusion Models via Adversarial Examples. In ICML (oral).

PDF Cite Code

Jiaru Zhang, Yang Hua, Tao Song, Hao Wang, Zhengui Xue, Ruhui Ma, Haibing Guan (2022). Improving Bayesian Neural Networks by Adversarial Sampling. In AAAI.

PDF Cite Code Poster Slides Video

Jiaru Zhang, Ruhui Ma, Tao Song, Yang Hua, Zhengui Xue, Chenyang Guan, Haibing Guan (2022). Hierarchical Satellite System Graph for Approximate Nearest Neighbor Search on Big Data. In ACM/IMS Transactions on Data Science.

PDF Cite Code

Jiaru Zhang, Yang Hua, Zhengui Xue, Tao Song, Chengyu Zheng, Ruhui Ma, Haibing Guan (2021). Robust Bayesian Neural Networks by Spectral Expectation Bound Regularization. In CVPR.

PDF Cite Code Poster Slides