Statistical Deep Learning (SDL) Lab


General Goal

In SDL, we focus on both theoretical foundation of machine learning/deep learning and theory-inspired applications to broad aspects of deep learning, including transformers, graph learning, neural architecture search, model compression and edge computing, etc.

Current PhD students: Yancheng Wang, Bowen Xi

Undergraduate students: Marko Jojic

Recent Publications

Yancheng Wang, Yingzhen Yang.
Low-Rank Graph Contrastive Learning for Noisy Node Classification.
Transactions on Machine Learning Research (TMLR) 2025. [Paper]

Yancheng Wang, Changyu Liu, Yingzhen Yang.
Diffusion on Graph: Augmentation of Graph Structure for Node Classification.
Transactions on Machine Learning Research (TMLR) 2025. [Paper] [Code]

Yancheng Wang, Rajeev Goel, Marko Jojic, Alvin C Silva, Teresa Wu, Yingzhen Yang.
Informative Synthetic Data Generation for Thorax Disease Classification.
Conference on Uncertainty in Artificial Intelligence (UAI) 2025.

Yancheng Wang, Yingzhen Yang.
Efficient Vision Transformer by Informative Token Merging.
Under Minor Revision for IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI) 2025. [Paper] [Code]

Yancheng Wang, Rajeev Goel, Utkarsh Nath, Alvin C Silva, Teresa Wu, Yingzhen Yang.
Learning Low-Rank Feature for Thorax Disease Classification.
Advances in Neural Information Processing Systems (NeurIPS) 2024. [Paper] [Code]

Yancheng Wang, Ping Li, Yingzhen Yang.
Visual Transformer with Differentiable Channel Selection: An Information Bottleneck Inspired Approach.
International Conference on Machine Learning (ICML) 2024. [Paper] [Code]

Utkarsh Nath, Yancheng Wang, Pavan Turaga, Yingzhen Yang.
RNAS-CL: Robust Neural Architecture Search by Cross-Layer Knowledge Distillation.
International Journal of Computer Vision (IJCV) 2024. [Paper] [Code]

Utkarsh Nath, Yancheng Wang, Yingzhen Yang.
Neural Architecture Search Finds Robust Models by Knowledge Distillation.
Conference on Uncertainty in Artificial Intelligence (UAI) 2024. [Paper] [Code]

Yancheng Wang, Ziyan Jiang, Zheng Chen, Fan Yang, Yingxue Zhou, Eunah Cho, Xing Fan, Xiaojiang Huang, Yanbin Lu, Yingzhen Yang.
RecMind: Large language Model Powered Agent for Recommendation.
Annual Conference of the North American Chapter of the Association for Computational Linguistics (NAACL) 2024. [Paper]

Utkarsh Nath, Rajeev Goel, Eun Som Jeon, Changhoon Kim, Kyle Min, Yezhou Yang, Yingzhen Yang, Pavan Turaga
Deep Geometric Moments Promote Shape Consistency in Text-to-3D Generation.
IEEE Winter Conference on Applications of Computer Vision (WACV) 2025. [Paper]