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
|
|
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
|
|
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
|
|
CUDA-LLM: LLMs Can Write Efficient CUDA Kernels
Wentao Chen,
Jiace Zhu,
Qi Fan,
Yehan Ma,
and
An Zou Preprint, 2025
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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.