← Search

Lei Sha

18 accepted papers

2026

Large Language Models Struggle with Unreasonability in Math Problems

AAAI 2026technical

Large Language Models (LLMs) have shown remarkable success on a wide range of math and reasoning benchmarks. However, we observe that they often struggle when faced with unreasonable math problems. Instead of recognizing these issues, models frequently proceed as if the problem is well-posed, produc

Cited by 0SourcePDFScholar
2026

MA-RWG: A Multi-Agent Framework for Thematically Structuring and Generation of Related Work

IJCAI 2026

AI-driven survey generation has advanced rapidly, yet related work generation (RWG) remains relatively underexplored. Unlike surveys that provide broad literature overviews, RWG synthesizes prior studies for a single focal paper, requiring contextual fit, cross-paper comparison, and accurate attribu

Cited by 0Scholar
2026

StepCodeReasoner: Aligning Code Reasoning with Stepwise Execution Traces via Reinforcement Learning

ICML 2026poster

Existing code reasoning methods primarily supervise final code outputs, ignoring intermediate states, often leading to reward hacking where correct answers are obtained through inconsistent reasoning. We propose StepCodeReasoner, a framework that introduces explicit intermediate execution-state supe…

Cited by 0SourceScholar
2025

DiffusionAttacker: Diffusion-Driven Prompt Manipulation for LLM Jailbreak

EMNLP 2025

Large Language Models (LLMs) are susceptible to generating harmful content when prompted with carefully crafted inputs, a vulnerability known as LLM jailbreaking. As LLMs become more powerful, studying jailbreak methods is critical to enhancing security and aligning models with human values. Traditi

Cited by 0SourcePDFScholar
2025

How Far are LLMs from Being Our Digital Twins? A Benchmark for Persona-Based Behavior Chain Simulation

ACL 2025finding

Recently, LLMs have garnered increasing attention across academic disciplines for their potential as human digital twins, virtual proxies designed to replicate individuals and autonomously perform tasks such as decision-making, problem-solving, and reasoning on their behalf.However, current evaluati…

2025

LLMs know their vulnerabilities: Uncover Safety Gaps through Natural Distribution Shifts

ACL 2025long

Safety concerns in large language models (LLMs) have gained significant attention due to their exposure to potentially harmful data during pre-training. In this paper, we identify a new safety vulnerability in LLMs: their susceptibility to natural distribution shifts between attack prompts and origi…

2025

Layer-Aware Representation Filtering: Purifying Finetuning Data to Preserve LLM Safety Alignment

EMNLP 2025

With rapid advancement and increasing accessibility of LLMs, fine-tuning aligned models has become a critical step for adapting them to real-world applications, which makes the safety of this fine-tuning process more important than ever. However, recent studies have highlighted a critical challenge:

2025

Omni-MATH: A Universal Olympiad Level Mathematic Benchmark for Large Language Models

ICLR 2025poster

Recent advancements in large language models (LLMs) have led to significant breakthroughs in mathematical reasoning capabilities. However, existing benchmarks like GSM8K or MATH are now being solved with high accuracy (e.g., OpenAI o1 achieves 94.8% on MATH dataset), indicating their inadequacy for…

Cited by 50SourcePDFScholar
2025

Reasoning-to-Defend: Safety-Aware Reasoning Can Defend Large Language Models from Jailbreaking

EMNLP 2025

Large Reasoning Models (LRMs) have recently demonstrated impressive performances across diverse domains. However, how the safety of Large Language Models (LLMs) benefits from enhanced reasoning capabilities against jailbreak queries remains unexplored. To bridge this gap, in this paper, we propose R

2025

Towards Harmonized Uncertainty Estimation for Large Language Models

ACL 2025long

To facilitate robust and trustworthy deployment of large language models (LLMs), it is essential to quantify the reliability of their generations through uncertainty estimation. While recent efforts have made significant advancements by leveraging the internal logic and linguistic features of LLMs t…

Cited by 0SourcePDFScholar
2024

A Stable, Fast, and Fully Automatic Learning Algorithm for Predictive Coding Networks

ICLR 2024poster

Predictive coding networks are neuroscience-inspired models with roots in both Bayesian statistics and neuroscience. Training such models, however, is quite inefficient and unstable. In this work, we show how by simply changing the temporal scheduling of the update rule for the synaptic weights lead…

Cited by 10SourcePDFScholar
2024

ASETF: A Novel Method for Jailbreak Attack on LLMs through Translate Suffix Embeddings

EMNLP 2024main

The safety defense methods of Large language models (LLMs) stays limited because the dangerous prompts are manually curated to just few known attack types, which fails to keep pace with emerging varieties. Recent studies found that attaching suffixes to harmful instructions can hack the defense of L…

Cited by 9SourcePDFScholar
2024

ATM: Adversarial Tuning Multi-agent System Makes a Robust Retrieval-Augmented Generator

EMNLP 2024main

Large language models (LLMs) are proven to benefit a lot from retrieval-augmented generation (RAG) in alleviating hallucinations confronted with knowledge-intensive questions. RAG adopts information retrieval techniques to inject external knowledge from semantic-relevant documents as input contexts.…

2024

Be a Multitude to Itself: A Prompt Evolution Framework for Red Teaming

EMNLP 2024finding

Large Language Models (LLMs) have gained increasing attention for their remarkable capacity, alongside concerns about safety arising from their potential to produce harmful content. Red teaming aims to find prompts that could elicit harmful responses from LLMs, and is essential to discover and mitig…

Cited by 0SourcePDFScholar
2024

ShieldLM: Empowering LLMs as Aligned, Customizable and Explainable Safety Detectors

EMNLP 2024finding

The safety of Large Language Models (LLMs) has gained increasing attention in recent years, but there still lacks a comprehensive approach for detecting safety issues within LLMs’ responses in an aligned, customizable and explainable manner. In this paper, we propose ShieldLM, an LLM-based safety de…

2021

Associative Memories via Predictive Coding

NeurIPS 2021poster

Associative memories in the brain receive and store patterns of activity registered by the sensory neurons, and are able to retrieve them when necessary. Due to their importance in human intelligence, computational models of associative memories have been developed for several decades now. In this p…

Cited by 84SourcePDFScholar
2021

Learning from the Best: Rationalizing Predictions by Adversarial Information Calibration

AAAI 2021technical

Explaining the predictions of AI models is paramount in safety-critical applications, such as in legal or medical domains. One form of explanation for a prediction is an extractive rationale, i.e., a subset of features of an instance that lead the model to give its prediction on the instance. Previo…