← Search

Jiazheng Li

24 accepted papers

2026

QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentation

ICLR 2026poster

Reinforcement learning (RL) has emerged as a central paradigm for training large language models (LLMs) in reasoning tasks. Yet recent studies question RL’s ability to incentivize reasoning capacity beyond the base model. This raises a key challenge: how can RL be adapted to solve harder reasoning p…

Cited by 0SourcecodeScholar
2026

SSA: Sparse Sparse Attention by Aligning Full and Sparse Attention Outputs in Feature Space

ICML 2026poster

Sparse attention reduces the quadratic complexity of full self-attention but faces two challenges: (1) an attention gap, where applying sparse attention to full-attention-trained models causes performance degradation due to train-inference distribution mismatch, and (2) a capability gap, where model…

Cited by 0SourceScholar
2026

Scaling Knowledge Editing in LLMs to 100,000 Facts with Neural KV Database

ICLR 2026poster

Efficiently editing knowledge stored in Large Language Models (LLMs) enables model updates without large-scale training. One promising solution is Locate-and-Edit (L\&E), allowing simultaneous modifications of a massive number of factual knowledge. However, such editing may compromise the general ab…

Cited by 0SourcecodeScholar
2025

An Automated Explainable Educational Assessment System Built on LLMs

AAAI 2025technical

In this demo, we present AERA Chat, an automated and explainable educational assessment system designed for interactive and visual evaluations of student responses. This system leverages large language models (LLMs) to generate automated marking and rationale explanations, addressing the challenge o…

Cited by 0SourcePDFScholar
2025

Data Mixing Can Induce Phase Transitions in Knowledge Acquisition

NeurIPS 2025spotlight

Large Language Models (LLMs) are typically trained on data mixtures: most data come from web scrapes, while a small portion is curated from high-quality sources with dense domain-specific knowledge. In this paper, we show that when training LLMs on such data mixtures, knowledge acquisition from know…

Cited by 0SourceScholar
2025

Drift: Enhancing LLM Faithfulness in Rationale Generation via Dual-Reward Probabilistic Inference

ACL 2025long

As Large Language Models (LLMs) are increasingly applied to complex reasoning tasks, achieving both accurate task performance and faithful explanations becomes crucial. However, LLMs often generate unfaithful explanations, partly because they do not consistently adhere closely to the provided contex…

2025

EnigmaToM: Improve LLMs’ Theory-of-Mind Reasoning Capabilities with Neural Knowledge Base of Entity States

ACL 2025finding

Theory-of-Mind (ToM), the ability to infer others’ perceptions and mental states, is fundamental to human interaction but remains challenging for Large Language Models (LLMs). While existing ToM reasoning methods show promise with reasoning via perceptual perspective-taking, they often rely excessiv…

2025

ExDDI: Explaining Drug-Drug Interaction Predictions with Natural Language

AAAI 2025technical

Predicting unknown drug-drug interactions (DDIs) is crucial for improving medication safety. Previous efforts in DDI prediction have typically focused on binary classification or predicting DDI categories, with the absence of explanatory insights that could enhance trust in these predictions. In thi…

2025

MASS: Mathematical Data Selection via Skill Graphs for Pretraining Large Language Models

ICML 2025poster

High-quality data plays a critical role in the pretraining and fine-tuning of large language models (LLMs), even determining their performance ceiling to some degree. Consequently, numerous data selection methods have been proposed to identify subsets of data that can effectively and efficiently enh…

Cited by 0SourcePDFScholar
2025

Position: LLMs Need a Bayesian Meta-Reasoning Framework for More Robust and Generalizable Reasoning

ICML 2025poster

Large language models (LLMs) excel in many reasoning tasks but continue to face significant challenges, such as lack of robustness in reasoning, struggling with cross-task generalization, and inefficiencies in scaling up reasoning capabilities. Current training paradigms, including next-token predi…

Cited by 0SourcePDFScholar
2025

RoleMRC: A Fine-Grained Composite Benchmark for Role-Playing and Instruction-Following

ACL 2025finding

Role-playing is important for Large Language Models (LLMs) to follow diverse instructions while maintaining role identity and the role’s pre-defined ability limits. Existing role-playing datasets mostly contribute to controlling role style and knowledge boundaries, but overlook role-playing in instr…

2025

Two Heads Are Better Than One: Dual-Model Verbal Reflection at Inference-Time

EMNLP 2025

Although preference optimization methods have improved reasoning performance in Large Language Models (LLMs), they often lack transparency regarding why one reasoning outcome is preferred over another. This limitation is especially critical in Automated Student Answer Scoring (ASAS), where explainab

2024

Calibrating LLMs with Preference Optimization on Thought Trees for Generating Rationale in Science Question Scoring

EMNLP 2024finding

Generating rationales that justify scoring decisions has been a promising way to facilitate explainability in automated scoring systems. However, existing methods do not match the accuracy of classifier-based methods. Plus, the generated rationales often contain hallucinated information. To address…

2024

Eliminating Biased Length Reliance of Direct Preference Optimization via Down-Sampled KL Divergence

EMNLP 2024main

Direct Preference Optimization (DPO) has emerged as a prominent algorithm for the direct and robust alignment of Large Language Models (LLMs) with human preferences, offering a more straightforward alternative to the complex Reinforcement Learning from Human Feedback (RLHF). Despite its promising ef…

2024

Large Language Models Fall Short: Understanding Complex Relationships in Detective Narratives

ACL 2024findings

Existing datasets for narrative understanding often fail to represent the complexity and uncertainty of relationships in real-life social scenarios. To address this gap, we introduce a new benchmark, Conan, designed for extracting and analysing intricate character relation graphs from detective narr…

2024

NarrativePlay: An Automated System for Crafting Visual Worlds in Novels for Role-Playing

AAAI 2024technical

In this demo, we present NarrativePlay -- an innovative system enabling users to role-play a fictional character and interact with dynamically generated narrative environments. Unlike existing predefined sandbox approaches, NarrativePlay centres around the main storyline events extracted from the na…

2024

The Mystery of In-Context Learning: A Comprehensive Survey on Interpretation and Analysis

EMNLP 2024main

Understanding in-context learning (ICL) capability that enables large language models (LLMs) to excel in proficiency through demonstration examples is of utmost importance. This importance stems not only from the better utilization of this capability across various tasks, but also from the proactive…

2023

CUE: An Uncertainty Interpretation Framework for Text Classifiers Built on Pre-Trained Language Models

UAI 2023poster

Text classifiers built on Pre-trained Language Models (PLMs) have achieved remarkable progress in various tasks including sentiment analysis, natural language inference, and question-answering. However, the occurrence of uncertain predictions by these classifiers poses a challenge to their reliabili…

2023

Distilling ChatGPT for Explainable Automated Student Answer Assessment

EMNLP 2023long findings

Providing explainable and faithful feedback is crucial for automated student answer assessment. In this paper, we introduce a novel framework that explores using ChatGPT, a cutting-edge large language model, for the concurrent tasks of student answer scoring and rationale generation. We identify the…

Cited by 0SourcecodeScholar
2022

NumHTML: Numeric-Oriented Hierarchical Transformer Model for Multi-Task Financial Forecasting

AAAI 2022technical

Financial forecasting has been an important and active area of machine learning research because of the challenges it presents and the potential rewards that even minor improvements in prediction accuracy or forecasting may entail. Traditionally, financial forecasting has heavily relied on quantitat…

Cited by 45SourcePDFScholar
2022

PHEE: A Dataset for Pharmacovigilance Event Extraction from Text

EMNLP 2022main

The primary goal of drug safety researchers and regulators is to promptly identify adverse drug reactions. Doing so may in turn prevent or reduce the harm to patients and ultimately improve public health. Evaluating and monitoring drug safety (i.e., pharmacovigilance) involves analyzing an ever grow…

2021

Exploring the Efficacy of Automatically Generated Counterfactuals for Sentiment Analysis

ACL 2021long

While state-of-the-art NLP models have been achieving the excellent performance of a wide range of tasks in recent years, important questions are being raised about their robustness and their underlying sensitivity to systematic biases that may exist in their training and test data. Such issues come…