← Search

Xiang Kong

14 accepted papers

2025

Checklists Are Better Than Reward Models For Aligning Language Models

NeurIPS 2025spotlight

Language models must be adapted to understand and follow user instructions. Reinforcement learning is widely used to facilitate this —typically using fixed criteria such as "helpfulness" and "harmfulness". In our work, we instead propose using flexible, instruction-specific criteria as a means of br…

Cited by 0SourceScholar
2025

MMAU: A Holistic Benchmark of Agent Capabilities Across Diverse Domains

NAACL 2025findings

Recent advances in large language models (LLMs) have increased the demand for comprehensive benchmarks to evaluate their capabilities as human-like agents. Existing benchmarks, while useful, often focus on specific application scenarios, emphasizing task completion but failing to dissect the underly…

2025

Mutual Reinforcement of LLM Dialogue Synthesis and Summarization Capabilities for Few-Shot Dialogue Summarization

NAACL 2025findings

In this work, we propose Mutual Reinforcing Data Synthesis (MRDS) within LLMs to improve few-shot dialogue summarization task. Unlike prior methods that require external knowledge, we mutually reinforce the LLM’s dialogue synthesis and summarization capabilities, allowing them to complement each oth…

Cited by 0SourcePDFScholar
2025

Step-by-Step Reasoning for Math Problems via Twisted Sequential Monte Carlo

ICLR 2025poster

Augmenting the multi-step reasoning abilities of Large Language Models (LLMs) has been a persistent challenge. Recently, verification has shown promise in improving solution consistency by evaluating generated outputs. However, current verification approaches suffer from sampling inefficiencies, req…

Cited by 1SourcePDFScholar
2025

TIS-DPO: Token-level Importance Sampling for Direct Preference Optimization With Estimated Weights

ICLR 2025poster

Direct Preference Optimization (DPO) has been widely adopted for preference alignment of Large Language Models (LLMs) due to its simplicity and effectiveness. However, DPO is derived as a bandit problem in which the whole response is treated as a single arm, ignoring the importance differences betw…

2024

"MM1: Methods, Analysis & Insights from Multimodal LLM Pre-training"

ECCV 2024poster

"In this work, we discuss building performant Multimodal Large Language Models (MLLMs). In particular, we study the importance of various architecture components and data choices. Through careful and comprehensive ablations of the image encoder, the vision language connector, and various pre-trainin…

2024

Direct Large Language Model Alignment Through Self-Rewarding Contrastive Prompt Distillation

ACL 2024long

Aligning large language models (LLMs) with human expectations without human-annotated preference data is an important problem. In this paper, we propose a method to evaluate the response preference by using the output probabilities of response pairs under contrastive prompt pairs, which could achiev…

2023

Mega: Moving Average Equipped Gated Attention

ICLR 2023poster

The design choices in the Transformer attention mechanism, including weak inductive bias and quadratic computational complexity, have limited its application for modeling long sequences. In this paper, we introduce Mega, a simple, theoretically grounded, single-head gated attention mechanism equippe…

2022

BLT: Bidirectional Layout Transformer for Controllable Layout Generation

ECCV 2022poster

"Creating visual layouts is a critical step in graphic design. Automatic generation of such layouts is essential for scalable and diverse visual designs. To advance conditional layout generation, we introduce BLT, a bidirectional layout transformer. BLT differs from previous work on transformers in…

2021

Decoupling Global and Local Representations via Invertible Generative Flows

ICLR 2021poster

In this work, we propose a new generative model that is capable of automatically decoupling global and local representations of images in an entirely unsupervised setting, by embedding a generative flow in the VAE framework to model the decoder. Specifically, the proposed model utilizes the variatio…

2021

Luna: Linear Unified Nested Attention

NeurIPS 2021poster

The quadratic computational and memory complexities of the Transformer's attention mechanism have limited its scalability for modeling long sequences. In this paper, we propose Luna, a linear unified nested attention mechanism that approximates softmax attention with two nested linear attention fun…

2017

Evaluating automatic speech recognition systems in comparison with human perception results using distinctive feature measures

ICASSP 2017accepted

This paper describes methods for evaluating automatic speech recognition (ASR) systems in comparison with human perception results, using measures derived from linguistic distinctive features. Error patterns in terms of manner, place and voicing are presented, along with an examination of confusion…

Cited by 0SourceScholar