Shuai Zhang
Assistant Professor, Data Science
2117 Guttenberg Information Technologies Center (GITC)
About Me
I am an assistant professor in the Department of Data Science, New Jersey Institute of Technology (NJIT), starting at Fall 2023. I earned my Ph.D. in Electrical and Computer Engineering from Rensselaer Polytechnic Institute under the supervision of Dr. Meng Wang. Previously, I was fortunately to work with IBM Thomas J. Watson Research Center and MIT-IBM Watson AI Lab.
Education
Ph.D.
; Rensselaer Polytechnic Institute
; Electrical and Computer Engineering
; 2021
B.E. ; University of Science and Technology of China ; ; 2016
B.E. ; University of Science and Technology of China ; ; 2016
2025 Fall Courses
DS 488 - INDEPENDENT STUDY IN DS
DS 726 - INDEPENDENT STUDY II
DS 675 - MACHINE LEARNING
DS 700B - MASTER'S PROJECT
DS 725 - INDEPENDENT STUDY I
DS 701C - MASTER'S THESIS
DS 790A - DOCT DISSERTATION & RES
DS 792B - PRE-DOCTORAL RESEARCH
CS 792 - PRE-DOCTORAL RESEARCH
DS 701B - MASTER'S THESIS
DS 726 - INDEPENDENT STUDY II
DS 675 - MACHINE LEARNING
DS 700B - MASTER'S PROJECT
DS 725 - INDEPENDENT STUDY I
DS 701C - MASTER'S THESIS
DS 790A - DOCT DISSERTATION & RES
DS 792B - PRE-DOCTORAL RESEARCH
CS 792 - PRE-DOCTORAL RESEARCH
DS 701B - MASTER'S THESIS
Past Courses
CS 732: ADVANCED MACHINE LEARNING
DS 675: MACHINE LEARNING
DS 675: MACHINE LEARNING
Research Interests
My research has been focused on the theoretical foundations of deep learning and the design of principled and fast algorithms for better, safer, and more efficient AI applications. My current research focuses on the theoretical foundation of foundation models and parameter-efficient transfer learning.
Journal Article
Xiaobing Chen, Boyang Zhang, Xiangwei Zhou, Mingxuan Sun, Shuai Zhang, Songyang Zhang, Geoffrey Ye Li.
2025. "Efficient Training of Large-Scale AI Models Through Federated Mixture-of-Experts: A System-Level Approach."
arXiv preprint arXiv:2507.05685 .
Yubo Huang, Xin Lai, Muyang Ye, Anran Zhu, Zixi Wang, Jingzehua Xu, Shuai Zhang, Zhiyuan Zhou, Weijie Niu. 2025. "LHQ-SVC: Lightweight and High Quality Singing Voice Conversion Modeling." ICASSP, 2025 .
Yimian Ding, Jingzehua Xu, Guanwen Xie, Shuai Zhang, Yi Li. 2025. "Make Your AUV Adaptive: An Environment-Aware Reinforcement Learning Framework For Underwater Tasks." IROS, 2025 .
Guanwen Xie, Jingzehua Xu, Yimian Ding, Zhi Zhang, Shuai Zhang, Yi Li. 2025. "Never too Prim to Swim: An LLM-Enhanced RL-based Adaptive S-Surface Controller for AUVs under Extreme Sea Conditions." IROS, 2025 .
Jingzehua Xu, Guanwen Xie, Zekai Zhang, Xiangwang Hou, Shuai Zhang, Yong Ren, Dusit Niyato. 2025. "UPEGSim: An RL-Enabled Simulator for Unmanned Underwater Vehicles Dedicated in the Underwater Pursuit-Evasion Game." IEEE Internet of Things Journal , vol. 12 , no. 3 , pp. 2334--2346.
Yubo Huang, Xin Lai, Muyang Ye, Anran Zhu, Zixi Wang, Jingzehua Xu, Shuai Zhang, Zhiyuan Zhou, Weijie Niu. 2025. "LHQ-SVC: Lightweight and High Quality Singing Voice Conversion Modeling." ICASSP, 2025 .
Yimian Ding, Jingzehua Xu, Guanwen Xie, Shuai Zhang, Yi Li. 2025. "Make Your AUV Adaptive: An Environment-Aware Reinforcement Learning Framework For Underwater Tasks." IROS, 2025 .
Guanwen Xie, Jingzehua Xu, Yimian Ding, Zhi Zhang, Shuai Zhang, Yi Li. 2025. "Never too Prim to Swim: An LLM-Enhanced RL-based Adaptive S-Surface Controller for AUVs under Extreme Sea Conditions." IROS, 2025 .
Jingzehua Xu, Guanwen Xie, Zekai Zhang, Xiangwang Hou, Shuai Zhang, Yong Ren, Dusit Niyato. 2025. "UPEGSim: An RL-Enabled Simulator for Unmanned Underwater Vehicles Dedicated in the Underwater Pursuit-Evasion Game." IEEE Internet of Things Journal , vol. 12 , no. 3 , pp. 2334--2346.
SHOW MORE
Jingzehua Xu, Guanwen Xie, Xinqi Wang, Yimian Ding, Shuai Zhang.
2025. "USV-AUV Collaboration Framework for Underwater Tasks under Extreme Sea Conditions."
ICASSP, 2025 .
Ziqi Zhang, Jingzehua Xu, Jinxin Liu, Zifeng Zhuang, Donglin Wang, Miao Liu, Shuai Zhang. 2024. "Context-former: Stitching via latent conditioned sequence modeling." arXiv preprint arXiv:2401.16452 .
Jingzehua Xu, Yimian Ding, Yiyuan Yang, Guanwen Xie, Shuai Zhang. 2024. "Enhancing Information Freshness: An AoI Optimized Markov Decision Process Dedicated In the Underwater Task." arXiv preprint arXiv:2409.02424 .
Shusen Jing, Anlan Yu, Shuai Zhang, Songyang Zhang. 2024. "FedSC: Provable Federated Self-supervised Learning with Spectral Contrastive Objective over Non-iid Data." ICML, 2024 .
Hongkang Li, Shuai Zhang, Yihua Zhang, Meng Wang, Sijia Liu, Pin-Yu Chen. 2024. "How does promoting the minority fraction affect generalization? a theoretical study of one-hidden-layer neural network on group imbalance." IEEE Journal of Selected Topics in Signal Processing , vol. 18 , no. 2 , pp. 216--231.
Shuai Zhang, Heshan Devaka Fernando, Miao Liu, Keerthiram Murugesan, Songtao Lu, Pin-Yu Chen, Tianyi Chen, Meng Wang. 2024. "SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning." ICML, 2024 .
Yihua Zhang, Hongkang Li, Yuguang Yao, Aochuan Chen, Shuai Zhang, Pin-Yu Chen, Meng Wang, Sijia Liu. 2024. "Visual prompting reimagined: The power of activation prompts." .
Jingzehua Xu, Guanwen Xie, Ziqi Zhang, Xiangwang Hou, Dongfang Ma, Shuai Zhang, Yong Ren, Dusit Niyato. 2023. "Is FISHER All You Need in The Multi-AUV Underwater Target Tracking Task?." arXiv preprint arXiv:2412.03959 .
Guanwen Xie, Jingzehua Xu, Yiyuan Yang, Yimian Ding, Shuai Zhang. 2023. "Large Language Models as Efficient Reward Function Searchers for Custom-Environment Multi-Objective Reinforcement Learning." arXiv preprint arXiv:2409.02428 .
Shuai Zhang, Hongkang Li, Meng Wang, Miao Liu, Pin-Yu Chen, Songtao Lu, Sijia Liu, Keerthiram Murugesan, Subhajit Chaudhury. 2023. "On the Convergence and Sample Complexity Analysis of Deep Q-Networks with $$\backslash$epsilon $-Greedy Exploration." NeurIPS, 2023 .
Mohammed Nowaz Rabbani Chowdhury, Shuai Zhang, Meng Wang, Sijia Liu, Pin-Yu Chen. 2023. "Patch-level Routing in Mixture-of-Experts is Provably Sample-efficient for Convolutional Neural Networks." ICML (Spotlight), 2023 .
Meng Wang, Joe H Chow, Denis Osipov, Stavros Konstantinopoulos, Shuai Zhang, Evangelos Farantatos, Mahendra Patel. 2021. "Review of low-rank data-driven methods applied to synchrophasor measurement." IEEE Open Access Journal of Power and Energy , vol. 8 , pp. 532--542.
Shuai Zhang, Meng Wang, Jinjun Xiong, Sijia Liu, Pin-Yu Chen. 2020. "Improved linear convergence of training cnns with generalizability guarantees: A one-hidden-layer case." IEEE Transactions on Neural Networks and Learning Systems , vol. 32 , no. 6 , pp. 2622--2635.
Shuai Zhang, Meng Wang. 2019. "Correction of corrupted columns through fast robust Hankel matrix completion." IEEE Transactions on Signal Processing , vol. 67 , no. 10 , pp. 2580--2594.
Shuai Zhang, Yingshuai Hao, Meng Wang, Joe H Chow. 2018. "Multichannel Hankel matrix completion through nonconvex optimization." IEEE Journal of Selected Topics in Signal Processing , vol. 12 , no. 4 , pp. 617--632.
Ziqi Zhang, Jingzehua Xu, Jinxin Liu, Zifeng Zhuang, Donglin Wang, Miao Liu, Shuai Zhang. 2024. "Context-former: Stitching via latent conditioned sequence modeling." arXiv preprint arXiv:2401.16452 .
Jingzehua Xu, Yimian Ding, Yiyuan Yang, Guanwen Xie, Shuai Zhang. 2024. "Enhancing Information Freshness: An AoI Optimized Markov Decision Process Dedicated In the Underwater Task." arXiv preprint arXiv:2409.02424 .
Shusen Jing, Anlan Yu, Shuai Zhang, Songyang Zhang. 2024. "FedSC: Provable Federated Self-supervised Learning with Spectral Contrastive Objective over Non-iid Data." ICML, 2024 .
Hongkang Li, Shuai Zhang, Yihua Zhang, Meng Wang, Sijia Liu, Pin-Yu Chen. 2024. "How does promoting the minority fraction affect generalization? a theoretical study of one-hidden-layer neural network on group imbalance." IEEE Journal of Selected Topics in Signal Processing , vol. 18 , no. 2 , pp. 216--231.
Shuai Zhang, Heshan Devaka Fernando, Miao Liu, Keerthiram Murugesan, Songtao Lu, Pin-Yu Chen, Tianyi Chen, Meng Wang. 2024. "SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning." ICML, 2024 .
Yihua Zhang, Hongkang Li, Yuguang Yao, Aochuan Chen, Shuai Zhang, Pin-Yu Chen, Meng Wang, Sijia Liu. 2024. "Visual prompting reimagined: The power of activation prompts." .
Jingzehua Xu, Guanwen Xie, Ziqi Zhang, Xiangwang Hou, Dongfang Ma, Shuai Zhang, Yong Ren, Dusit Niyato. 2023. "Is FISHER All You Need in The Multi-AUV Underwater Target Tracking Task?." arXiv preprint arXiv:2412.03959 .
Guanwen Xie, Jingzehua Xu, Yiyuan Yang, Yimian Ding, Shuai Zhang. 2023. "Large Language Models as Efficient Reward Function Searchers for Custom-Environment Multi-Objective Reinforcement Learning." arXiv preprint arXiv:2409.02428 .
Shuai Zhang, Hongkang Li, Meng Wang, Miao Liu, Pin-Yu Chen, Songtao Lu, Sijia Liu, Keerthiram Murugesan, Subhajit Chaudhury. 2023. "On the Convergence and Sample Complexity Analysis of Deep Q-Networks with $$\backslash$epsilon $-Greedy Exploration." NeurIPS, 2023 .
Mohammed Nowaz Rabbani Chowdhury, Shuai Zhang, Meng Wang, Sijia Liu, Pin-Yu Chen. 2023. "Patch-level Routing in Mixture-of-Experts is Provably Sample-efficient for Convolutional Neural Networks." ICML (Spotlight), 2023 .
Meng Wang, Joe H Chow, Denis Osipov, Stavros Konstantinopoulos, Shuai Zhang, Evangelos Farantatos, Mahendra Patel. 2021. "Review of low-rank data-driven methods applied to synchrophasor measurement." IEEE Open Access Journal of Power and Energy , vol. 8 , pp. 532--542.
Shuai Zhang, Meng Wang, Jinjun Xiong, Sijia Liu, Pin-Yu Chen. 2020. "Improved linear convergence of training cnns with generalizability guarantees: A one-hidden-layer case." IEEE Transactions on Neural Networks and Learning Systems , vol. 32 , no. 6 , pp. 2622--2635.
Shuai Zhang, Meng Wang. 2019. "Correction of corrupted columns through fast robust Hankel matrix completion." IEEE Transactions on Signal Processing , vol. 67 , no. 10 , pp. 2580--2594.
Shuai Zhang, Yingshuai Hao, Meng Wang, Joe H Chow. 2018. "Multichannel Hankel matrix completion through nonconvex optimization." IEEE Journal of Selected Topics in Signal Processing , vol. 12 , no. 4 , pp. 617--632.
COLLAPSE
Conference Proceeding
"Mixture-of-Experts for Distributed Edge Computing with Channel-Aware Gating Function"
2025.
"On the Training Dynamics of Contrastive Learning with Imbalanced Feature Distributions: A Theoretical Study of Feature Learning"
2025.
"Theoretical Guarantees and Training Dynamics of Contrastive Learning: How Misaligned Data Influence Feature Purity"
2025.
"When is Task Vector Provably Effective for Model Editing? A Generalization Analysis of Nonlinear Transformers"
2025.
"Learning on transformers is provable low-rank and sparse: A one-layer analysis"
2024.
2025.
"On the Training Dynamics of Contrastive Learning with Imbalanced Feature Distributions: A Theoretical Study of Feature Learning"
2025.
"Theoretical Guarantees and Training Dynamics of Contrastive Learning: How Misaligned Data Influence Feature Purity"
2025.
"When is Task Vector Provably Effective for Model Editing? A Generalization Analysis of Nonlinear Transformers"
2025.
"Learning on transformers is provable low-rank and sparse: A one-layer analysis"
2024.
SHOW MORE
"Learning and generalization of one-hidden-layer neural networks, going beyond standard gaussian data"
2022.
"Guaranteed convergence of training convolutional neural networks via accelerated gradient descent"
2020.
"A low-rank framework of pmu data recovery and event identification"
2019.
"Correction of simultaneous bad measurements by exploiting the low-rank hankel structure"
2018.
"Multi-channel missing data recovery by exploiting the low-rank hankel structures"
2017.
2022.
"Guaranteed convergence of training convolutional neural networks via accelerated gradient descent"
2020.
"A low-rank framework of pmu data recovery and event identification"
2019.
"Correction of simultaneous bad measurements by exploiting the low-rank hankel structure"
2018.
"Multi-channel missing data recovery by exploiting the low-rank hankel structures"
2017.
COLLAPSE
Conference Paper
"Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning"
ICML 2024, 2024.
"Joint Edge-Model Sparse Learning is Provably Efficient for Graph Neural Networks "
ICLR, January (1st Quarter/Winter), 2023.
"How unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysis"
ICLR, 2022.
"Why lottery ticket wins? a theoretical perspective of sample complexity on sparse neural networks"
Advances in Neural Information Processing Systems (NeurIPS), 2021.
"Fast learning of graph neural networks with guaranteed generalizability: one-hidden-layer case"
International Conference on Machine Learning (ICML), 2020.
ICML 2024, 2024.
"Joint Edge-Model Sparse Learning is Provably Efficient for Graph Neural Networks "
ICLR, January (1st Quarter/Winter), 2023.
"How unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysis"
ICLR, 2022.
"Why lottery ticket wins? a theoretical perspective of sample complexity on sparse neural networks"
Advances in Neural Information Processing Systems (NeurIPS), 2021.
"Fast learning of graph neural networks with guaranteed generalizability: one-hidden-layer case"
International Conference on Machine Learning (ICML), 2020.
Other
"Non-Convex Optimizations for Machine Learning with Theoretical Guarantee: Robust Matrix Completion and Neural Network Learning"
Rensselaer Polytechnic Institute, 2021.
Rensselaer Polytechnic Institute, 2021.