← Search

Xinyao Niu

4 accepted papers

2026

A$^2$Search: Ambiguity-Aware Question Answering with Reinforcement Learning

ICLR 2026poster

Recent advances in Large Language Models (LLMs) and Reinforcement Learning (RL) have led to strong performance in open-domain question answering (QA). However, existing models still struggle with questions that admit multiple valid answers. Standard QA benchmarks, which typically assume a single gol…

Cited by 0Source ↗code ↗Scholar ↗
2024

Data Engineering for Scaling Language Models to 128K Context

ICML 2024poster

We study continual pretraining recipe for scaling language models' context lengths to 128K, with a focus on data engineering. We hypothesize that long context modeling, in particular *the ability to utilize information at arbitrary input locations*, is a capability that is mostly already acquired th…

2024

Machine Unlearning of Pre-trained Large Language Models

ACL 2024long

This study investigates the concept of the ‘right to be forgotten’ within the context of large language models (LLMs). We explore machine unlearning as a pivotal solution, with a focus on pre-trained models–a notably under-researched area. Our research delineates a comprehensive framework for machin…