← Search

Hengzhi Pei

8 accepted papers

2026

Knowledge Distillation for Large Language Models through Residual Learning

ICLR 2026poster

Knowledge distillation has become a crucial technique to transfer the capacities of large language models (LLMs) to smaller, more efficient models for practical deployment. While recent work exploits rich information from intermediate states of the teacher model for more effective knowledge transfer…

Cited by 0SourceScholar
2025

Understanding Silent Data Corruption in LLM Training

ACL 2025long

As the scale of training large language models (LLMs) increases, one emergent failure is silent data corruption (SDC), where hardware produces incorrect computations without explicit failure signals. In this work, we are the first to investigate the impact of real-world SDCs on LLM training by compa…

2024

Fine-tuning Language Models for Joint Rewriting and Completion of Code with Potential Bugs

ACL 2024findings

Handling drafty partial code remains a notable challenge in real-time code suggestion applications. Previous work has demonstrated shortcomings of large language models of code (CodeLLMs) in completing partial code with potential bugs. In this study, we view partial code as implementation hints and…

Cited by 1SourcePDFScholar
2023

Better Context Makes Better Code Language Models: A Case Study on Function Call Argument Completion

AAAI 2023technical

Pretrained code language models have enabled great progress towards program synthesis. However, common approaches only consider in-file local context and thus miss information and constraints imposed by other parts of the codebase and its external dependencies. Existing code completion benchmarks al…

2023

DecodingTrust: A Comprehensive Assessment of Trustworthiness in GPT Models

NeurIPS 2023oral

Generative Pre-trained Transformer (GPT) models have exhibited exciting progress in capabilities, capturing the interest of practitioners and the public alike. Yet, while the literature on the trustworthiness of GPT models remains limited, practitioners have proposed employing capable GPT models for…

2023

Zero-Shot Classification by Logical Reasoning on Natural Language Explanations

ACL 2023findings

Humans can classify data of an unseen category by reasoning on its language explanations. This ability is owing to the compositional nature of language: we can combine previously seen attributes to describe the new category. For example, we might describe a sage thrasher as “it has a slim straight r…

2022

Improving Certified Robustness via Statistical Learning with Logical Reasoning

NeurIPS 2022accept

Intensive algorithmic efforts have been made to enable the rapid improvements of certificated robustness for complex ML models recently. However, current robustness certification methods are only able to certify under a limited perturbation radius. Given that existing pure data-driven statistical ap…

2020

The Secret Revealer: Generative Model-Inversion Attacks Against Deep Neural Networks

CVPR 2020oral

This paper studies model-inversion attacks, in which the access to a model is abused to infer information about the training data. Since its first introduction by [??], such attacks have raised serious concerns given that training data usually contain privacy sensitive information. Thus far, success…

Cited by 614PDFcodeScholar