← Search

Kaiyuan Zhang

22 accepted papers

2026

FinSearchComp: Towards a Realistic, Expert-Level Evaluation of Financial Search and Reasoning

ICLR 2026poster

Search has emerged as core infrastructure for LLM-based agents and is widely viewed as critical on the path toward more general intelligence. Finance is a particularly demanding proving ground: analysts routinely conduct complex, multi-step searches over time-sensitive, domain-specific data, making…

Cited by 0SourcecodeScholar
2026

FutureX: An Advanced Live Benchmark for LLM Agents in Future Prediction

ICLR 2026poster

Future prediction is a complex task for LLM agents, requiring a high level of analytical thinking, information gathering, contextual understanding, and decision-making under uncertainty. Agents must not only gather and interpret vast amounts of dynamic information but also integrate diverse data sou…

Cited by 0SourceScholar
2026

MIND THE SHIFT: USING DELTA SSL EMBEDDINGS TO ENHANCE CHILD ASR

ICASSP 2026poster

Self-supervised learning (SSL) models have achieved impressive results across many speech tasks, yet child automatic speech recognition (ASR) remains challenging due to limited data and pretraining domain mismatch. Fine-tuning SSL models on child speech induces shifts in the representation space. We…

Cited by 0SourcePDFScholar
2026

Membership Inference Attacks Against Fine-tuned Diffusion Language Models

ICLR 2026poster

Diffusion Language Models (DLMs) represent a promising alternative to autoregressive language models, using bidirectional masked token prediction. Yet their susceptibility to privacy leakage via Membership Inference Attacks (MIA) remains critically underexplored. This paper presents the first system…

Cited by 0SourceScholar
2026

STACODEC: SEMANTIC TOKEN ASSIGNMENT FOR BALANCING ACOUSTIC FIDELITY AND SEMANTIC INFORMATION IN AUDIO CODECS

ICASSP 2026poster

Neural audio codecs are widely used for audio compression and can be integrated into token-based language models. Traditional codecs preserve acoustic details well but lack semantic information. Recent hybrid codecs attempt to incorporate semantic information through distillation, but this often deg…

Cited by 0SourcePDFScholar
2025

Equipping Retrieval-Augmented Large Language Models with Document Structure Awareness

EMNLP 2025

While large language models (LLMs) demonstrate impressive capabilities, their reliance on parametric knowledge often leads to factual inaccuracies. Retrieval-Augmented Generation (RAG) mitigates this by leveraging external documents, yet existing approaches treat retrieved passages as isolated chunk

2025

MADAWSD: Multi-Agent Debate Framework for Adversarial Word Sense Disambiguation

EMNLP 2025

Word sense disambiguation (WSD) is a fundamental yet challenging task in natural language processing. In recent years, the advent of large language models (LLMs) has led to significant advancements in regular WSD tasks. However, most existing LLMs face two major issues that hinder their performance

2025

MARS-Bench: A Multi-turn Athletic Real-world Scenario Benchmark for Dialogue Evaluation

EMNLP 2025

Large Language Models ( LLMs ), e.g. ChatGPT, have been widely adopted in real-world dialogue applications. However, LLMs’ robustness, especially in handling long complex dialogue sessions, including frequent motivation transfer, sophisticated cross-turn dependency, is criticized all along. Neverthe

2025

ProSec: Fortifying Code LLMs with Proactive Security Alignment

ICML 2025poster

While recent code-specific large language models (LLMs) have greatly enhanced their code generation capabilities, the safety of these models remains under-explored, posing potential risks as insecure code generated by these models may introduce vulnerabilities into real-world systems. Existing metho…

2025

Profiler: Black-box AI-generated Text Origin Detection via Context-aware Inference Pattern Analysis

EMNLP 2025

With the increasing capabilities of Large Language Models (LLMs), the proliferation of AI-generated texts has become a serious concern. Given the diverse range of organizations providing LLMs, it is crucial for governments and third-party entities to identify the origin LLM of a given AI-generated t

Cited by 0SourcePDFScholar
2025

RoDEval: A Robust Word Sense Disambiguation Evaluation Framework for Large Language Models

EMNLP 2025

Accurately evaluating the word sense disambiguation (WSD) capabilities of large language models (LLMs) remains challenging, as existing studies primarily rely on single-task evaluations and classification-based metrics that overlook the fundamental differences between generative LLMs and traditional

2025

System Prompt Hijacking via Permutation Triggers in LLM Supply Chains

ACL 2025finding

LLMs are increasingly developed through distributed supply chains, where model providers create base models that deployers customize with system prompts for task-specific applications and safety alignment. We introduce SHIP, a novel post-deployment attack that bypasses system prompts, enabling unres…

2025

TAI3: Testing Agent Integrity in Interpreting User Intent

NeurIPS 2025poster

LLM agents are increasingly deployed to automate real-world tasks by invoking APIs through natural language instructions. While powerful, they often suffer from misinterpretation of user intent, leading to the agent’s actions that diverge from the user’s intended goal, especially as external toolkit…

Cited by 0SourceScholar
2024

BiScope: AI-generated Text Detection by Checking Memorization of Preceding Tokens

NeurIPS 2024poster

Detecting text generated by Large Language Models (LLMs) is a pressing need in order to identify and prevent misuse of these powerful models in a wide range of applications, which have highly undesirable consequences such as misinformation and academic dishonesty. Given a piece of subject text, m…

2024

Elijah: Eliminating Backdoors Injected in Diffusion Models via Distribution Shift

AAAI 2024technical

Diffusion models (DM) have become state-of-the-art generative models because of their capability of generating high-quality images from noises without adversarial training. However, they are vulnerable to backdoor attacks as reported by recent studies. When a data input (e.g., some Gaussian noise)…

2024

LOTUS: Evasive and Resilient Backdoor Attacks through Sub-Partitioning

CVPR 2024poster

Backdoor attack poses a significant security threat to Deep Learning applications. Existing attacks are often not evasive to established backdoor detection techniques. This susceptibility primarily stems from the fact that these attacks typically leverage a universal trigger pattern or transformatio…

2024

Source Code Foundation Models are Transferable Binary Analysis Knowledge Bases

NeurIPS 2024poster

Human-Oriented Binary Reverse Engineering (HOBRE) lies at the intersection of binary and source code, aiming to lift binary code to human-readable content relevant to source code, thereby bridging the binary-source semantic gap. Recent advancements in uni-modal code model pre-training, particularly…

2024

UNIT: Backdoor Mitigation via Automated Neural Distribution Tightening

ECCV 2024poster

"Deep neural networks (DNNs) have demonstrated effectiveness in various fields. However, DNNs are vulnerable to backdoor attacks, which inject a unique pattern, called trigger, into the input to cause misclassification to an attack-chosen target label. While existing works have proposed various meth…

2023

Detecting Backdoors in Pre-Trained Encoders

CVPR 2023poster

Self-supervised learning in computer vision trains on unlabeled data, such as images or (image, text) pairs, to obtain an image encoder that learns high-quality embeddings for input data. Emerging backdoor attacks towards encoders expose crucial vulnerabilities of self-supervised learning, since dow…

2023

Django: Detecting Trojans in Object Detection Models via Gaussian Focus Calibration

NeurIPS 2023poster

Object detection models are vulnerable to backdoor or trojan attacks, where an attacker can inject malicious triggers into the model, leading to altered behavior during inference. As a defense mechanism, trigger inversion leverages optimization to reverse-engineer triggers and identify compromised m…

2023

FLIP: A Provable Defense Framework for Backdoor Mitigation in Federated Learning

ICLR 2023poster

Federated Learning (FL) is a distributed learning paradigm that enables different parties to train a model together for high quality and strong privacy protection. In this scenario, individual participants may get compromised and perform backdoor attacks by poisoning the data (or gradients). Existin…

2023

ParaFuzz: An Interpretability-Driven Technique for Detecting Poisoned Samples in NLP

NeurIPS 2023poster

Backdoor attacks have emerged as a prominent threat to natural language processing (NLP) models, where the presence of specific triggers in the input can lead poisoned models to misclassify these inputs to predetermined target classes. Current detection mechanisms are limited by their inability to a…

Cited by 21SourcePDFScholar