← Search

An Zhang

46 accepted papers

2026

AgentNoiseBench: Benchmarking Robustness of Tool-Using LLM Agents Under Noisy Condition

ICML 2026poster

As LLM-based agents are increasingly deployed in real-world workflows, existing agent benchmarks---often built on idealized, noise-free assumptions---fall short of characterizing agents' robustness under imperfect user instructions and unreliable tool feedback. To address this gap, we introduce **Ag…

Cited by 0SourceScholar
2026

AlphaAlign: Incentivizing Safety Alignment with Extremely Simplified Reinforcement Learning

ICLR 2026poster

Large language models (LLMs), despite possessing latent safety understanding from their vast pretraining data, remain vulnerable to generating harmful content and exhibit issues such as over-refusal and utility degradation after safety alignment. Current safety alignment methods often result in supe…

Cited by 0SourceScholar
2026

AlphaSteer: Learning Refusal Steering with Principled Null-Space Constraint

ICLR 2026poster

As LLMs are increasingly deployed in real-world applications, ensuring their ability to refuse malicious prompts, especially jailbreak attacks, is essential for safe and reliable use. Recently, activation steering has emerged as an effective approach for enhancing LLM safety by adding a refusal dire…

Cited by 0SourcecodeScholar
2026

Denoising Neural Reranker for Recommender Systems

ICLR 2026poster

For multi-stage recommenders in industry, a user request would first trigger a simple and efficient retriever module that selects and ranks a list of relevant items, then the recommender calls a slower but more sophisticated reranking model that refines the item list exposure to the user. To consist…

Cited by 0SourcecodeScholar
2026

Internalizing Safety Understanding in Large Reasoning Models via Verification

ICML 2026poster

While explicit Chain-of-Thought (CoT) empowers large reasoning models (LRMs), it enables the generation of riskier final answers. Current alignment paradigms primarily rely on externally enforced compliance, optimizing models to detect malicious prompts rather than evaluating the safety of their own…

Cited by 0SourceScholar
2026

Learning to Self-Verify Makes Language Models Better Reasoners

ICML 2026poster

Recent large language models (LLMs) achieve strong performance in generating promising reasoning paths for complex tasks. However, despite powerful generation ability, LLMs remain weak at verifying their own answers, revealing a persistent capability asymmetry between generation and self-verificatio…

Cited by 0SourceScholar
2026

Look Back to Reason Forward: Revisitable Memory for Long-Context LLM Agents

ICLR 2026poster

Large language models face challenges in long-context question answering, where key evidence of a query may be dispersed across millions of tokens. Existing works equip large language models with a memory corpus that is dynamically updated during a single-pass document scan, also known as the "memor…

Cited by 0SourcecodeScholar
2026

MemOCR: Layout-Aware Visual Memory for Efficient Long-Horizon Reasoning

ICML 2026poster

Long-horizon agentic reasoning necessitates effectively compressing growing interaction histories into a limited context window. Most existing memory systems serialize history as text, where token-level cost is uniform and scales linearly with length, often spending scarce budget on low-value detail…

Cited by 0SourceScholar
2026

Quantile Advantage Estimation for Entropy-Safe Reasoning

ICLR 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) strengthens LLM reasoning but training often oscillates between {entropy collapse} and {entropy explosion}. We trace both hazards to the mean-baseline used in value-free RL (\eg GRPO/DAPO), which improperly penalizes negative-advantage samples un…

Cited by 0SourcecodeScholar
2026

Reasoning Can Be Restored by Correcting a Few Decision Tokens

ICML 2026poster

Large reasoning models (LRMs) substantially outperform their base LLM counterparts on challenging reasoning benchmarks, yet it remains poorly understood where base models go wrong during token-by-token generation and how to narrow this gap efficiently. We study the base–reasoning gap by quantifying …

Cited by 0SourceScholar
2026

SafeSieve: From Heuristics to Experience in Progressive Pruning for LLM-based Multi-Agent Communication

AAAI 2026technical

LLM-based multi-agent systems exhibit strong collaborative capabilities but often suffer from redundant communication and excessive token overhead. Existing methods typically enhance efficiency through pretrained GNNs or greedy algorithms, but often isolate pre- and post-task optimization, lacking a

Cited by 0SourcePDFScholar
2026

Tiny Brains, Giant Impact: Uncovering the Keystone Neurons of LLM with Just a Few Prompts

ICML 2026poster

Large language models (LLMs) display strong comprehensive abilities, yet the internal mechanisms that support these behaviors remain insufficiently understood. In this work, we show that across a wide range of open-weight Transformers, a subset of neurons remains consistently highly activated during…

Cited by 0SourceScholar
2026

Transport and Merge: Cross-Architecture Merging for Large Language Models

ICML 2026poster

Large language models (LLMs) achieve strong capabilities by scaling model capacity and training data, yet many real-world deployments rely on smaller models trained or adapted from low-resource data. This gap motivates the need for mechanisms to transfer knowledge from large, high-resource models to…

Cited by 0SourceScholar
2026

When to Memorize and When to Stop: Gated Recurrent Memory for Long-Context Reasoning

ICML 2026poster

While reasoning over long context is crucial for various real-world applications, it remains challenging for large language models (LLMs) as they suffer from performance degradation as the context length grows. Recent work MemAgent has tried to tackle this by processing context chunk-by-chunk in an …

Cited by 0SourceScholar
2025

AdaSteer: Your Aligned LLM is Inherently an Adaptive Jailbreak Defender

EMNLP 2025

Despite extensive efforts in safety alignment, large language models (LLMs) remain vulnerable to jailbreak attacks. Activation steering offers a training-free defense method but relies on fixed steering coefficients, resulting in suboptimal protection and increased false rejections of benign inputs.

2025

AgentRecBench: Benchmarking LLM Agent-based Personalized Recommender Systems

NeurIPS 2025spotlight

The emergence of agentic recommender systems powered by Large Language Models (LLMs) represents a paradigm shift in personalized recommendations, leveraging LLMs’ advanced reasoning and role-playing capabilities to enable autonomous, adaptive decision-making. Unlike traditional recommendation approa…

Cited by 0SourcecodeScholar
2025

Beware of Your Po! Measuring and Mitigating AI Safety Risks in Role-Play Fine-Tuning of LLMs

ACL 2025long

Role-playing enables large language models (LLMs) to engage users in immersive and personalized interactions, but it also introduces significant safety risks. Existing role-play fine-tuning techniques improve role adaptability but may degrade safety performance, particularly for villainous character…

Cited by 0SourcePDFScholar
2025

Fading to Grow: Growing Preference Ratios via Preference Fading Discrete Diffusion for Recommendation

NeurIPS 2025poster

Recommenders aim to rank items from a discrete item corpus in line with user interests, yet suffer from extremely sparse user preference data. Recent advances in diffusion models have inspired diffusion-based recommenders, which alleviate sparsity by injecting noise during a forward process to preve…

Cited by 0SourceScholar
2025

Fine-Grained Verifiers: Preference Modeling as Next-token Prediction in Vision-Language Alignment

ICLR 2025poster

The recent advancements in large language models (LLMs) and pre-trained vision models have accelerated the development of vision-language large models (VLLMs), enhancing the interaction between visual and linguistic modalities. Despite their notable success across various domains, VLLMs face challen…

Cited by 6SourcePDFScholar
2025

Hello Again! LLM-powered Personalized Agent for Long-term Dialogue

NAACL 2025long

Open-domain dialogue systems have seen remarkable advancements with the development of large language models (LLMs). Nonetheless, most existing dialogue systems predominantly focus on brief single-session interactions, neglecting the real-world demands for long-term companionship and personalized in…

2025

LaMP-Val: Large Language Models Empower Personalized Valuation in Auction

EMNLP 2025

Auctions are a vital economic mechanism used to determine the market value of goods or services through competitive bidding within a specific framework. However, much of the current research primarily focuses on the bidding algorithms used within auction mechanisms. This often neglects the potential

2025

Language Representations Can be What Recommenders Need: Findings and Potentials

ICLR 2025oral

Recent studies empirically indicate that language models (LMs) encode rich world knowledge beyond mere semantics, attracting significant attention across various fields. However, in the recommendation domain, it remains uncertain whether LMs implicitly encode user preference information. Contrary to…

2025

MPO: Multilingual Safety Alignment via Reward Gap Optimization

ACL 2025long

Large language models (LLMs) have become increasingly central to AI applications worldwide, necessitating robust multilingual safety alignment to ensure secure deployment across diverse linguistic contexts. Existing preference learning methods for safety alignment, such as RLHF and DPO, are primaril…

2025

Neuron-Level Sequential Editing for Large Language Models

ACL 2025long

This work explores sequential model editing in large language models (LLMs), a critical task that involves modifying internal knowledge within LLMs continuously through multi-round editing, each incorporating updates or corrections to adjust the model’s outputs without the need for costly retraining…

2025

On Reasoning Strength Planning in Large Reasoning Models

NeurIPS 2025poster

Recent studies empirically reveal that large reasoning models (LRMs) can automatically allocate more reasoning strengths (\ie the number of reasoning tokens) for harder problems, exhibiting difficulty-awareness for better task performance. While this automatic reasoning strength allocation phenomeno…

Cited by 0SourcecodeScholar
2025

RSafe: Incentivizing proactive reasoning to build robust and adaptive LLM safeguards

NeurIPS 2025poster

Large Language Models (LLMs) continue to exhibit vulnerabilities despite deliberate safety alignment efforts, posing significant risks to users and society. To safeguard against the risk of policy-violating content, system-level moderation via external guard models—designed to monitor LLM inputs and…

Cited by 0SourcecodeScholar
2025

Safe + Safe = Unsafe? Exploring How Safe Images Can Be Exploited to Jailbreak Large Vision-Language Models

NeurIPS 2025poster

Recent advances in Large Vision-Language Models (LVLMs) have showcased strong reasoning abilities across multiple modalities, achieving significant breakthroughs in various real-world applications. Despite this great success, the safety guardrail of LVLMs may not cover the unforeseen domains introdu…

Cited by 0SourcecodeScholar
2025

Search and Refine During Think: Facilitating Knowledge Refinement for Improved Retrieval-Augmented Reasoning

NeurIPS 2025poster

Large language models have demonstrated impressive reasoning capabilities but are inherently limited by their knowledge reservoir. Retrieval-augmented reasoning mitigates this limitation by allowing LLMs to query external resources, but existing methods often retrieve irrelevant or noisy information…

Cited by 0SourceScholar
2025

The Emergence of Abstract Thought in Large Language Models Beyond Any Language

NeurIPS 2025poster

As large language models (LLMs) continue to advance, their capacity to function effectively across a diverse range of languages has shown marked improvement. Preliminary studies observe that the hidden activations of LLMs often resemble English, even when responding to non-English prompts. This has…

Cited by 0SourceScholar
2024

ALI-Agent: Assessing LLMs' Alignment with Human Values via Agent-based Evaluation

NeurIPS 2024poster

Large Language Models (LLMs) can elicit unintended and even harmful content when misaligned with human values, posing severe risks to users and society. To mitigate these risks, current evaluation benchmarks predominantly employ expert-designed contextual scenarios to assess how well LLMs align with…

2024

Customizing Language Models with Instance-wise LoRA for Sequential Recommendation

NeurIPS 2024poster

Sequential recommendation systems predict the next interaction item based on users' past interactions, aligning recommendations with individual preferences. Leveraging the strengths of Large Language Models (LLMs) in knowledge comprehension and reasoning, recent approaches are eager to apply LLMs t…

2024

On Softmax Direct Preference Optimization for Recommendation

NeurIPS 2024poster

Recommender systems aim to predict personalized rankings based on user preference data. With the rise of Language Models (LMs), LM-based recommenders have been widely explored due to their extensive world knowledge and powerful reasoning abilities. Most of the LM-based recommenders convert historica…

2024

ProtT3: Protein-to-Text Generation for Text-based Protein Understanding

ACL 2024long

Language Models (LMs) excel in understanding textual descriptions of proteins, as evident in biomedical question-answering tasks. However, their capability falters with raw protein data, such as amino acid sequences, due to a deficit in pretraining on such data. Conversely, Protein Language Models (…

2024

ReactXT: Understanding Molecular “Reaction-ship” via Reaction-Contextualized Molecule-Text Pretraining

ACL 2024findings

Molecule-text modeling, which aims to facilitate molecule-relevant tasks with a textual interface and textual knowledge, is an emerging research direction. Beyond single molecules, studying reaction-text modeling holds promise for helping the synthesis of new materials and drugs. However, previous w…

2024

Towards Neuron Attributions in Multi-Modal Large Language Models

NeurIPS 2024poster

As Large Language Models (LLMs) demonstrate impressive capabilities, demystifying their internal mechanisms becomes increasingly vital. Neuron attribution, which attributes LLM outputs to specific neurons to reveal the semantic properties they learn, has emerged as a key interpretability approach. H…

Cited by 3SourcePDFScholar
2023

Boosting Causal Discovery via Adaptive Sample Reweighting

ICLR 2023poster

Under stringent model type and variable distribution assumptions, score-based causal discovery methods learn the directed acyclic graph (DAG) from observational data by evaluating candidate graphs over an averaged score function. Despite the great success in low-dimensional linear systems, it has be…

2023

Empowering Collaborative Filtering with Principled Adversarial Contrastive Loss

NeurIPS 2023poster

Contrastive Learning (CL) has achieved impressive performance in self-supervised learning tasks, showing superior generalization ability. Inspired by the success, adopting CL into collaborative filtering (CF) is prevailing in semi-supervised topK recommendations. The basic idea is to routinely condu…

2023

Evaluating Post-hoc Explanations for Graph Neural Networks via Robustness Analysis

NeurIPS 2023oral

This work studies the evaluation of explaining graph neural networks (GNNs), which is crucial to the credibility of post-hoc explainability in practical usage. Conventional evaluation metrics, and even explanation methods -- which mainly follow the paradigm of feeding the explanatory subgraph and me…

Cited by 36SourcePDFScholar
2023

ReLM: Leveraging Language Models for Enhanced Chemical Reaction Prediction

EMNLP 2023short findings

Predicting chemical reactions, a fundamental challenge in chemistry, involves forecasting the resulting products from a given reaction process. Conventional techniques, notably those employing Graph Neural Networks (GNNs), are often limited by insufficient training data and their inability to utiliz…

Cited by 0SourcecodeScholar
2023

Rethinking Tokenizer and Decoder in Masked Graph Modeling for Molecules

NeurIPS 2023poster

Masked graph modeling excels in the self-supervised representation learning of molecular graphs. Scrutinizing previous studies, we can reveal a common scheme consisting of three key components: (1) graph tokenizer, which breaks a molecular graph into smaller fragments (\ie subgraphs) and converts th…

2022

Discovering Invariant Rationales for Graph Neural Networks

ICLR 2022poster

Intrinsic interpretability of graph neural networks (GNNs) is to find a small subset of the input graph's features --- rationale --- which guides the model prediction. Unfortunately, the leading rationalization models often rely on data biases, especially shortcut features, to compose rationales and…

2022

Incorporating Bias-aware Margins into Contrastive Loss for Collaborative Filtering

NeurIPS 2022accept

Collaborative filtering (CF) models easily suffer from popularity bias, which makes recommendation deviate from users’ actual preferences. However, most current debiasing strategies are prone to playing a trade-off game between head and tail performance, thus inevitably degrading the overall recommen…

2022

Let Invariant Rationale Discovery Inspire Graph Contrastive Learning

ICML 2022spotlight

Leading graph contrastive learning (GCL) methods perform graph augmentations in two fashions: (1) randomly corrupting the anchor graph, which could cause the loss of semantic information, or (2) using domain knowledge to maintain salient features, which undermines the generalization to other domains…

2022

SMPL: Simulated Industrial Manufacturing and Process Control Learning Environments

NeurIPS 2022accept

Traditional biological and pharmaceutical manufacturing plants are controlled by human workers or pre-defined thresholds. Modernized factories have advanced process control algorithms such as model predictive control (MPC). However, there is little exploration of applying deep reinforcement learning…

2021

Towards Multi-Grained Explainability for Graph Neural Networks

NeurIPS 2021poster

When a graph neural network (GNN) made a prediction, one raises question about explainability: “Which fraction of the input graph is most influential to the model’s decision?” Producing an answer requires understanding the model’s inner workings in general and emphasizing the insights on the decision…