← 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 0SourcecodeScholar
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…