← Search

Tianxiang Sun

24 accepted papers

2026

MuKV: Multi-Grained KV Cache Compression for Long Streaming Video Question-Answering

CVPR 2026

Long streaming video QA remains challenging due to growing visual tokens and limited reasoning length of large language models (LLMs). KV-caching stores the Key-Value (KV) of the historical tokens via LLM prefill and enables more efficient streaming QA. However, existing methods cache every one or t

Cited by 0SourceScholar
2025

Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performance

ICLR 2025poster

Pretraining data of large language models composes multiple domains (e.g., web texts, academic papers, codes), whose mixture proportions crucially impact the competence of outcome models. While existing endeavors rely on heuristics or qualitative strategies to tune the proportions, we discover the q…

2024

Aggregation of Reasoning: A Hierarchical Framework for Enhancing Answer Selection in Large Language Models

COLING 2024main

Recent advancements in Chain-of-Thought prompting have facilitated significant breakthroughs for Large Language Models (LLMs) in complex reasoning tasks. Current research enhances the reasoning performance of LLMs by sampling multiple reasoning chains and ensembling based on the answer frequency. Ho…

2024

AnyGPT: Unified Multimodal LLM with Discrete Sequence Modeling

ACL 2024long

We introduce AnyGPT, an any-to-any multimodal language model that utilizes discrete representations for the unified processing of various modalities, including speech, text, images, and music. AnyGPT can be trained stably without any alterations to the current large language model (LLM) architecture…

2024

Can AI Assistants Know What They Don't Know?

ICML 2024poster

AI assistants powered by Large Language Models (LLMs) have demonstrated impressive performance in various tasks. However, LLMs still make factual errors in knowledge-intensive tasks such as open-domain question answering. These untruthful responses from AI assistants can pose significant risks in pr…

2024

DenoSent: A Denoising Objective for Self-Supervised Sentence Representation Learning

AAAI 2024technical

Contrastive-learning-based methods have dominated sentence representation learning. These methods regularize the representation space by pulling similar sentence representations closer and pushing away the dissimilar ones and have been proven effective in various NLP tasks, e.g., semantic textual si…

2024

Flames: Benchmarking Value Alignment of LLMs in Chinese

NAACL 2024long

The widespread adoption of large language models (LLMs) across various regions underscores the urgent need to evaluate their alignment with human values. Current benchmarks, however, fall short of effectively uncovering safety vulnerabilities in LLMs. Despite numerous models achieving high scores an…

2024

LLM can Achieve Self-Regulation via Hyperparameter Aware Generation

ACL 2024findings

In the realm of Large Language Models (LLMs), users commonly employ diverse decoding strategies and adjust hyperparameters to control the generated text. However, a critical question emerges: Are LLMs conscious of the existence of these decoding strategies and capable of regulating themselves? The c…

Cited by 3SourcePDFScholar
2024

LLatrieval: LLM-Verified Retrieval for Verifiable Generation

NAACL 2024long

Verifiable generation aims to let the large language model (LLM) generate text with supporting documents, which enables the user to flexibly verify the answer and makes the LLM’s output more reliable. Retrieval plays a crucial role in verifiable generation. Specifically, the retrieved documents not…

2024

Turn Waste into Worth: Rectifying Top-k Router of MoE

EMNLP 2024main

Sparse Mixture of Experts (MoE) models are popular for training large language models due to their computational efficiency. However, the commonly used top-k routing mechanism suffers from redundancy computation and memory costs due to the unbalanced routing. Some experts are overflow, where the exc…

Cited by 2SourcePDFScholar
2024

Unified Active Retrieval for Retrieval Augmented Generation

EMNLP 2024finding

In Retrieval-Augmented Generation (RAG), retrieval is not always helpful and applying it to every instruction is sub-optimal. Therefore, determining whether to retrieve is crucial for RAG, which is usually referred to as Active Retrieval. However, existing active retrieval methods face two challenge…

2023

CodeIE: Large Code Generation Models are Better Few-Shot Information Extractors

ACL 2023long

Large language models (LLMs) pre-trained on massive corpora have demonstrated impressive few-shot learning ability on many NLP tasks. A common practice is to recast the task into a text-to-text format such that generative LLMs of natural language (NL-LLMs) like GPT-3 can be prompted to solve it. How…

2023

DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models

ACL 2023long

We present DiffusionBERT, a new generative masked language model based on discrete dif- fusion models. Diffusion models and many pre- trained language models have a shared training objective, i.e., denoising, making it possible to combine the two powerful models and enjoy the best of both worlds. On…

2023

Improving Contrastive Learning of Sentence Embeddings from AI Feedback

ACL 2023findings

Contrastive learning has become a popular approach in natural language processing, particularly for the learning of sentence embeddings.However, the discrete nature of natural language makes it difficult to ensure the quality of positive and negative sample pairs generated through data augmentation…

2023

Multitask Pre-training of Modular Prompt for Chinese Few-Shot Learning

ACL 2023long

Prompt tuning is a parameter-efficient approach to adapting pre-trained language models to downstream tasks. Although prompt tuning has been shown to match the performance of full model tuning when training data is sufficient, it tends to struggle in few-shot learning settings. In this paper, we pre…

2022

A Simple Hash-Based Early Exiting Approach For Language Understanding and Generation

ACL 2022findings

Early exiting allows instances to exit at different layers according to the estimation of difficulty. Previous works usually adopt heuristic metrics such as the entropy of internal outputs to measure instance difficulty, which suffers from generalization and threshold-tuning. In contrast, learning t…

2022

BBTv2: Towards a Gradient-Free Future with Large Language Models

EMNLP 2022main

Most downstream adaptation methods tune all or part of the parameters of pre-trained models (PTMs) through gradient descent, where the tuning cost increases linearly with the growth of the model size.By contrast, gradient-free methods only require the forward computation of the PTM to tune the promp…

2022

BERTScore is Unfair: On Social Bias in Language Model-Based Metrics for Text Generation

EMNLP 2022main

Automatic evaluation metrics are crucial to the development of generative systems. In recent years, pre-trained language model (PLM) based metrics, such as BERTScore, have been commonly adopted in various generation tasks. However, it has been demonstrated that PLMs encode a range of stereotypical s…

2022

Black-Box Tuning for Language-Model-as-a-Service

ICML 2022spotlight

Extremely large pre-trained language models (PTMs) such as GPT-3 are usually released as a service. It allows users to design task-specific prompts to query the PTMs through some black-box APIs. In such a scenario, which we call Language-Model-as-a-Service (LMaaS), the gradients of PTMs are usually…

2022

Late Prompt Tuning: A Late Prompt Could Be Better Than Many Prompts

EMNLP 2022finding

Prompt tuning is a parameter-efficient tuning (PETuning) method for utilizing pre-trained models (PTMs) that simply prepends a soft prompt to the input and only optimizes the prompt to adapt PTMs to downstream tasks. Although it is parameter- and deployment-efficient, its performance still lags behi…

2022

Towards Efficient NLP: A Standard Evaluation and A Strong Baseline

NAACL 2022long

Supersized pre-trained language models have pushed the accuracy of various natural language processing (NLP) tasks to a new state-of-the-art (SOTA). Rather than pursuing the reachless SOTA accuracy, more and more researchers start paying attention to model efficiency and usability. Different from ac…

2021

Accelerating BERT Inference for Sequence Labeling via Early-Exit

ACL 2021long

Both performance and efficiency are crucial factors for sequence labeling tasks in many real-world scenarios. Although the pre-trained models (PTMs) have significantly improved the performance of various sequence labeling tasks, their computational cost is expensive. To alleviate this problem, we ex…

2021

Does syntax matter? A strong baseline for Aspect-based Sentiment Analysis with RoBERTa

NAACL 2021long

Aspect-based Sentiment Analysis (ABSA), aiming at predicting the polarities for aspects, is a fine-grained task in the field of sentiment analysis. Previous work showed syntactic information, e.g. dependency trees, can effectively improve the ABSA performance. Recently, pre-trained models (PTMs) als…

2020

CoLAKE: Contextualized Language and Knowledge Embedding

COLING 2020main

With the emerging branch of incorporating factual knowledge into pre-trained language models such as BERT, most existing models consider shallow, static, and separately pre-trained entity embeddings, which limits the performance gains of these models. Few works explore the potential of deep contextu…