Ph.D. Candidate
Electrical and Computer Engineering
Global College
Shanghai Jiao Tong University
wentaochen [at] sjtu [dot] edu [dot] cn

I am a Ph.D. candidate at Global College, Shanghai Jiao Tong University. My advisor is Prof. An Zou. Previously, I worked with Prof. Weimin Zhou from University of Arizona. I received my B.E. degree from Beijing University of Posts and Telecommunications in 2023.

Research: My research interests lie in designing scalable and self-improving AI systems for high-performance computing, including better programming models and systems for domain-specific architectures, as well as optimizing GPU kernels for emerging applications. [Awesome-LLM4Kernel] [KernelPilot] [CUDABench]

Previously, my research was dedicated to exploring mathematical principles of image science, explores modern AI/ML technologies, and utilizes a wide variety of computational tools to advance image formation, perception, and interpretation, including deep generative models for medical image analysis, AI/ML-based image perception and observer designs for objective assessment of image quality, as well as task-oriented image acquisition, reconstruction, and enhancement.

I am always open to research collaborations. Feel free to contact me.

Publications

CUDABench: Benchmarking LLMs for Text-to-CUDA Generation
Jiace Zhu, Wentao Chen, Qi Fan, Zhixing Ren, Junying Wu, Xing Zhe Chai, Chotiwit Rungrueangwutthinon, Yehan Ma, and An Zou
Preprint, 2026
pdf
Medical Image Denoising with Task-based Image-adaptive Denoising CNN
Wentao Chen and Weimin Zhou
Medical Imaging 2026: Image Perception, Observer Performance, and Technology Assessment, 2026
pdf
Diffusion Models with Awareness of Binary Signal Detection Tasks for Medical Image Denoising
Yaozhi Zhang, Wentao Chen, Weimin Zhou, and Yang Liu
Medical Imaging 2026: Image Perception, Observer Performance, and Technology Assessment, 2026
pdf
CUDA-LLM: LLMs Can Write Efficient CUDA Kernels
Wentao Chen, Jiace Zhu, Qi Fan, Yehan Ma, and An Zou
Preprint, 2025
pdf
Ambient Denoising Diffusion Generative Adversarial Networks for Establishing Stochastic Object Models from Noisy Image Data
Xichen Xu, Wentao Chen, and Weimin Zhou
Medical Imaging 2025: Image Perception, Observer Performance, and Technology Assessment, 2025
pdf
Task-based Regularization in Penalized Least-Squares for Binary Signal Detection Tasks in Medical Image Denoising
Wentao Chen, Tianming Xu, and Weimin Zhou
Medical Imaging 2025: Image Perception, Observer Performance, and Technology Assessment, 2025
pdf
MLAE: Masked LoRA Experts for Parameter-Efficient Fine-Tuning
Junjie Wang, Guangjin Yang, Wentao Chen, Huahui Yi, Xiaohu Wu, Zhouchen Lin, and Qicheng Lao
† co-first author
Preprint, 2024
pdf
Ambient-Pix2PixGAN for Translating Medical Images from Noisy Data
Wentao Chen, Xichen Xu, Jie Luo, and Weimin Zhou
Medical Imaging 2024: Image Perception, Observer Performance, and Technology Assessment, 2024
pdf
AmbientCycleGAN for Establishing Interpretable Stochastic Object Models Based on Mathematical Phantoms and Medical Imaging Measurements
Xichen Xu, Wentao Chen, and Weimin Zhou
Medical Imaging 2024: Image Perception, Observer Performance, and Technology Assessment, 2024
pdf
Unsupervised Generation of Pseudo Normal PET from MRI with Diffusion Model for Epileptic Focus Localization
Wentao Chen, Jiwei Li, Xichen Xu, Hui Huang, Siyu Yuan, Miao Zhang, Tianming Xu, Jie Luo, and Weimin Zhou
Medical Imaging 2024: Image Perception, Observer Performance, and Technology Assessment, 2024
pdf
Coarse-to-Fine Covid-19 Segmentation via Vision-Language Alignment
Dandan Shan, Zihan Li, Wentao Chen, Qingde Li, Jie Tian, and Qingqi Hong
† co-first author
IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2023
pdf
Semi-WTC: A Practical Semi-supervised Framework for Attack Categorization through Weight-Task Consistency
Zihan Li, Wentao Chen, Zhiqing Wei, Xingqi Luo, and Bing Su
† co-first author
Preprint, 2022
pdf

Teaching

At Shanghai Jiao Tong University:

  • ECE6501G, Principles of Medical Imaging, Graduate Course TA (Spring 2024).
  • ECE7609J, Principles of Imaging Science, Graduate Course TA (Summer 2023).
  • VE445/ECE4450J, Introduction to Machine Learning, Undergraduate Course TA (Fall 2022, Fall 2023).
  • VE490, Undergraduate Research, Undergraduate Course TA (Fall 2022).

Miscellaneous

  • An ENTJ. I love traveling and reading books.
  • A Real Madrid & DFB fan, also a railway & aviation enthusiast.
  • My hometown is Huainan, a prefecture-level city in north-central Anhui province, China. It is known for its coal industry and thermal power plants. It is also considered to be the hometown and birthplace of tofu.