← Search

Lei Shu

16 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
2025

Accelerating Inference of Retrieval-Augmented Generation via Sparse Context Selection

ICLR 2025poster

Large language models (LLMs) augmented with retrieval exhibit robust performance and extensive versatility by incorporating external contexts. However, the input length grows linearly in the number of retrieved documents, causing a dramatic increase in latency. In this paper, we propose a novel para…

Cited by 16SourcePDFScholar
2024

Enhancing Reinforcement Learning with Dense Rewards from Language Model Critic

EMNLP 2024main

Reinforcement learning (RL) can align language models with non-differentiable reward signals, such as human preferences. However, a major challenge arises from the sparsity of these reward signals - typically, there is only a single reward for an entire output. This sparsity of rewards can lead to i…

Cited by 9SourcePDFScholar
2024

Fusion-Eval: Integrating Assistant Evaluators with LLMs

EMNLP 2024industry

Evaluating natural language generation (NLG) systems automatically poses significant challenges.Recent studies have employed large language models (LLMs) as reference-free metrics for NLG evaluation, enhancing adaptability to new tasks tasks. However, these methods still show lower correspondence wi…

2024

RewriteLM: An Instruction-Tuned Large Language Model for Text Rewriting

AAAI 2024technical

Large Language Models (LLMs) have demonstrated impressive capabilities in creative tasks such as storytelling and E-mail generation. However, as LLMs are primarily trained on final text results rather than intermediate revisions, it might be challenging for them to perform text rewriting tasks. Mos…

2024

Towards an On-device Agent for Text Rewriting

NAACL 2024findings

Large Language Models (LLMs) have demonstrated impressive capabilities for text rewriting. However creating a smaller yet potent language model for text rewriting presents two formidable challenges: costly data collection and absence of emergent capabilities.In this paper we present solutions to add…

Cited by 11SourcePDFScholar
2022

Adapting a Language Model While Preserving its General Knowledge

EMNLP 2022main

Domain-adaptive pre-training (or DA-training for short), also known as post-training, aimsto train a pre-trained general-purpose language model (LM) using an unlabeled corpus of aparticular domain to adapt the LM so that end-tasks in the domain can give improved performances. However, existing DA-tr…

2022

Continual Training of Language Models for Few-Shot Learning

EMNLP 2022main

Recent work on applying large language models (LMs) achieves impressive performance in many NLP applications. Adapting or posttraining an LM using an unlabeled domain corpus can produce even better performance for end-tasks in the domain. This paper proposes the problem of continually extending an L…

2022

Measuring and Reducing Model Update Regression in Structured Prediction for NLP

NeurIPS 2022accept

Recent advance in deep learning has led to rapid adoption of machine learning based NLP models in a wide range of applications. Despite the continuous gain in accuracy, backward compatibility is also an important aspect for industrial applications, yet it received little research attention. Backward…

Cited by 10SourcePDFScholar
2022

Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System

ACL 2022long

Pre-trained language models have been recently shown to benefit task-oriented dialogue (TOD) systems. Despite their success, existing methods often formulate this task as a cascaded generation problem which can lead to error accumulation across different sub-tasks and greater data annotation overhea…

2022

TaCL: Improving BERT Pre-training with Token-aware Contrastive Learning

NAACL 2022findings

Masked language models (MLMs) such as BERT have revolutionized the field of Natural Language Understanding in the past few years. However, existing pre-trained MLMs often output an anisotropic distribution of token representations that occupies a narrow subset of the entire representation space. Suc…

2022

Zero-Shot Out-of-Distribution Detection Based on the Pre-trained Model CLIP

AAAI 2022technical

In an out-of-distribution (OOD) detection problem, samples of known classes (also called in-distribution classes) are used to train a special classifier. In testing, the classifier can (1) classify the test samples of known classes to their respective classes and also (2) detect samples that do not…

2021

Achieving Forgetting Prevention and Knowledge Transfer in Continual Learning

NeurIPS 2021poster

Continual learning (CL) learns a sequence of tasks incrementally with the goal of achieving two main objectives: overcoming catastrophic forgetting (CF) and encouraging knowledge transfer (KT) across tasks. However, most existing techniques focus only on overcoming CF and have no mechanism to encour…

2021

CLASSIC: Continual and Contrastive Learning of Aspect Sentiment Classification Tasks

EMNLP 2021main

This paper studies continual learning (CL) of a sequence of aspect sentiment classification (ASC) tasks in a particular CL setting called domain incremental learning (DIL). Each task is from a different domain or product. The DIL setting is particularly suited to ASC because in testing the system ne…

2021

ODIST: Open World Classification via Distributionally Shifted Instances

EMNLP 2021finding

In this work, we address the open-world classification problem with a method called ODIST, open world classification via distributionally shifted instances. This novel and straightforward method can create out-of-domain instances from the in-domain training instances with the help of a pre-trained g…