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, 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. [Paper] [Code] Yancheng Wang, Yingzhen Yang. Efficient Visual Transformer by Learnable Token Merging. Accepted by 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]