← Search

Yikai Zhang

29 accepted papers

2025

ARIA: Training Language Agents with Intention-driven Reward Aggregation

NeurIPS 2025spotlight

Large language models (LLMs) have enabled agents to perform complex reasoning and decision-making through free-form language interactions. However, in open-ended language action environments (e.g., negotiation or question-asking games), the action space can be formulated as a joint distribution over…

Cited by 0SourceScholar
2025

DEEPER Insight into Your User: Directed Persona Refinement for Dynamic Persona Modeling

ACL 2025long

To advance personalized applications such as recommendation systems and user behavior prediction, recent research increasingly adopts large language models (LLMs) for human-readable persona modeling. In dynamic real-world scenarios, effective persona modeling necessitates leveraging streaming behavi…

2025

Deep Transfer Regression for EEG-based Driving Fatigue Detection

ICASSP 2025accepted

Recently, Electroencephalography (EEG) has been increasingly utilized in driving fatigue detection tasks. However, the inter-subject variabilities in EEG data render models trained on one subject ineffective for being directly applied to others. Transfer learning has been widely used to address this…

Cited by 0SourceScholar
2025

MCiteBench: A Multimodal Benchmark for Generating Text with Citations

EMNLP 2025

Multimodal Large Language Models (MLLMs) have advanced in integrating diverse modalities but frequently suffer from hallucination. A promising solution to mitigate this issue is to generate text with citations, providing a transparent chain for verification. However, existing work primarily focuses

2025

Multi-agent KTO: Enhancing Strategic Interactions of Large Language Model in Language Game

NeurIPS 2025poster

Achieving Artificial General Intelligence (AGI) requires AI agents that can not only make strategic decisions but also engage in flexible and meaningful communication. Inspired by Wittgenstein's language game theory, we propose that language agents can learn through in-context interaction rather tha…

Cited by 0SourcecodeScholar
2025

R-KV: Redundancy-aware KV Cache Compression for Reasoning Models

NeurIPS 2025poster

Reasoning models have demonstrated impressive performance in self-reflection and chain-of-thought reasoning. However, they often produce excessively long outputs, leading to prohibitively large key-value (KV) caches during inference. While chain-of-thought inference significantly improves performanc…

Cited by 0SourceScholar
2025

Revealing the Barriers of Language Agents in Planning

NAACL 2025long

Autonomous planning has been an ongoing pursuit since the inception of artificial intelligence. Based on curated problem solvers, early planning agents could deliver precise solutions for specific tasks but lacked generalization. The emergence of large language models (LLMs) and their powerful reaso…

2025

SELFGOAL: Your Language Agents Already Know How to Achieve High-level Goals

NAACL 2025long

Language agents powered by large language models (LLMs) are increasingly valuable as decision-making tools in domains such as gaming and programming. However, these agents often face challenges in achieving high-level goals without detailed instructions and in adapting to environments where feedback…

Cited by 9SourcePDFScholar
2025

The Lighthouse of Language: Enhancing LLM Agents via Critique-Guided Improvement

NeurIPS 2025poster

Large language models (LLMs) have recently transformed from text-based assistants to autonomous agents capable of planning, reasoning, and iteratively improving their actions. While numerical reward signals and verifiers can effectively rank candidate actions, they often provide limited contextual g…

Cited by 0SourceScholar
2024

DetectBench: Can Large Language Model Detect and Piece Together Implicit Evidence?

EMNLP 2024finding

Detecting evidence within the context is a key step in the process of reasoning task. Evaluating and enhancing the capabilities of LLMs in evidence detection will strengthen context-based reasoning performance. This paper proposes a benchmark called DetectBench for verifying the ability to detect an…

2024

Finite Smoothing Algorithm for High-Dimensional Support Vector Machines and Quantile Regression

ICML 2024poster

This paper introduces a finite smoothing algorithm (FSA), a novel approach to tackle computational challenges in applying support vector machines (SVM) and quantile regression to high-dimensional data. The critical issue with these methods is the non-smooth nature of their loss functions, which trad…

Cited by 0SourcePDFScholar
2024

Light Up the Shadows: Enhance Long-Tailed Entity Grounding with Concept-Guided Vision-Language Models

ACL 2024findings

Multi-Modal Knowledge Graphs (MMKGs) have proven valuable for various downstream tasks. However, scaling them up is challenging because building large-scale MMKGs often introduces mismatched images (i.e., noise). Most entities in KGs belong to the long tail, meaning there are few images of them avai…

2024

OlympicArena: Benchmarking Multi-discipline Cognitive Reasoning for Superintelligent AI

NeurIPS 2024poster

The evolution of Artificial Intelligence (AI) has been significantly accelerated by advancements in Large Language Models (LLMs) and Large Multimodal Models (LMMs), gradually showcasing potential cognitive reasoning abilities in problem-solving and scientific discovery (i.e., AI4Science) once exclus…

2024

TimeArena: Shaping Efficient Multitasking Language Agents in a Time-Aware Simulation

ACL 2024long

Despite remarkable advancements in emulating human-like behavior through Large Language Models (LLMs), current textual simulations do not adequately address the notion of time. To this end, we introduce TimeArena, a novel textual simulated environment that incorporates complex temporal dynamics and…

Cited by 13SourcePDFScholar
2023

C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Models

NeurIPS 2023poster

New NLP benchmarks are urgently needed to align with the rapid development of large language models (LLMs). We present C-Eval, the first comprehensive Chinese evaluation suite designed to assess advanced knowledge and reasoning abilities of foundation models in a Chinese context. C-Eval comprises mu…

2023

HAUSER: Towards Holistic and Automatic Evaluation of Simile Generation

ACL 2023long

Similes play an imperative role in creative writing such as story and dialogue generation. Proper evaluation metrics are like a beacon guiding the research of simile generation (SG). However, it remains under-explored as to what criteria should be considered, how to quantify each criterion into metr…

2023

Learning to Segment from Noisy Annotations: A Spatial Correction Approach

ICLR 2023poster

Noisy labels can significantly affect the performance of deep neural networks (DNNs). In medical image segmentation tasks, annotations are error-prone due to the high demand in annotation time and in the annotators' expertise. Existing methods mostly tackle label noise in classification tasks. Their…

2023

Provably Convergent Schrödinger Bridge with Applications to Probabilistic Time Series Imputation

ICML 2023poster

The Schrödinger bridge problem (SBP) is gaining increasing attention in generative modeling and showing promising potential even in comparison with the score-based generative models (SGMs). SBP can be interpreted as an entropy-regularized optimal transport problem, which conducts projections onto ev…

2023

Risk Bounds on Aleatoric Uncertainty Recovery

AISTATS 2023poster

Quantifying aleatoric uncertainty is a challenging task in machine learning. It is important for decision making associated with data-dependent uncertainty in model outcomes. Recently, many empirical studies in modeling aleatoric uncertainty under regression settings primarily rely on either a Gauss…

2023

Topology-Aware Uncertainty for Image Segmentation

NeurIPS 2023poster

Segmentation of curvilinear structures such as vasculature and road networks is challenging due to relatively weak signals and complex geometry/topology. To facilitate and accelerate large scale annotation, one has to adopt semi-automatic approaches such as proofreading by experts. In this work, we…

2022

A Manifold View of Adversarial Risk

AISTATS 2022poster

The adversarial risk of a machine learning model has been widely studied. Most previous works assume that the data lies in the whole ambient space. We propose to take a new angle and take the manifold assumption into consideration. Assuming data lies in a manifold, we investigate two new types of ad…

Cited by 4SourcePDFScholar
2022

Stability of SGD: Tightness analysis and improved bounds

UAI 2022poster

Stochastic Gradient Descent (SGD) based methods have been widely used for training large-scale machine learning models that also generalize well in practice. Several explanations have been offered for this generalization performance, a prominent one being algorithmic stability Hardt et al [2016]. Ho…

Cited by 41SourcePDFScholar
2021

Learning with Feature-Dependent Label Noise: A Progressive Approach

ICLR 2021spotlight

Label noise is frequently observed in real-world large-scale datasets. The noise is introduced due to a variety of reasons; it is heterogeneous and feature-dependent. Most existing approaches to handling noisy labels fall into two categories: they either assume an ideal feature-independent noise, or…

2021

Topological Detection of Trojaned Neural Networks

NeurIPS 2021poster

Deep neural networks are known to have security issues. One particular threat is the Trojan attack. It occurs when the attackers stealthily manipulate the model's behavior through Trojaned training samples, which can later be exploited. Guided by basic neuroscientific principles, we discover subtle…

Cited by 58SourcePDFScholar
2020

Learn distributed GAN with Temporary Discriminators

ECCV 2020poster

In this work, we propose a method for training distributed GAN with sequential temporary discriminators. Our proposed method tackles the challenge of training GAN in the federated learning manner: How to update the generator with a flow of temporary discriminators? We apply our proposed method to le…

2020

Synthetic Learning: Learn From Distributed Asynchronized Discriminator GAN Without Sharing Medical Image Data

CVPR 2020poster

In this paper, we propose a data privacy-preserving and communication efficient distributed GAN learning framework named Distributed Asynchronized Discriminator GAN (AsynDGAN). Our proposed framework aims to train a central generator learns from distributed discriminator, and use the generated synth…

Cited by 113PDFcodeScholar
2018

Robust Vertex Enumeration for Convex Hulls in High Dimensions

AISTATS 2018poster

We design a fast and robust algorithm named {All Vertex Traingle Algorithm (AVTA)} for detecting the vertices of the convex hull of a set of points in high dimensions. Our proposed algorithm is very general and works for arbitrary convex hulls. In addition to being a fundamental problem in computati…