← Search

Jiayi Yuan

19 accepted papers

2026

Deterministic Inference across Tensor Parallel Sizes That Eliminates Training-Inference Mismatch

ICML 2026poster

Deterministic inference is increasingly critical for large language model (LLM) applications such as LLM-as-a-judge evaluation, multi-agent systems, and Reinforcement Learning (RL). However, existing LLM serving frameworks exhibit non-deterministic behavior: identical inputs can yield different outp…

Cited by 0SourceScholar
2026

FAFO: Lossy KV Cache Compression for Lossless Inference Acceleration via Draftless Fumble Decoding

ICML 2026poster

Lossy KV cache compression is a well-explored subfield of machine learning efficiency, with improved latency being one of its major gains. However, lossy compression techniques can fumble from time to time, exhibiting various — and often catastrophic — failure patterns that are not only difficult to…

Cited by 0SourceScholar
2026

RouterArena: An Open Platform for Comprehensive Comparison of LLM Routers

ICLR 2026poster

Today's LLM ecosystem comprises a wide spectrum of models that differ in size, capability, and cost. No single model is optimal for all scenarios; hence, LLM routers have become essential for selecting the most appropriate model under varying circumstances. However, the rapid emergence of various ro…

Cited by 0SourceScholar
2025

DHP Benchmark: Are LLMs Good NLG Evaluators?

NAACL 2025findings

Large Language Models (LLMs) are increasingly serving as evaluators in Natural Language Generation (NLG) tasks; this is often referred to as “LLM-as-a-judge” paradigm. However, the capabilities of LLMs in evaluating NLG quality remain underexplored. Current studies depend on human assessments and si…

Cited by 6SourcePDFScholar
2025

InvestESG: A multi-agent reinforcement learning benchmark for studying climate investment as a social dilemma

ICLR 2025poster

**InvestESG** is a novel multi-agent reinforcement learning (MARL) benchmark designed to study the impact of Environmental, Social, and Governance (ESG) disclosure mandates on corporate climate investments. The benchmark models an intertemporal social dilemma where companies balance short-term profi…

2025

LoRATK: LoRA Once, Backdoor Everywhere in the Share-and-Play Ecosystem

EMNLP 2025

Backdoor attacks are powerful and effective, but distributing LLMs without a proven track record like ‘meta-llama‘ or ‘qwen‘ rarely gains community traction. We identify LoRA sharing as a unique scenario where users are more willing to try unendorsed assets, since such shared LoRAs allow them to enj

2025

ReasonerRank: Redefining Language Model Evaluation with Ground-Truth-Free Ranking Frameworks

ACL 2025finding

Large Language Models (LLMs) are increasingly adopted across real-world applications, yet traditional evaluations rely on expensive, domain-specific ground-truth labels that are often unavailable or infeasible. We introduce a ground-truth-free evaluation framework focused on reasoning consistency an…

Cited by 0SourcePDFScholar
2025

Understanding and Mitigating Numerical Sources of Nondeterminism in LLM Inference

NeurIPS 2025oral

Large Language Models (LLMs) are now integral across various domains and have demonstrated impressive performance. Progress, however, rests on the premise that benchmark scores are both accurate and reproducible. We demonstrate that the reproducibility of LLM performance is fragile: changing system…

Cited by 0SourcecodeScholar
2024

GNNs Also Deserve Editing, and They Need It More Than Once

ICML 2024poster

Suppose a self-driving car is crashing into pedestrians, or a chatbot is instructing its users to conduct criminal wrongdoing; the stakeholders of such products will undoubtedly want to patch these catastrophic errors as soon as possible. To address such concerns, *Model Editing:* the study of effic…

2024

KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache

ICML 2024poster

Efficiently serving large language models (LLMs) requires batching many requests together to reduce the cost per request. Yet, the key-value (KV) cache, which stores attention keys and values to avoid re-computations, significantly increases memory demands and becomes the new bottleneck in speed and…

2024

KV Cache Compression, But What Must We Give in Return? A Comprehensive Benchmark of Long Context Capable Approaches

EMNLP 2024finding

Long context capability is a crucial competency for large language models (LLMs) as it mitigates the human struggle to digest long-form texts. This capability enables complex task-solving scenarios such as book summarization, code assistance, and many more tasks that are traditionally manpower-inten…

2024

Taylor Unswift: Secured Weight Release for Large Language Models via Taylor Expansion

EMNLP 2024main

Ensuring the security of released large language models (LLMs) poses a significant dilemma, as existing mechanisms either compromise ownership rights or raise data privacy concerns. To address this dilemma, we introduce TaylorMLP to protect the ownership of released LLMs and prevent their abuse. Spe…

2023

ERSAM: Neural Architecture Search for Energy-Efficient and Real-Time Social Ambiance Measurement

ICASSP 2023accepted

Social ambiance describes the context in which social interactions happen, and can be measured using speech audio by counting the number of concurrent speakers. This measurement has enabled various mental health tracking and human-centric IoT applications. While on-device Socal Ambiance Measure (SAM…

Cited by 0SourceScholar
2023

Recurrent Structure Attention Guidance for Depth Super-resolution

AAAI 2023technical

Image guidance is an effective strategy for depth super-resolution. Generally, most existing methods employ hand-crafted operators to decompose the high-frequency (HF) and low-frequency (LF) ingredients from low-resolution depth maps and guide the HF ingredients by directly concatenating them with i…

2023

Setting the Trap: Capturing and Defeating Backdoors in Pretrained Language Models through Honeypots

NeurIPS 2023poster

In the field of natural language processing, the prevalent approach involves fine-tuning pretrained language models (PLMs) using local samples. Recent research has exposed the susceptibility of PLMs to backdoor attacks, wherein the adversaries can embed malicious prediction behaviors by manipulating…

Cited by 18SourcePDFScholar
2023

Structure Flow-Guided Network for Real Depth Super-resolution

AAAI 2023technical

Real depth super-resolution (DSR), unlike synthetic settings, is a challenging task due to the structural distortion and the edge noise caused by the natural degradation in real-world low-resolution (LR) depth maps. These defeats result in significant structure inconsistency between the depth map an…

2022

DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks

ICML 2022spotlight

Efficient deep neural network (DNN) models equipped with compact operators (e.g., depthwise convolutions) have shown great potential in reducing DNNs’ theoretical complexity (e.g., the total number of weights/operations) while maintaining a decent model accuracy. However, existing efficient DNNs are…