← Search

Yuhang Lai

6 accepted papers

2026

Step-Level Sparse Autoencoder for Reasoning Process Interpretation

ICML 2026poster

Large Language Models (LLMs) have achieved strong complex reasoning capabilities through Chain-of-Thought (CoT) reasoning. However, their reasoning patterns remain too complicated to analyze. While Sparse Autoencoders (SAEs) have emerged as a powerful tool for interpretability, existing approaches p…

Cited by 0SourceScholar
2025

HAF-RM: A Hybrid Alignment Framework for Reward Model Training

ACL 2025long

The reward model has become increasingly important in alignment, assessment, and data construction for large language models (LLMs). Most existing researchers focus on enhancing reward models through data improvements, following the conventional training framework for reward models that directly opt…

2025

How Jailbreak Defenses Work and Ensemble? A Mechanistic Investigation

EMNLP 2025

Jailbreak attacks, where harmful prompts bypass generative models’ built-in safety, raise serious concerns about model vulnerability. While many defense methods have been proposed, the trade-offs between safety and helpfulness, and their application to Large Vision-Language Models (LVLMs), are not w

Cited by 0SourcePDFScholar
2024

ALaRM: Align Language Models via Hierarchical Rewards Modeling

ACL 2024findings

We introduce ALaRM, the first framework modeling hierarchical rewards in reinforcement learning from human feedback (RLHF), which is designed to enhance the alignment of large language models (LLMs) with human preferences. The framework addresses the limitations of current alignment approaches, whic…

2024

EvoR: Evolving Retrieval for Code Generation

EMNLP 2024finding

Recently the retrieval-augmented generation (RAG) has been successfully applied in code generation. However, existing pipelines for retrieval-augmented code generation (RACG) employ static knowledge bases with a single source, limiting the adaptation capabilities of Large Language Models (LLMs) to d…

2023

DS-1000: A Natural and Reliable Benchmark for Data Science Code Generation

ICML 2023poster

We introduce DS-1000, a code generation benchmark with a thousand data science problems spanning seven Python libraries, such as Numpy and Pandas. Compared to prior works, DS-1000 incorporates three core features. First, our problems reflect diverse, realistic, and practical use cases since we colle…