← Search

Yao Fu

33 accepted papers

2026

Ego2Web: A Web Agent Benchmark Grounded in Egocentric Videos

CVPR 2026

Multimodal AI agents are increasingly automating complex real-world workflows that involve online web execution. However, current web-agent benchmarks suffer from a critical limitation: they focus entirely on web-based interaction and perception, lacking grounding in the user's real-world physical s

Cited by 0SourcecodeScholar
2026

Hierarchical Procedural Meta-Reasoning for Generalizable Multimodal Agents

ICML 2026poster

While multimodal agents can achieve strong performance through fine-tuning, their ability to generalize remains limited in complex real-world tasks such as mobile navigation, where diverse applications, frequent system changes, and customized workflows are common in practice. We argue that a fundame…

Cited by 0SourceScholar
2025

Digest the Knowledge: Large Language Models empowered Message Passing for Knowledge Graph Question Answering

ACL 2025long

Despite their success, large language models (LLMs) suffer from notorious hallucination issue. By introducing external knowledge stored in knowledge graphs (KGs), existing methods use paths as the medium to represent the graph information that send into LLMs. However, paths only contain limited grap…

2025

DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads

ICLR 2025poster

Deploying long-context large language models (LLMs) is essential but poses significant computational and memory challenges. Caching all Key and Value (KV) states across all attention heads consumes substantial memory. Existing KV cache pruning methods either damage the long-context capabilities of L…

2025

FAEDKV: Infinite-Window Fourier Transform for Unbiased KV Cache Compression

EMNLP 2025

The efficacy of Large Language Models (LLMs) in long-context tasks is often hampered by the substantial memory footprint and computational demands of the Key-Value (KV) cache. Current compression strategies, including token eviction and learned projections, frequently lead to biased representations—

Cited by 0SourcePDFScholar
2025

Interactive and Expressive Code-Augmented Planning with Large Language Models

ACL 2025long

Large Language Models (LLMs) demonstrate strong abilities in common-sense reasoning and interactive decision-making, but often struggle with complex, long-horizon planning tasks. Recent techniques have sought to structure LLM outputs using control flow and code to improve planning performance. Howev…

Cited by 0SourcePDFScholar
2025

MoE-CAP: Benchmarking Cost, Accuracy and Performance of Sparse Mixture-of-Experts Systems

NeurIPS 2025poster

The sparse Mixture-of-Experts (MoE) architecture is increasingly favored for scaling Large Language Models (LLMs) efficiently, but it depends on heterogeneous compute and memory resources. These factors jointly affect system Cost, Accuracy, and Performance (CAP), making trade-offs inevitable. Existi…

Cited by 0SourcecodeScholar
2025

Pruning Weights but Not Truth: Safeguarding Truthfulness While Pruning LLMs

EMNLP 2025

Neural network pruning has emerged as a promising approach for deploying LLMs in low-resource scenarios while preserving downstream task performance. However, for the first time, we reveal that such pruning disrupts LLMs’ internal activation features crucial for lie detection, where probing classifi

Cited by 0SourcePDFScholar
2025

Quantized but Deceptive? A Multi-Dimensional Truthfulness Evaluation of Quantized LLMs

EMNLP 2025

Quantization enables efficient deployment of large language models (LLMs) in resource-constrained environments by significantly reducing memory and computation costs. While quantized LLMs often maintain performance on perplexity and zero-shot tasks, their impact on truthfulness—whether generating tr

2025

RAM-W600: A Multi-Task Wrist Dataset and Benchmark for Rheumatoid Arthritis

NeurIPS 2025poster

Rheumatoid arthritis (RA) is a common autoimmune disease that has been the focus of research in computer-aided diagnosis (CAD) and disease monitoring. In clinical settings, conventional radiography (CR) is widely used for the screening and evaluation of RA due to its low cost and accessibility. The…

Cited by 0SourcecodeScholar
2025

Retrieval Head Mechanistically Explains Long-Context Factuality

ICLR 2025oral

Despite the recent progress in long-context language models, it remains elusive how transformer-based models exhibit the capability to retrieve relevant information from arbitrary locations within the long context. This paper aims to address this question. Our systematic investigation across a wide…

2025

When Truthful Representations Flip Under Deceptive Instructions?

EMNLP 2025

Large language models (LLMs) tend to follow maliciously crafted instructions to generate deceptive responses, posing safety challenges. How deceptive instructions alter the internal representations of LLM compared to truthful ones remains poorly understood beyond output analysis. To bridge this gap,

2024

AutoGuide: Automated Generation and Selection of Context-Aware Guidelines for Large Language Model Agents

NeurIPS 2024poster

Recent advances in large language models (LLMs) have empowered AI agents capable of performing various sequential decision-making tasks. However, effectively guiding LLMs to perform well in unfamiliar domains like web navigation, where they lack sufficient knowledge, has proven to be difficult with…

Cited by 7SourcePDFScholar
2024

Data Engineering for Scaling Language Models to 128K Context

ICML 2024poster

We study continual pretraining recipe for scaling language models' context lengths to 128K, with a focus on data engineering. We hypothesize that long context modeling, in particular *the ability to utilize information at arbitrary input locations*, is a capability that is mostly already acquired th…

2024

MAmmoTH: Building Math Generalist Models through Hybrid Instruction Tuning

ICLR 2024spotlight

We introduce MAmmoTH, a series of open-source large language models (LLMs) specifically tailored for general math problem-solving. The MAmmoTH models are trained on MathInstruct, our meticulously curated instruction tuning dataset. MathInstruct is compiled from 13 math datasets with intermediate rat…

Cited by 332SourcePDFScholar
2024

OpenMoE: An Early Effort on Open Mixture-of-Experts Language Models

ICML 2024poster

To help the open-source community have a better understanding of Mixture-of-Experts (MoE) based large language models (LLMs), we train and release OpenMoE, a series of fully open-sourced and reproducible decoder-only MoE LLMs, ranging from 650M to 34B parameters and trained on up to over 1T tokens.…

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

Complexity-Based Prompting for Multi-step Reasoning

ICLR 2023poster

We study the task of prompting large-scale language models to perform multi-step reasoning. Existing work shows that when prompted with a chain of thoughts (CoT), sequences of short sentences describing intermediate reasoning steps towards a final answer, large language models can generate new reaso…

Cited by 416SourcePDFScholar
2023

Decomposed Prompting: A Modular Approach for Solving Complex Tasks

ICLR 2023poster

Few-shot prompting is a surprisingly powerful way to use Large Language Models (LLMs) to solve various tasks. However, this approach struggles as the task complexity increases or when the individual reasoning steps of the task themselves are hard to learn, especially when embedded in more complex ta…

2023

Specializing Smaller Language Models towards Multi-Step Reasoning

ICML 2023oral

The surprising ability of Large Language Models (LLMs) to perform well on complex reasoning with only few-shot chain-of-thought prompts is believed to emerge only in very large-scale models. We show that such abilities can, in fact, be distilled down from GPT-3.5 (≥ 175B) to T5 variants (≤ 11B). We…

2023

To Repeat or Not To Repeat: Insights from Scaling LLM under Token-Crisis

NeurIPS 2023poster

Recent research has highlighted the importance of dataset size in scaling language models. However, large language models (LLMs) are notoriously token-hungry during pre-training, and high-quality text data on the web is likely to be approaching its scaling limit for LLMs. To further enhance LLMs, a…

Cited by 81SourcePDFScholar
2021

Analyzing the Confidentiality of Undistillable Teachers in Knowledge Distillation

NeurIPS 2021poster

Knowledge distillation (KD) has recently been identified as a method that can unintentionally leak private information regarding the details of a teacher model to an unauthorized student. Recent research in developing undistillable nasty teachers that can protect model confidentiality has gained sig…

2021

Nested Named Entity Recognition with Partially-Observed TreeCRFs

AAAI 2021technical

Named entity recognition (NER) is a well-studied task in natural language processing. However, the widely-used sequence labeling framework is difficult to detect entities with nested structures. In this work, we view nested NER as constituency parsing with partially-observed trees and model it with…

2021

Noisy-Labeled NER with Confidence Estimation

NAACL 2021long

Recent studies in deep learning have shown significant progress in named entity recognition (NER). However, most existing works assume clean data annotation, while real-world scenarios typically involve a large amount of noises from a variety of sources (e.g., pseudo, weak, or distant annotations).…

2021

Probing BERT in Hyperbolic Spaces

ICLR 2021poster

Recently, a variety of probing tasks are proposed to discover linguistic properties learned in contextualized word embeddings. Many of these works implicitly assume these embeddings lay in certain metric spaces, typically the Euclidean space. This work considers a family of geometrically special spa…

2021

Prototypical Representation Learning for Relation Extraction

ICLR 2021poster

Recognizing relations between entities is a pivotal task of relational learning. Learning relation representations from distantly-labeled datasets is difficult because of the abundant label noise and complicated expressions in human language. This paper aims to learn predictive, interpretable, a…

2020

Latent Template Induction with Gumbel-CRFs

NeurIPS 2020poster

Learning to control the structure of sentences is a challenging problem in text generation. Existing work either relies on simple deterministic approaches or RL-based hard structures. We explore the use of structured variational autoencoders to infer latent templates for sentence generation using a…