← Search

Yushi Bai

21 accepted papers

2026

LongWriter-Zero: Mastering Ultra-Long Text Generation via Reinforcement Learning

ICLR 2026oral

Ultra-long generation by large language models (LLMs) is a widely demanded scenario, yet it remains a significant challenge due to their maximum generation length limit and overall quality degradation as sequence length increases. Previous approaches, exemplified by LongWriter, typically rely on ''t…

Cited by 0SourceScholar
2025

CogCoM: A Visual Language Model with Chain-of-Manipulations Reasoning

ICLR 2025poster

Vision-Language Models (VLMs) have shown broad effectiveness due to extensive training that aligns visual inputs with corresponding language responses. However, this conclusive alignment training causes models to overlook essential visual reasoning, leading to failures in handling detailed visual ta…

2025

How do Transformers Learn Implicit Reasoning?

NeurIPS 2025spotlight

Recent work suggests that large language models (LLMs) can perform multi-hop reasoning implicitly---producing correct answers without explicitly verbalizing intermediate steps---but the underlying mechanisms remain poorly understood. In this paper, we study how such implicit reasoning emerges by tra…

Cited by 0SourceScholar
2025

LongBench v2: Towards Deeper Understanding and Reasoning on Realistic Long-context Multitasks

ACL 2025long

This paper introduces LongBench v2, a benchmark designed to assess the ability of LLMs to handle long-context problems requiring deep understanding and reasoning across real-world multitasks. LongBench v2 consists of 503 challenging multiple-choice questions, with contexts ranging from 8k to 2M word…

2025

LongCite: Enabling LLMs to Generate Fine-grained Citations in Long-Context QA

ACL 2025finding

Though current long-context large language models (LLMs) have demonstrated impressive capacities in answering various questions based on extensive text, the lack of citations in their responses makes user verification difficult, leading to concerns about their trustworthiness due to the potential ha…

2025

LongWriter: Unleashing 10,000+ Word Generation from Long Context LLMs

ICLR 2025poster

Current long context large language models (LLMs) can process inputs up to 100,000 tokens, yet struggle to generate outputs exceeding even a modest length of 2,000 words. Through controlled experiments, we find that the model's effective generation length is inherently bounded by the sample it has s…

2025

Pre-training Distillation for Large Language Models: A Design Space Exploration

ACL 2025long

Knowledge distillation (KD) aims to transfer knowledge from a large teacher model to a smaller student model. Previous work applying KD in the field of large language models (LLMs) typically focused on the post-training phase, where the student LLM learns directly from instructions and corresponding…

Cited by 0SourcePDFScholar
2025

Towards Understanding Safety Alignment: A Mechanistic Perspective from Safety Neurons

NeurIPS 2025poster

Large language models (LLMs) excel in various capabilities but pose safety risks such as generating harmful content and misinformation, even after safety alignment. In this paper, we explore the inner mechanisms of safety alignment through the lens of mechanistic interpretability, focusing on identi…

Cited by 0SourceScholar
2024

AlphaTablets: A Generic Plane Representation for 3D Planar Reconstruction from Monocular Videos

NeurIPS 2024poster

We introduce AlphaTablets, a novel and generic representation of 3D planes that features continuous 3D surface and precise boundary delineation. By representing 3D planes as rectangles with alpha channels, AlphaTablets combine the advantages of current 2D and 3D plane representations, enabling accur…

Cited by 0SourcePDFScholar
2024

Automating Dataset Updates Towards Reliable and Timely Evaluation of Large Language Models

NeurIPS 2024poster

Large language models (LLMs) have achieved impressive performance across various natural language benchmarks, prompting a continual need to curate more difficult datasets for larger LLMs, which is costly and time-consuming. In this paper, we propose to automate dataset updating and provide systemati…

2024

KoLA: Carefully Benchmarking World Knowledge of Large Language Models

ICLR 2024poster

The unprecedented performance of large language models (LLMs) necessitates improvements in evaluations. Rather than merely exploring the breadth of LLM abilities, we believe meticulous and thoughtful designs are essential to thorough, unbiased, and applicable evaluations. Given the importance of wor…

2024

Large Language Models Can Be Contextual Privacy Protection Learners

EMNLP 2024main

The proliferation of Large Language Models (LLMs) has driven considerable interest in fine-tuning them with domain-specific data to create specialized language models. Nevertheless, such domain-specific fine-tuning data often contains contextually sensitive personally identifiable information (PII).…

2024

LongAlign: A Recipe for Long Context Alignment of Large Language Models

EMNLP 2024finding

Extending large language models to effectively handle long contexts requires instruction fine-tuning on input sequences of similar length. To address this, we present LongAlign—a recipe of the instruction data, training, and evaluation for long context alignment. First, we construct a long instructi…

2024

LongBench: A Bilingual, Multitask Benchmark for Long Context Understanding

ACL 2024long

Although large language models (LLMs) demonstrate impressive performance for many language tasks, most of them can only handle texts a few thousand tokens long, limiting their applications on longer sequence inputs, such as books, reports, and codebases. Recent works have proposed methods to improve…

2024

MM-MATH: Advancing Multimodal Math Evaluation with Process Evaluation and Fine-grained Classification

EMNLP 2024finding

To advance the evaluation of multimodal math reasoning in large multimodal models (LMMs), this paper introduces a novel benchmark, MM-MATH. MM-MATH consists of 5,929 open-ended middle school math problems with visual contexts, with fine-grained classification across difficulty, grade level, and know…

2024

WaterBench: Towards Holistic Evaluation of Watermarks for Large Language Models

ACL 2024long

To mitigate the potential misuse of large language models (LLMs), recent research has developed watermarking algorithms, which restrict the generation process to leave an invisible trace for watermark detection. Due to the two-stage nature of the task, most studies evaluate the generation and detect…

2023

Answering Complex Logical Queries on Knowledge Graphs via Query Computation Tree Optimization

ICML 2023poster

Answering complex logical queries on incomplete knowledge graphs is a challenging task, and has been widely studied. Embedding-based methods require training on complex queries and may not generalize well to out-of-distribution query structures. Recent work frames this task as an end-to-end optimiza…

2023

Benchmarking Foundation Models with Language-Model-as-an-Examiner

NeurIPS 2023poster

Numerous benchmarks have been established to assess the performance of foundation models on open-ended question answering, which serves as a comprehensive test of a model's ability to understand and generate language in a manner similar to humans. Most of these works focus on proposing new datasets,…

Cited by 141SourcePDFScholar
2022

Envy-Free and Pareto-Optimal Allocations for Agents with Asymmetric Random Valuations

IJCAI 2022poster

We study the problem of allocating m indivisible items to n agents with additive utilities. It is desirable for the allocation to be both fair and efficient, which we formalize through the notions of envy-freeness and Pareto-optimality. While envy-free and Pareto-optimal allocations may not exist fo…

2022

SQUIRE: A Sequence-to-sequence Framework for Multi-hop Knowledge Graph Reasoning

EMNLP 2022main

Multi-hop knowledge graph (KG) reasoning has been widely studied in recent years to provide interpretable predictions on missing links with evidential paths. Most previous works use reinforcement learning (RL) based methods that learn to navigate the path towards the target entity. However, these me…

2021

Modeling Heterogeneous Hierarchies with Relation-specific Hyperbolic Cones

NeurIPS 2021poster

Hierarchical relations are prevalent and indispensable for organizing human knowledge captured by a knowledge graph (KG). The key property of hierarchical relations is that they induce a partial ordering over the entities, which needs to be modeled in order to allow for hierarchical reasoning. Howev…