| 2026 | ACL | SFTMix: Elevating Language Model Instruction Tuning with Mixup Recipe. | Yuxin Xiao, Shujian Zhang, Marzyeh Ghassemi, Wenxuan Zhou |
| 2025 | ACL | T-REG: Preference Optimization with Token-Level Reward Regularization. | Wenxuan Zhou, Shujian Zhang, Lingxiao Zhao, Tao Meng |
| 2025 | ICLR | Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Models. | Tianqi Chen, Shujian Zhang, Mingyuan Zhou |
| 2025 | ICLR | Statistical Advantages of Perturbing Cosine Router in Mixture of Experts. | Huy Nguyen, Pedram Akbarian, Huyen Trang Pham, Thien Trang Nguyen Vu, Shujian Zhang, Nhat Ho |
| 2025 | ICLR | Instructional Segment Embedding: Improving LLM Safety with Instruction Hierarchy. | Tong Wu, Shujian Zhang, Kaiqiang Song, Silei Xu, Sanqiang Zhao, Ravi Agrawal, Sathish Reddy Indurthi, Chong Xiang, Prateek Mittal, Wenxuan Zhou |
| 2024 | EMNLP | WPO: Enhancing RLHF with Weighted Preference Optimization. | Wenxuan Zhou, Ravi Agrawal, Shujian Zhang, Sathish Reddy Indurthi, Sanqiang Zhao, Kaiqiang Song, Silei Xu, Chenguang Zhu |
| 2024 | ICML | Sliced Wasserstein with Random-Path Projecting Directions. | Khai Nguyen, Shujian Zhang, Tam Le, Nhat Ho |
| 2024 | ICML | Switchable Decision: Dynamic Neural Generation Networks. | Shujian Zhang, Korawat Tanwisuth, Chengyue Gong, Pengcheng He, Mingyuan Zhou |
| 2024 | NAACL | LanguageFlow: Advancing Diffusion Language Generation with Probabilistic Flows. | Shujian Zhang, Lemeng Wu, Chengyue Gong, Xingchao Liu |
| 2023 | CVPR | FlowGrad: Controlling the Output of Generative ODEs with Gradients. | Xingchao Liu, Lemeng Wu, Shujian Zhang, Chengyue Gong, Wei Ping, Qiang Liu |
| 2023 | ICLR | Fantastic Rewards and How to Tame Them: A Case Study on Reward Learning for Task-oriented Dialogue Systems. | Yihao Feng, Shentao Yang, Shujian Zhang, Jianguo Zhang, Caiming Xiong, Mingyuan Zhou, Huan Wang |
| 2023 | ICML | POUF: Prompt-Oriented Unsupervised Fine-tuning for Large Pre-trained Models. | Korawat Tanwisuth, Shujian Zhang, Huangjie Zheng, Pengcheng He, Mingyuan Zhou |
| 2022 | EMNLP | Passage-Mask: A Learnable Regularization Strategy for Retriever-Reader Models. | Shujian Zhang, Chengyue Gong, Xingchao Liu |
| 2022 | ICML | Regularizing a Model-based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning. | Shentao Yang, Yihao Feng, Shujian Zhang, Mingyuan Zhou |
| 2022 | NAACL | ALLSH: Active Learning Guided by Local Sensitivity and Hardness. | Shujian Zhang, Chengyue Gong, Xingchao Liu, Pengcheng He, Weizhu Chen, Mingyuan Zhou |
| 2021 | ACL | Knowing More About Questions Can Help: Improving Calibration in Question Answering. | Shujian Zhang, Chengyue Gong, Eunsol Choi |
| 2021 | EMNLP | Learning with Different Amounts of Annotation: From Zero to Many Labels. | Shujian Zhang, Chengyue Gong, Eunsol Choi |
| 2021 | ICLR | Contextual Dropout: An Efficient Sample-Dependent Dropout Module. | Xinjie Fan, Shujian Zhang, Korawat Tanwisuth, Xiaoning Qian, Mingyuan Zhou |
| 2021 | ICML | Bayesian Attention Belief Networks. | Shujian Zhang, Xinjie Fan, Bo Chen, Mingyuan Zhou |
| 1994 | ISCAS | Why Cellular Automata are better than LFSRs as Built-in Self-test Generators for Sequential-type Faults. | Shujian Zhang, Rod Byrne, Jon C. Muzio, D. Michael Miller |
| 1992 | ICCD | BIST Generators for Sequential Faults. | Shujian Zhang, Rod Byrne, D. Michael Miller |
| 1990 | ICCD | Estimating aliasing in CA and LFSR based signature registers. | D. Michael Miller, Shujian Zhang, Werner Pries, Robert D. McLeod |