← Search

Xujiang Zhao

21 accepted papers

2026

Escaping Whack-a-Mole: Code Documentation Optimization via Dependency-Guided Bi-level Search

ICML 2026poster

As large language models increasingly serve as autonomous coding agents, code documentation must be optimized for agent comprehension rather than human readability. We frame agent-oriented documentation generation as a black-box optimization problem over the documentation space, where quality is mea…

Cited by 0SourceScholar
2026

MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery

AAAI 2026technical

Uncovering causal structures from observational data is crucial for understanding complex systems and making informed decisions. While reinforcement learning (RL) has shown promise in identifying these structures in the form of a directed acyclic graph (DAG), existing methods often lack efficiency,

Cited by 0SourcePDFScholar
2026

Out-of-Distribution Detection with Positive and Negative Prompt Supervision Using Large Language Models

AAAI 2026technical

Out-of-distribution (OOD) detection is committed to delineating the classification boundaries between in-distribution (ID) and OOD images. Recent advances in vision-language models (VLMs) have demonstrated remarkable OOD detection performance by integrating both visual and textual modalities. In thi

Cited by 0SourcePDFScholar
2025

Human Texts Are Outliers: Detecting LLM-generated Texts via Out-of-distribution Detection

NeurIPS 2025poster

The rapid advancement of large language models (LLMs) such as ChatGPT, DeepSeek, and Claude has significantly increased the presence of AI-generated text in digital communication. This trend has heightened the need for reliable detection methods to distinguish between human-authored and machine-gene…

Cited by 0SourceScholar
2025

MixLLM: Dynamic Routing in Mixed Large Language Models

NAACL 2025long

Large Language Models (LLMs) exhibit potential artificial generic intelligence recently, however, their usage is costly with high response latency. Given mixed LLMs with their own strengths and weaknesses, LLM routing aims to identify the most suitable model for each query in the stream to maximize…

Cited by 4SourcePDFScholar
2025

Position Really Matters: Towards a Holistic Approach for Prompt Tuning

NAACL 2025findings

Prompt tuning is highly effective in efficiently extracting knowledge from foundation models, encompassing both language, vision, and vision-language models. However, the efficacy of employing fixed soft prompts with a predetermined position for concatenation with inputs for all instances, irrespect…

2025

SFS: Smarter Code Space Search improves LLM Inference Scaling

ICLR 2025poster

We frame code generation as a black-box optimization problem within the code space and demonstrate how optimization-inspired techniques can enhance inference scaling over text. Based on this perspective, we propose **SCATTERED FOREST SEARCH (SFS)**, a novel approach that improves solution diversity…

Cited by 0SourcePDFScholar
2025

SolverLLM: Leveraging Test-Time Scaling for Optimization Problem via LLM-Guided Search

NeurIPS 2025poster

Large Language Models (LLMs) offer promising capabilities for tackling complex reasoning tasks, including optimization problems. However, existing methods either rely on prompt engineering, which leads to poor generalization across problem types, or require costly supervised training. We introduce S…

Cited by 0SourceScholar
2024

Large Language Models Can Be Contextual Privacy Protection Learners

EMNLP 2024main

The proliferation of Large Language Models (LLMs) has driven considerable interest in fine-tuning them with domain-specific data to create specialized language models. Nevertheless, such domain-specific fine-tuning data often contains contextually sensitive personally identifiable information (PII).…

2024

Pruning as a Domain-specific LLM Extractor

NAACL 2024findings

Large Language Models (LLMs) have exhibited remarkable proficiency across a wide array of NLP tasks. However, the escalation in model size also engenders substantial deployment costs. While few efforts have explored model pruning techniques to reduce the size of LLMs, they mainly center on general o…

2024

Towards Counterfactual Fairness-aware Domain Generalization in Changing Environments

IJCAI 2024poster

Recognizing domain generalization as a commonplace challenge in machine learning, data distribution might progressively evolve across a continuum of sequential domains in practical scenarios. While current methodologies primarily concentrate on bolstering model effectiveness within these new domains…

Cited by 3SourcePDFScholar
2024

Uncertainty Quantification for In-Context Learning of Large Language Models

NAACL 2024long

In-context learning has emerged as a groundbreaking ability of Large Language Models (LLMs) and revolutionized various fields by providing a few task-relevant demonstrations in the prompt. However, trustworthy issues with LLM’s response, such as hallucination, have also been actively discussed. Exis…

2023

Multi-Label Temporal Evidential Neural Networks for Early Event Detection

ICASSP 2023accepted

Early event detection aims to detect events even before the event is complete. However, most of the existing methods focus on an event with a single label but fail to be applied to cases with multiple labels. Another non-negligible issue for early event detection is a prediction with overconfidence…

Cited by 0SourceScholar
2023

Open-ended Commonsense Reasoning with Unrestricted Answer Candidates

EMNLP 2023long findings

Open-ended Commonsense Reasoning is defined as solving a commonsense question without providing 1) a short list of answer candidates and 2) a pre-defined answer scope. Conventional ways of formulating the commonsense question into a question-answering form or utilizing external knowledge to learn re…

Cited by 0SourceScholar
2022

Seed: Sound Event Early Detection Via Evidential Uncertainty

ICASSP 2022accepted

Sound Event Early Detection (SEED) is an essential task in recognizing the acoustic environments and soundscapes. However, most of the existing methods focus on the offline sound event detection, which suffers from the over-confidence issue of early-stage event detection and usually yield unreliable…

Cited by 0SourceScholar
2021

Boosting Cross-Lingual Transfer via Self-Learning with Uncertainty Estimation

EMNLP 2021main

Recent multilingual pre-trained language models have achieved remarkable zero-shot performance, where the model is only finetuned on one source language and directly evaluated on target languages. In this work, we propose a self-learning framework that further utilizes unlabeled data of target langu…

2021

Multidimensional Uncertainty-Aware Evidential Neural Networks

AAAI 2021technical

Traditional deep neural networks (NNs) have significantly contributed to the state-of-the-art performance in the task of classification under various application domains. However, NNs have not considered inherent uncertainty in data associated with the class probabilities where misclassification un…

2021

RETRIEVE: Coreset Selection for Efficient and Robust Semi-Supervised Learning

NeurIPS 2021poster

Semi-supervised learning (SSL) algorithms have had great success in recent years in limited labeled data regimes. However, the current state-of-the-art SSL algorithms are computationally expensive and entail significant compute time and energy requirements. This can prove to be a huge limitation for…

2020

Multifaceted Uncertainty Estimation for Label-Efficient Deep Learning

NeurIPS 2020poster

We present a novel multi-source uncertainty prediction approach that enables deep learning (DL) models to be actively trained with much less labeled data. By leveraging the second-order uncertainty representation provided by subjective logic (SL), we conduct evidence-based theoretical analysis and f…

Cited by 41SourcePDFScholar
2020

Uncertainty Aware Semi-Supervised Learning on Graph Data

NeurIPS 2020spotlight

Thanks to graph neural networks (GNNs), semi-supervised node classification has shown the state-of-the-art performance in graph data. However, GNNs have not considered different types of uncertainties associated with class probabilities to minimize risk of increasing misclassification under uncerta…