← Search

Tao Ge

27 accepted papers

2026

FormAct: Agentic Source Editing for Rich-Format Document Generation

ICML 2026poster

Rich-format documents are essential for everyday operations yet costly to author, motivating the need for automated generation to enhance productivity. To this end, we present FormAct, an agentic system that generates professional rich-format documents from scratch. FormAct operates on an HTML sourc…

Cited by 0SourceScholar
2025

Improving LLM General Preference Alignment via Optimistic Online Mirror Descent

NeurIPS 2025spotlight

Reinforcement learning from human feedback (RLHF) has demonstrated remarkable effectiveness in aligning large language models (LLMs) with human preferences. Many existing alignment approaches rely on the Bradley-Terry (BT) model assumption, which assumes the existence of a ground-truth reward for ea…

Cited by 0SourceScholar
2025

K-Level Reasoning: Establishing Higher Order Beliefs in Large Language Models for Strategic Reasoning

NAACL 2025long

Strategic reasoning is a complex yet essential capability for intelligent agents. It requires Large Language Model (LLM) agents to adapt their strategies dynamically in multi-agent environments. Unlike static reasoning tasks, success in these contexts depends on anticipating other agents’ beliefs an…

Cited by 0SourcePDFScholar
2025

Learn to Memorize: Scalable Continual Learning in Semiparametric Models with Mixture-of-Neighbors Induction Memory

ACL 2025long

Semiparametric language models (LMs) have shown promise in various Natural Language Processing (NLP) tasks. However, they utilize non-parametric memory as static storage, which lacks learning capability and remains disconnected from the internal information flow of the parametric models, limiting sc…

Cited by 0SourcePDFScholar
2025

Low-Bit Quantization Favors Undertrained LLMs

ACL 2025long

Low-bit quantization improves machine learning model efficiency but surprisingly favors undertrained large language models (LLMs). Larger models or those trained on fewer tokens exhibit less quantization-induced degradation (QiD), while smaller, well-trained models face significant performance losse…

Cited by 0SourcePDFScholar
2025

Router-Tuning: A Simple and Effective Approach for Dynamic Depth

EMNLP 2025

The Mixture of Depths (MoD) was introduced to improve computational efficiency by dynamically skipping less important layers, reducing redundant computation while maintaining model capacity. Despite its promise, existing MoD approaches remain under-explored and face two main challenges: (1) high tra

2024

In-context Autoencoder for Context Compression in a Large Language Model

ICLR 2024poster

We propose the In-context Autoencoder (ICAE), leveraging the power of a large language model (LLM) to compress a long context into short compact memory slots that can be directly conditioned on by the LLM for various purposes. ICAE is first pretrained using both autoencoding and language modeling ob…

2024

Learn Beyond The Answer: Training Language Models with Reflection for Mathematical Reasoning

EMNLP 2024main

Supervised fine-tuning enhances the problem-solving abilities of language models across various mathematical reasoning tasks. To maximize such benefits, existing research focuses on *broadening* the training set with various data augmentation techniques, which is effective for standard single-round…

2024

Low-code LLM: Graphical User Interface over Large Language Models

NAACL 2024system demonstrations

Utilizing Large Language Models (LLMs) for complex tasks is challenging, often involving a time-consuming and uncontrollable prompt engineering process. This paper introduces a novel human-LLM interaction framework, Low-code LLM. It incorporates six types of simple low-code visual programming intera…

2024

Refining Corpora from a Model Calibration Perspective for Chinese Spelling Correction

ACL 2024findings

Chinese Spelling Correction (CSC) commonly lacks large-scale high-quality corpora, due to the labor-intensive labeling of spelling errors in real-life human writing or typing scenarios. Two data augmentation methods are widely adopted: (1) *Random Replacement* with the guidance of confusion sets and…

Cited by 1SourcePDFScholar
2024

SCALE: Synergized Collaboration of Asymmetric Language Translation Engines

ACL 2024findings

In this paper, we introduce SCALE, a collaborative framework that connects a compact Specialized Translation Model (STM) and a general-purpose Large Language Model (LLM) as one unified translation engine. By introducing translation from STM into the triplet in-context demonstrations, SCALE unlocks r…

2024

Unleashing the Emergent Cognitive Synergy in Large Language Models: A Task-Solving Agent through Multi-Persona Self-Collaboration

NAACL 2024long

Human intelligence thrives on cognitive synergy, where collaboration among different minds yield superior outcomes compared to isolated individuals. In this work, we propose Solo Performance Prompting (SPP), which transforms a single LLM into a cognitive synergist by engaging in multi-turn self-coll…

2024

Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding

ACL 2024findings

To mitigate the high inference latency stemming from autoregressive decoding in Large Language Models (LLMs), Speculative Decoding has emerged as a novel decoding paradigm for LLM inference. In each decoding step, this method first drafts several future tokens efficiently and then verifies them in p…

2024

xRAG: Extreme Context Compression for Retrieval-augmented Generation with One Token

NeurIPS 2024poster

This paper introduces xRAG, an innovative context compression method tailored for retrieval-augmented generation. xRAG reinterprets document embeddings in dense retrieval--traditionally used solely for retrieval--as features from the retrieval modality. By employing a modality fusion methodology, xR…

2023

Extensible Prompts for Language Models on Zero-shot Language Style Customization

NeurIPS 2023poster

We propose eXtensible Prompt (X-Prompt) for prompting a large language model (LLM) beyond natural language (NL). X-Prompt instructs an LLM with not only NL but also an extensible vocabulary of imaginary words. Registering new imaginary words allows us to instruct the LLM to comprehend concepts that…

Cited by 3SourcePDFScholar
2023

Smart Word Suggestions for Writing Assistance

ACL 2023findings

Enhancing word usage is a desired feature for writing assistance. To further advance research in this area, this paper introduces “Smart Word Suggestions” (SWS) task and benchmark. Unlike other works, SWS emphasizes end-to-end evaluation and presents a more realistic writing assistance scenario. Thi…

2023

Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generation

EMNLP 2023long findings

We propose Speculative Decoding (SpecDec), for the first time ever, to formally study exploiting the idea of speculative execution to accelerate autoregressive (AR) decoding. Speculative Decoding has two innovations: Spec-Drafter -- an independent model specially optimized for efficient and accurate…

Cited by 0SourcecodeScholar
2022

A Unified Strategy for Multilingual Grammatical Error Correction with Pre-trained Cross-Lingual Language Model

IJCAI 2022poster

Synthetic data construction of Grammatical Error Correction (GEC) for non-English languages relies heavily on human-designed and language-specific rules, which produce limited error-corrected patterns. In this paper, we propose a generic and language-independent strategy for multilingual GEC, which…

2022

EdgeFormer: A Parameter-Efficient Transformer for On-Device Seq2seq Generation

EMNLP 2022main

We introduce EdgeFormer – a parameter-efficient Transformer for on-device seq2seq generation under the strict computation and memory constraints. Compared with the previous parameter-efficient Transformers, EdgeFormer applies two novel principles for cost-effective parameterization, allowing it to p…

2022

Text Revision By On-the-Fly Representation Optimization

AAAI 2022technical

Text revision refers to a family of natural language generation tasks, where the source and target sequences share moderate resemblance in surface form but differentiate in attributes, such as text formality and simplicity. Current state-of-the-art methods formulate these tasks as sequence-to-sequen…

2021

Beyond Preserved Accuracy: Evaluating Loyalty and Robustness of BERT Compression

EMNLP 2021main

Recent studies on compression of pretrained language models (e.g., BERT) usually use preserved accuracy as the metric for evaluation. In this paper, we propose two new metrics, label loyalty and probability loyalty that measure how closely a compressed model (i.e., student) mimics the original model…

2021

Blow the Dog Whistle: A Chinese Dataset for Cant Understanding with Common Sense and World Knowledge

NAACL 2021long

Cant is important for understanding advertising, comedies and dog-whistle politics. However, computational research on cant is hindered by a lack of available datasets. In this paper, we propose a large and diverse Chinese dataset for creating and understanding cant from a computational linguistics…

2021

Improving Sequence-to-Sequence Pre-training via Sequence Span Rewriting

EMNLP 2021main

In this paper, we propose Sequence Span Rewriting (SSR), a self-supervised task for sequence-to-sequence (Seq2Seq) pre-training. SSR learns to refine the machine-generated imperfect text spans into ground truth text. SSR provides more fine-grained and informative supervision in addition to the origi…

2021

Instantaneous Grammatical Error Correction with Shallow Aggressive Decoding

ACL 2021long

In this paper, we propose Shallow Aggressive Decoding (SAD) to improve the online inference efficiency of the Transformer for instantaneous Grammatical Error Correction (GEC). SAD optimizes the online inference efficiency for GEC by two innovations: 1) it aggressively decodes as many tokens as possi…

2020

BERT Loses Patience: Fast and Robust Inference with Early Exit

NeurIPS 2020poster

In this paper, we propose Patience-based Early Exit, a straightforward yet effective inference method that can be used as a plug-and-play technique to simultaneously improve the efficiency and robustness of a pretrained language model (PLM). To achieve this, our approach couples an internal-classifi…

2020

Self-Adversarial Learning with Comparative Discrimination for Text Generation

ICLR 2020poster

Conventional Generative Adversarial Networks (GANs) for text generation tend to have issues of reward sparsity and mode collapse that affect the quality and diversity of generated samples. To address the issues, we propose a novel self-adversarial learning (SAL) paradigm for improving GANs' performa…

Cited by 31SourceScholar