← Search

Kai Song

11 accepted papers

2026

Sparser Block-Sparse Attention via Token Permutation

ICML 2026poster

Scaling the context length of large language models (LLMs) offers significant benefits but is computationally expensive. This expense stems primarily from the self-attention mechanism, whose $O(N^2)$ complexity with respect to sequence length presents a major bottleneck for both memory and latency. …

Cited by 0SourceScholar
2025

An Empirical Study of Iterative Refinements for Non-autoregressive Translation

ACL 2025long

Iterative non-autoregressive (NAR) models share a spirit of mixed autoregressive (AR) and fully NAR models, seeking a balance between generation quality and inference efficiency. These models have recently demonstrated impressive performance in varied generation tasks, surpassing the autoregressive…

2025

CAMIEval: Enhancing NLG Evaluation through Multidimensional Comparative Instruction-Following Analysis

NAACL 2025long

With the rapid development of large language models (LLMs), due to their strong performance across various fields, LLM-based evaluation methods (LLM-as-a-Judge) have become widely used in natural language generation (NLG) evaluation. However, these methods encounter the following challenges: (1) dis…

Cited by 0SourcePDFScholar
2025

ConvSearch-R1: Enhancing Query Reformulation for Conversational Search with Reasoning via Reinforcement Learning

EMNLP 2025

Conversational search systems require effective handling of context-dependent queries that often contain ambiguity, omission, and coreference. Conversational Query Reformulation (CQR) addresses this challenge by transforming these queries into self-contained forms suitable for off-the-shelf retrieve

2025

FiNE: Filtering and Improving Noisy Data Elaborately with Large Language Models

NAACL 2025long

Data is the lifeblood of large language models (LLMs). While the quantity of open-source data available for training LLMs is substantial, its integrity often falls short. For instance, the open-source chat version of Yi-1.5-9B scores 5.20 on AlignBench, while the Chinese Alpaca-GPT4 version scores 4…

Cited by 0SourcePDFScholar
2025

MRO: Enhancing Reasoning in Diffusion Language Models via Multi-Reward Optimization

NeurIPS 2025poster

Recent advances in diffusion language models (DLMs) have presented a promising alternative to traditional autoregressive large language models (LLMs). However, DLMs still lag behind LLMs in reasoning performance, especially as the number of denoising steps decreases. Our analysis reveals that this s…

Cited by 0SourceScholar
2024

Revealing the Parallel Multilingual Learning within Large Language Models

EMNLP 2024main

Large language models (LLMs) can handle multilingual and cross-lingual text within a single input; however, previous works leveraging multilingualism in LLMs primarily focus on using English as the pivot language to enhance language understanding and reasoning. Given that multiple languages are a co…

2024

Semformer: Transformer Language Models with Semantic Planning

EMNLP 2024main

Next-token prediction serves as the dominant component in current neural language models.During the training phase, the model employs teacher forcing, which predicts tokens based on all preceding ground truth tokens.However, this approach has been found to create shortcuts, utilizing the revealed pr…

2024

VOLTA: Improving Generative Diversity by Variational Mutual Information Maximizing Autoencoder

NAACL 2024findings

The natural language generation domain has witnessed great success thanks to Transformer models. Although they have achieved state-of-the-art generative quality, they often neglect generative diversity. Prior attempts to tackle this issue suffer from either low model capacity or over-complicated arc…

Cited by 0SourcePDFScholar