← Search

Kexin Huang

36 accepted papers

2026

Clipping Bottleneck: Stabilizing RLVR via Stochastic Recovery of Near-Boundary Signals

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a central paradigm for scaling LLM reasoning, yet its optimization often suffers from training instability and suboptimal convergence. Through a systematic dissection of the GRPO-based objective, we reveal that the rigid clipping d…

Cited by 0SourceScholar
2026

Experience Augmented Policy Optimization for LLM Reasoning

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is a powerful paradigm for improving the reasoning capabilities of large language models (LLMs). However, existing RLVR methods typically rely on on-policy optimization from scratch, resulting in high sampling costs and inefficient utilization of…

Cited by 0SourceScholar
2026

Mitigating Reward Hacking in LLM-based Recommendation: A Preference Optimization Approach

ICML 2026poster

Post-training adaptation has become the central paradigm for leveraging large language models (LLMs) in recommendation. While recent preference optimization methods, such as Direct Preference Optimization (DPO), enhance pairwise preference discrimination, they remain vulnerable to \emph{reward hacki…

Cited by 0SourceScholar
2026

On the Direction of RLVR Updates for LLM Reasoning: Identification and Exploitation

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has substantially improved the reasoning capabilities of large language models. While existing analyses identify that RLVR-induced changes are sparse, they primarily focus on the **magnitude** of these updates, largely overlooking their **direct…

Cited by 0SourcecodeScholar
2026

One-Way Policy Optimization for Self-Evolving LLMs

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has become a promising paradigm for scaling reasoning capabilities of Large Language Models (LLMs). However, the sparsity of binary verifier rewards often leads to low efficiency and optimization instability. To stabilize training, existing metho…

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

RoboOmni: Proactive Robot Manipulation in Omni-modal Context

ICLR 2026poster

Recent advances in Multimodal Large Language Models (MLLMs) have driven rapid progress in Vision–Language–Action (VLA) models for robotic manipulation. Although effective in many scenarios, current approaches largely rely on explicit instructions, whereas in real-world interactions, humans rarely is…

Cited by 0SourcecodeScholar
2026

Scaling Transformers for End-to-End Discrete Audio Tokenization

ICML 2026poster

Discrete audio tokenizers are fundamental to empowering large language models with native audio processing and generation capabilities. Despite recent progress, existing approaches often rely on pretrained encoders, semantic distillation, or heterogeneous CNN-based architectures. These designs intro…

Cited by 0SourceScholar
2026

Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has significantly improved reasoning in large language models (LLMs), yet the token-level mechanisms through which they reshape model behavior remain unclear. We present a systematic empirical study of RLVR’s distributional effects across three c…

Cited by 0SourceScholar
2025

Automated Hypothesis Validation with Agentic Sequential Falsifications

ICML 2025poster

Hypotheses are central to information acquisition, decision-making, and discovery. However, many real-world hypotheses are abstract, high-level statements that are difficult to validate directly. This challenge is further intensified by the rise of hypothesis generation from Large Language Models (L…

2025

BioDiscoveryAgent: An AI Agent for Designing Genetic Perturbation Experiments

ICLR 2025poster

Agents based on large language models have shown great potential in accelerating scientific discovery by leveraging their rich background knowledge and reasoning capabilities. In this paper, we introduce BioDiscoveryAgent, an agent that designs new experiments, reasons about their outcomes, and effi…

2025

From Evasion to Concealment: Stealthy Knowledge Unlearning for LLMs

ACL 2025finding

LLM Unlearning plays a crucial role in removing sensitive information from language models to mitigate potential misuse. However, previous approaches often treat nonsensical responses or template-based refusals (e.g., “Sorry, I cannot answer.”) as the unlearning target, which can give the impression…

Cited by 0SourcePDFScholar
2025

Invisible Entropy: Towards Safe and Efficient Low-Entropy LLM Watermarking

EMNLP 2025

Logit-based LLM watermarking traces and verifies AI-generated content by maintaining green and red token lists and increasing the likelihood of green tokens during generation. However, it struggles in low-entropy scenarios, where predictable outputs make green token selection difficult without disru

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

Larger or Smaller Reward Margins to Select Preferences for LLM Alignment?

ICML 2025poster

Preference learning is critical for aligning large language models (LLMs) with human values, with the quality of preference datasets playing a crucial role in this process. While existing metrics primarily assess data quality based on either *explicit* or *implicit* reward margins, their single-mar…

Cited by 0SourcePDFScholar
2025

Learning Bayesian Nash Equilibrium in Auction Games via Approximate Best Response

ICML 2025poster

Auction plays a crucial role in many modern trading environments, including online advertising and public resource allocation. As the number of competing bidders increases, learning Bayesian Nash Equilibrium (BNE) in auctions faces significant scalability challenges. Existing methods often experienc…

Cited by 0SourcePDFScholar
2025

RePO: Understanding Preference Learning Through ReLU-Based Optimization

NeurIPS 2025poster

Preference learning has become a common approach in various recent methods for aligning large language models with human values. These methods optimize the preference margin between chosen and rejected responses, subject to certain constraints for avoiding over-optimization. In this paper, we report…

Cited by 0SourceScholar
2025

Toward Generalizing Visual Brain Decoding to Unseen Subjects

ICLR 2025poster

Visual brain decoding aims to decode visual information from human brain activities. Despite the great progress, one critical limitation of current brain decoding research lies in the lack of generalization capability to unseen subjects. Prior work typically focuses on decoding brain activity of ind…

2024

Auctionformer: A Unified Deep Learning Algorithm for Solving Equilibrium Strategies in Auction Games

ICML 2024poster

Auction games have been widely used in plenty of trading environments such as online advertising and real estate. The complexity of real-world scenarios, characterized by diverse auction mechanisms and bidder asymmetries, poses significant challenges in efficiently solving for equilibria. Traditiona…

Cited by 0SourcePDFScholar
2024

AvaTaR: Optimizing LLM Agents for Tool Usage via Contrastive Reasoning

NeurIPS 2024poster

Large language model (LLM) agents have demonstrated impressive capabilities in utilizing external tools and knowledge to boost accuracy and reduce hallucinations. However, developing prompting techniques that enable LLM agents to effectively use these tools and knowledge remains a heuristic and labo…

2024

ESC-Eval: Evaluating Emotion Support Conversations in Large Language Models

EMNLP 2024main

Emotion Support Conversation (ESC) is a crucial application, which aims to reduce human stress, offer emotional guidance, and ultimately enhance human mental and physical well-being. With the advancement of Large Language Models (LLMs), many researchers have employed LLMs as the ESC models. However,…

2024

Fake Alignment: Are LLMs Really Aligned Well?

NAACL 2024long

The growing awareness of safety concerns in large language models (LLMs) has sparked considerable interest in the evaluation of safety. This study investigates an under-explored issue about the evaluation of LLMs, namely the substantial discrepancy in performance between multiple-choice questions an…

2024

Flames: Benchmarking Value Alignment of LLMs in Chinese

NAACL 2024long

The widespread adoption of large language models (LLMs) across various regions underscores the urgent need to evaluate their alignment with human values. Current benchmarks, however, fall short of effectively uncovering safety vulnerabilities in LLMs. Despite numerous models achieving high scores an…

2024

MLLMGuard: A Multi-dimensional Safety Evaluation Suite for Multimodal Large Language Models

NeurIPS 2024poster

Powered by remarkable advancements in Large Language Models (LLMs), Multimodal Large Language Models (MLLMs) demonstrate impressive capabilities in manifold tasks. However, the practical application scenarios of MLLMs are intricate, exposing them to potential malicious instructions and thereby posin…

2024

On the Analysis of GAN-based Image-to-Image Translation with Gaussian Noise Injection

ICLR 2024poster

Image-to-image (I2I) translation is vital in computer vision tasks like style transfer and domain adaptation. While recent advances in GAN have enabled high-quality sample generation, real-world challenges such as noise and distortion remain significant obstacles. Although Gaussian noise injection d…

Cited by 2SourcePDFScholar
2024

Position: Relational Deep Learning - Graph Representation Learning on Relational Databases

ICML 2024poster

Much of the world's most valued data is stored in relational databases and data warehouses, where the data is organized into tables connected by primary-foreign key relations. However, building machine learning models using this data is both challenging and time consuming because no ML algorithm can…

Cited by 12SourcePDFScholar
2024

RelBench: A Benchmark for Deep Learning on Relational Databases

NeurIPS 2024poster

We present RelBench, a public benchmark for solving predictive tasks in relational databases with deep learning. RelBench provides databases and tasks spanning diverse domains, scales, and database dimensions, and is intended to be a foundational infrastructure for future research in this direction…

Cited by 11SourcePDFScholar
2024

STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Bases

NeurIPS 2024poster

Answering real-world complex queries, such as complex product search, often requires accurate retrieval from semi-structured knowledge bases that involve blend of unstructured (e.g., textual descriptions of products) and structured (e.g., entity relations of products) information. However, many prev…

2023

High dimensional, tabular deep learning with an auxiliary knowledge graph

NeurIPS 2023poster

Machine learning models exhibit strong performance on datasets with abundant labeled samples. However, for tabular datasets with extremely high $d$-dimensional features but limited $n$ samples (i.e. $d \gg n$), machine learning models struggle to achieve strong performance due to the risk of overfit…

2023

Uncertainty Quantification over Graph with Conformalized Graph Neural Networks

NeurIPS 2023spotlight

Graph Neural Networks (GNNs) are powerful machine learning prediction models on graph-structured data. However, GNNs lack rigorous uncertainty estimates, limiting their reliable deployment in settings where the cost of errors is significant. We propose conformalized GNN (CF-GNN), extending conformal…

2022

Graphein - a Python Library for Geometric Deep Learning and Network Analysis on Biomolecular Structures and Interaction Networks

NeurIPS 2022accept

Geometric deep learning has broad applications in biology, a domain where relational structure in data is often intrinsic to modelling the underlying phenomena. Currently, efforts in both geometric deep learning and, more broadly, deep learning applied to biomolecular tasks have been hampered by a…

Cited by 32SourcePDFScholar
2021

Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Development

NeurIPS 2021poster

Therapeutics machine learning is an emerging field with incredible opportunities for innovation and impact. However, advancement in this field requires the formulation of meaningful tasks and careful curation of datasets. Here, we introduce Therapeutics Data Commons (TDC), the first unifying platfor…

Cited by 354SourcecodeScholar