Zhang, Shujian

19 publications

ICLR 2025 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

ICLR 2025 Score Forgetting Distillation: A Swift, Data-Free Method for Machine Unlearning in Diffusion Models Tianqi Chen, Shujian Zhang, Mingyuan Zhou

ICLR 2025 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

NeurIPSW 2024 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

ICML 2024 Sliced Wasserstein with Random-Path Projecting Directions Khai Nguyen, Shujian Zhang, Tam Le, Nhat Ho

ICML 2024 Switchable Decision: Dynamic Neural Generation Networks Shujian Zhang, Korawat Tanwisuth, Chengyue Gong, Pengcheng He, Mingyuan Zhou

ICLR 2023 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

CVPR 2023 FlowGrad: Controlling the Output of Generative ODEs with Gradients Xingchao Liu, Lemeng Wu, Shujian Zhang, Chengyue Gong, Wei Ping, Qiang Liu

ICML 2023 POUF: Prompt-Oriented Unsupervised Fine-Tuning for Large Pre-Trained Models Korawat Tanwisuth, Shujian Zhang, Huangjie Zheng, Pengcheng He, Mingyuan Zhou

NeurIPS 2023 Preference-Grounded Token-Level Guidance for Language Model Fine-Tuning Shentao Yang, Shujian Zhang, Congying Xia, Yihao Feng, Caiming Xiong, Mingyuan Zhou

NeurIPS 2022 A Unified Framework for Alternating Offline Model Training and Policy Learning Shentao Yang, Shujian Zhang, Yihao Feng, Mingyuan Zhou

NeurIPSW 2022 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

NeurIPSW 2022 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

ICML 2022 Regularizing a Model-Based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning Shentao Yang, Yihao Feng, Shujian Zhang, Mingyuan Zhou

NeurIPS 2021 A Prototype-Oriented Framework for Unsupervised Domain Adaptation Korawat Tanwisuth, Xinjie Fan, Huangjie Zheng, Shujian Zhang, Hao Zhang, Bo Chen, Mingyuan Zhou

NeurIPS 2021 Alignment Attention by Matching Key and Query Distributions Shujian Zhang, Xinjie Fan, Huangjie Zheng, Korawat Tanwisuth, Mingyuan Zhou

ICML 2021 Bayesian Attention Belief Networks Shujian Zhang, Xinjie Fan, Bo Chen, Mingyuan Zhou

ICLR 2021 Contextual Dropout: An Efficient Sample-Dependent Dropout Module Xinjie Fan, Shujian Zhang, Korawat Tanwisuth, Xiaoning Qian, Mingyuan Zhou

NeurIPS 2020 Bayesian Attention Modules Xinjie Fan, Shujian Zhang, Bo Chen, Mingyuan Zhou