Hengzhi Pei
Hengzhi Pei is an applied scientist at Amazon who works on the trustworthiness and reliability of large language models. He coauthored DecodingTrust, a benchmark that tests GPT-4 and GPT-3.5 for toxicity, stereotype bias, privacy, machine ethics and fairness, which won an Outstanding Datasets and Benchmarks Paper award at NeurIPS 2023. Earlier papers covered multi-criteria Chinese word segmentation and the Secret Revealer model-inversion attack in 2019, and the RTSGAN time-series generator in 2021. At the University of Illinois at Urbana-Champaign, he completed a master's degree in computer science in 2023 with Bo Li as advisor, writing a thesis on TextGuard, a provable defense against backdoor attacks on text classification. His Amazon papers study silent data corruption in LLM training, knowledge distillation and LLM inference simulation.
Timeline
-
2019
Multi-criteria Chinese word segmentation
In 2019, he coauthors with Xipeng Qiu, Hang Yan and Xuanjing Huang a unified Transformer-encoder model for multi-criteria Chinese word segmentation. The model segments simplified and traditional Chinese text according to a criterion token.
-
2019
The Secret Revealer attack
With Yuheng Zhang, Ruoxi Jia, Wenxiao Wang, Bo Li and Dawn Song, he coauthors "The Secret Revealer" in 2019. The paper presents a generative model-inversion attack that uses generative adversarial networks to invert deep neural networks and reconstruct face images from a face recognition model.
-
2021
RTSGAN for real-world time series
In 2021, he is first author of "Towards Generating Real-World Time Series Data" with Kan Ren, Yuqing Yang, Chang Liu, Tao Qin and Dongsheng Li. The paper proposes RTSGAN, a generative framework for long, variable-length time series with missing values.
-
2023
Context for code language models
At AAAI 2023, he is first author of "Better context makes better code language models," a case study on function call argument completion written with Jinman Zhao, Leonard Lausen, Sheng Zha and George Karypis. The work curates a new dataset of permissively licensed Python code.
-
2023
A master's degree at Illinois
He earns an M.S. in computer science at the University of Illinois at Urbana-Champaign in 2023, advised by Bo Li. His thesis presents TextGuard, a provable defense against backdoor attacks on text classification that splits training data into disjoint subsets and ensembles the resulting classifiers.
-
2023
A NeurIPS Outstanding Paper Award
DecodingTrust, which he coauthors with Boxin Wang, Weixin Chen, Bo Li and others, wins an Outstanding Datasets and Benchmarks Paper award at NeurIPS 2023 in the benchmark category. The benchmark evaluates GPT-4 and GPT-3.5 for toxicity, stereotype bias, adversarial robustness, privacy, machine ethics and fairness.
-
2025
Silent data corruption and Amazon Nova
At ACL 2025, he coauthors "Understanding silent data corruption in LLM training" with Jeffrey Ma, Leonard Lausen and George Karypis, comparing training on healthy production nodes with unhealthy nodes that show silent data corruption. In 2025, he is also among the authors of the technical report and model card for Amazon Nova, a family of foundation models.
-
As of October 2026
Applied scientist at Amazon
He is an applied scientist at Amazon. His 2026 papers include an IEEE Micro study of silent data corruption across training characteristics, an ICLR 2026 paper on knowledge distillation for large language models through residual learning, and KernelSight-LM, a kernel-level LLM inference simulator.
Career
- NowAmazonApplied Scientist
Education
Sources and credits
Photo: We have no published photo of Hengzhi Pei yet. Send us a free-licensed photo.
- A Concise Model for Multi-Criteria Chinese Word Segmentation with Transformer Encoder
- The Secret Revealer: Generative Model-Inversion Attacks Against Deep Neural Networks
- Towards Generating Real-World Time Series Data
- Hengzhi Pei - Amazon Science
- TextGuard: Provable defense against backdoor attacks on text classification | IDEALS
- Announcing the NeurIPS 2023 Paper Awards – NeurIPS Blog
- NeurIPS 2023 Awards
- DecodingTrust: A Comprehensive Assessment of Trustworthiness in GPT Models
- The Amazon Nova Family of Models: Technical Report and Model Card
- KernelSight-LM: A Kernel-Level LLM Inference Simulator


