About Me
I am currently a Ph.D. student in Computer Science at North Carolina State University, where I am fortunate to be advised by Prof. Xiaorui Liu. Prior to that, I received my M.S. degree in Mathematics from the Chinese Academy of Sciences, and my B.S. degree in Mathematics from Beijing Normal University. I have gained diverse industry research experience at Trustworthy Shopping at Amazon (2026), Automated Reasoning at AWS AI (2025), Amazon GuardDuty (2024), Baidu, Inc. (2023), and Tsinghua AIR (2022).
My research focuses on building scalable trustworthy AI systems that stay reliable across dynamic environments while remaining efficient to train, deploy, and store, which can be divided into three pieces:
-
▶ Architecture Level β Robustness-Informed Neural Networks (RINNs) Framework
Grounded in the theoretical foundations of robust statistics and non-convex optimization, I propose robustness-informed neural networks (RINNs), a principled framework applicable across diverse architectures and domains. I instantiate RINNs for transformers via robust attention [ProTransformer, NeurIPS'24], for graph neural networks via unbiased robust aggregation [RUNG, NeurIPS'24], and for convolutional networks via robust dictionary structure [EDLNets].
-
▶ Deployment Level β Training Acceleration · Resource Limit · Dynamic Risk
Real-world deployment is constrained by limited training budgets, heterogeneous compute, and ever-changing environments, raising three practical questions I pursue:
- How to accelerate robust training? I cut the cost of robust training by removing redundancy across iterative attack/defense steps, matching standard adversarial training's robustness with only 30% of the training budget [Spiking Attacks, ICLR'26].
- How to robustify models with limited resource? I enable resource-adaptive robustification of pretrained models through training-free integration [ProTransformer, NeurIPS'24] and versatile robustness reprogramming across paradigms spanning training-free adaptation to full fine-tuning [Robustness Reprogramming, ICLR'25].
- How can a single model handle dynamic environments? I train one all-in-one model conditioned on risk type, so new risks are met with a lightweight adapter rather than retraining the backbone [CAN].
-
▶ Computing Level β Algorithmic Acceleration · Kernel Efficiency · Precision Reduction
I systematically cut computational overhead across all three levels:
- Algorithm-wise: I eliminate redundant operations across iterative trustworthy-AI procedures, accelerating iterative attacks and adversarial training [Spiking Attacks, ICLR'26], randomized smoothing [SRS, NeurIPS'24], and memory-enhanced graph learning [IncAgg-GNN, NeurIPS'26].
- Kernel-wise: I design hardware-aware GPU kernels that minimize memory traffic, fusing robust reweighting into IO-aware attention kernels to safeguard LLM agents in real time [ProFlashAttention].
- Precision-wise: I compute with fewer bits by quantizing the small, stabilizing updates in iterative procedures rather than full activations, as in accelerated generative modeling [ModDiff, ICML'25], extending to quantized iterative attacks and adversarial training.
News
Selected Publications
Full Publications →Robustness Reprogramming for Representation Learning
Fine-Grained Iterative Adversarial Attacks with Limited Computation Budget
ProTransformer: Robustify Transformers via Plug-and-Play Paradigm
Equivariant Spatio-Temporal Attentive Graph Networks to Simulate Physical Dynamics
Boosting Adversarial Robustness and Generalization with Dictionary Structure
Towards One-for-All Robustness Across a Continuum of Threat Levels
Internship Experience
Applied Scientist Intern
Amazon Web ServicesTrustworthy Shopping AI Team SEATTLE, WA
Collaborators: Rui Song, Lingdao Shao, Xueyu Mao, Yang Liu, Peijie Qiu
Building token- and context-efficient LLM systems for modeling behavioral and structured data. → TEFM , CEMBS (pending for final approval).
Applied Scientist Intern
Amazon Web ServicesAutomated Reasoning at AWS AI NEW YORK, NY
Collaborators: Ferhat Erata, MohamadAli Torkamani
Improving natural language to logical formula translation and reasoning fidelity for LLM-based automated reasoning. → Amazon Research Award , SCD .
Applied Scientist Intern
Amazon Web ServicesSecurity Analytics and AI Research (SAAR) NEW YORK, NY
Collaborator: MohamadAli Torkamani
Designing efficient hierarchical transformer architectures for representing and analyzing large-scale log data. → HLogformer .
Research Intern
Tsinghua UniversityInstitute for AI Industry Research BEIJING, CHINA
Collaborators: Wenbing Huang, Yu Rong
Developing equivariant spatio-temporal graph neural networks to simulate physical dynamics. → ESTAG [NeurIPS 2023] .
Ball Is Life 
Brotherhood is why we play.
🏆 Champion I
🏆 Champion II