Publications

2025

1. Blog

   DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas

   Zirui Wu*, Lin Zheng*, Zhihui Xie, and 7 more authors

   2025
2. Arxiv

   Dream 7B: Diffusion Large Language Models

   Jiacheng Ye, Zhihui Xie, Lin Zheng, and 5 more authors

   arXiv preprint arXiv:2508.15487, 2025
3. Arxiv

   Dream-Coder 7B: An Open Diffusion Language Model for Code

   Zhihui Xie, Jiacheng Ye, Lin Zheng, and 8 more authors

   arXiv preprint arXiv:2509.01142, 2025
4. Arxiv

   RefTool: Enhancing Model Reasoning with Reference-Guided Tool Creation

   Xiao Liu, Da Yin, Zirui Wu, and 1 more author

   arXiv preprint arXiv:2505.21413, 2025
5. EMNLP Findings

   Haste Makes Waste: Evaluating Planning Abilities of LLMs for Efficient and Feasible Multitasking with Time Constraints Between Actions

   Zirui Wu, Xiao Liu, Jiayi Li, and 2 more authors

   EMNLP Findings, 2025
6. Arxiv

   Automated Annotation of Evolving Corpora for Augmenting Longitudinal Network Data: A Framework Integrating Large Language Models and Expert Knowledge

   Xiao Liu, Zirui Wu, Jiayi Li, and 3 more authors

   arXiv preprint arXiv:2503.01672, 2025

2024

1. EMNLP Findings

   ProTrix: Building Models for Planning and Reasoning over Tables with Sentence Context

   Zirui Wu, and Yansong Feng

   EMNLP Findings, 2024

   Abs HTML

   Tables play a crucial role in conveying information in various domains. We propose a Plan-then-Reason framework to answer different types of user queries over tables with sentence context. The framework first plans the reasoning paths over the context, then assigns each step to program-based or textual reasoning to reach the final answer. This framework enhances the table reasoning abilities for both in-context learning and fine-tuning methods. GPT-3.5-Turbo following Plan-then-Reason framework surpasses other prompting baselines without self-consistency while using less API calls and in-context demonstrations. We also construct an instruction tuning set TrixInstruct to evaluate the effectiveness of fine-tuning with this framework. We present ProTrix model family by finetuning models on TrixInstruct. Our experiments show that ProTrix family generalizes to diverse unseen tabular tasks with only 6k training instances. We further demonstrate that ProTrix can generate accurate and faithful explanations to answer complex free-form questions. Our work underscores the importance of the planning and reasoning abilities towards a model over tabular tasks with generalizability and interpretability. We open-source our dataset and models at https://github.com/WilliamZR/ProTrix

2. ACL Findings

   Are llms capable of data-based statistical and causal reasoning? benchmarking advanced quantitative reasoning with data

   Xiao Liu, Zirui Wu, Xueqing Wu, and 3 more authors

   ACL Findings, 2024

   Abs HTML

   Quantitative reasoning is a critical skill to analyze data, yet the assessment of such ability remains limited. To address this gap, we introduce the Quantitative Reasoning with Data (QRData) benchmark, aiming to evaluate Large Language Models’ capability in statistical and causal reasoning with real-world data. The benchmark comprises a carefully constructed dataset of 411 questions accompanied by data sheets from textbooks, online learning materials, and academic papers. To compare models’ quantitative reasoning abilities on data and text, we enrich the benchmark with an auxiliary set of 290 text-only questions, namely QRText. We evaluate natural language reasoning, program-based reasoning, and agent reasoning methods including Chain-of-Thought, Program-of-Thoughts, ReAct, and code interpreter assistants on diverse models. The strongest model GPT-4 achieves an accuracy of 58%, which has much room for improvement. Among open-source models, Deepseek-coder-instruct, a code LLM pretrained on 2T tokens, gets the highest accuracy of 37%. Analysis reveals that models encounter difficulties in data analysis and causal reasoning, and struggle in using causal knowledge and provided data simultaneously. Code and data are in https://github.com/xxxiaol/QRData

2023

1. EMNLP

   Enhancing Structured Evidence Extraction for Fact Verification

   Zirui Wu, Nan Hu, and Yansong Feng

   EMNLP, 2023

   Abs HTML

   Open-domain fact verification is the task of verifying claims in natural language texts against extracted evidence. FEVEROUS is a benchmark that requires extracting and integrating both unstructured and structured evidence to verify a given claim. Previous models suffer from low recall of structured evidence extraction, i.e., table extraction and cell selection. In this paper, we propose a simple but effective method to enhance the extraction of structured evidence by leveraging the row and column semantics of tables. Our method comprises two components: (i) a coarse-grained table extraction module that selects tables based on rows and columns relevant to the claim and (ii) a fine-grained cell selection graph that combines both formats of evidence and enables multi-hop and numerical reasoning. We evaluate our method on FEVEROUS and achieve an evidence recall of 60.01% on the test set, which is 6.14% higher than the previous state-of-the-art performance. Our results demonstrate that our method can extract tables and select cells effectively, and provide better evidence sets for verdict prediction. Our code is released at https://github.com/WilliamZR/see-st

2. EACL

   UnifEE: Unified Evidence Extraction for Fact Verification

   Nan Hu*, Zirui Wu*, Yuxuan Lai, and 2 more authors

   EACL, 2023

   Abs HTML

   FEVEROUS is a fact extraction and verification task that requires systems to extract evidence of both sentences and table cells from a Wikipedia dump, then predict the veracity of the given claim accordingly. Existing works extract evidence in the two formats separately, ignoring potential connections between them. In this paper, we propose a Unified Evidence Extraction model (UnifEE), which uses a mixed evidence graph to extract the evidence in both formats. With the carefully-designed unified evidence graph, UnifEE allows evidence interactions among all candidates in both formats at similar granularity. Experiments show that, with information aggregated from related evidence candidates in the fusion graph, UnifEE can make better decisions about which evidence should be kept, especially for claims requiring multi-hop reasoning or a combination of tables and texts. Thus it outperforms all previous evidence extraction methods and brings significant improvement in the subsequent claim verification step.

3. Arxiv

   Lawyer LLaMA Technical Report

   Quzhe Huang, Mingxu Tao, Zhenwei An, and 5 more authors

   arXiv preprint , 2023

2022

1. NAACL

   Dual-Channel Evidence Fusion for Fact Verification over Texts and Tables

   Nan Hu, Zirui Wu, Yuxuan Lai, and 2 more authors

   NAACL, 2022

   Abs HTML

   Different from previous fact extraction and verification tasks that only consider evidence of a single format, FEVEROUS brings further challenges by extending the evidence format to both plain text and tables. Existing works convert all candidate evidence into either sentences or tables, thus often failing to fully capture the rich context in their original format from the converted evidence, let alone the context information lost during conversion. In this paper, we propose a Dual Channel Unified Format fact verification model (DCUF), which unifies various evidence into parallel streams, i.e., natural language sentences and a global evidence table, simultaneously. With carefully-designed evidence conversion and organization methods, DCUF makes the most of pre-trained table/language models to encourage each evidence piece to perform early and thorough interactions with other pieces in its original format. Experiments show that our model can make better use of existing pre-trained models to absorb evidence of two formats, thus outperforming previous works by a large margin. Our code and models are publicly available.

2020

1. Cellbow: a robust customizable cell segmentation program

   Huixia Ren, Mengdi Zhao, Bo Liu, and 7 more authors

   Quantitative Biology, 2020