← Search

Kewei Tu

60 accepted papers

2026

Repurposing Synthetic Data for Fine-grained Search Agent Supervision

ICLR 2026poster

LLM-based search agents are increasingly trained on entity-centric synthetic data to solve complex, knowledge-intensive tasks. However, prevailing training methods like Group Relative Policy Optimization (GRPO) discard this rich entity information, relying instead on sparse, outcome-based rewards. T…

Cited by 0SourceScholar
2026

Scaling Agents via Continual Pre-training

ICLR 2026poster

Large language models (LLMs) have evolved into agentic systems capable of autonomous tool use and multi-step reasoning for complex problem-solving. However, post-training approaches building upon general-purpose foundation models consistently underperform in agentic tasks, particularly in open-sourc…

Cited by 0SourcecodeScholar
2025

A Systematic Study of Compositional Syntactic Transformer Language Models

ACL 2025long

Syntactic language models (SLMs) enhance Transformers by incorporating syntactic biases through the modeling of linearized syntactic parse trees alongside surface sentences. This paper focuses on compositional SLMs that are based on constituency parse trees and contain explicit bottom-up composition…

2025

Detecting Knowledge Boundary of Vision Large Language Models by Sampling-Based Inference

EMNLP 2025

Despite the advancements made in Vision Large Language Models (VLLMs), like text Large Language Models (LLMs), they have limitations in addressing questions that require real-time information or are knowledge-intensive. Indiscriminately adopting Retrieval Augmented Generation (RAG) techniques is an

2025

Efficient Length-Generalizable Attention via Causal Retrieval for Long-Context Language Modeling

ICML 2025poster

Despite the success of Transformers, handling longer contexts remains challenging due to the limited length generalization and quadratic complexity of self-attention, which often requires post-training with a larger attention window, significantly increasing computational and memory costs. In this p…

Cited by 0SourcePDFScholar
2025

EvolveSearch: An Iterative Self-Evolving Search Agent

EMNLP 2025

The rapid advancement of large language models (LLMs) has transformed the landscape of agentic information seeking capabilities through the integration of tools such as search engines and web browsers. However, current mainstream approaches for enabling LLM web search proficiency face significant ch

Cited by 0SourcePDFScholar
2025

Hardware-aligned Hierarchical Sparse Attention for Efficient Long-term Memory Access

NeurIPS 2025poster

A key advantage of Recurrent Neural Networks (RNNs) over Transformers is their linear computational and space complexity enables faster training and inference for long sequences. However, RNNs are fundamentally unable to randomly access historical context, and simply integrating attention mechanisms…

Cited by 0SourceScholar
2025

Look Both Ways and No Sink: Converting LLMs into Text Encoders without Training

ACL 2025long

Recent advancements have demonstrated the advantage of converting pretrained large language models into powerful text encoders by enabling bidirectional attention in transformer layers. However, existing methods often require extensive training on large-scale datasets, posing challenges in low-resou…

2025

Tree-Structured Non-Autoregressive Decoding for Sequence-to-Sequence Text Generation

EMNLP 2025

Autoregressive Transformer (AT) dominates sequence-to-sequence generation tasks but suffers from high inference latency due to sequential token generation. Non-Autoregressive Transformer (NAT) improves inference efficiency by parallelizing token prediction, yet degrades generation quality. To addres

2024

Augmenting Transformers with Recursively Composed Multi-grained Representations

ICLR 2024poster

We present ReCAT, a recursive composition augmented Transformer that is able to explicitly model hierarchical syntactic structures of raw texts without relying on gold trees during both learning and inference. Existing research along this line restricts data to follow a hierarchical tree structure…

2024

Dependency Transformer Grammars: Integrating Dependency Structures into Transformer Language Models

ACL 2024long

Syntactic Transformer language models aim to achieve better generalization through simultaneously modeling syntax trees and sentences. While prior work has been focusing on adding constituency-based structures to Transformers, we introduce Dependency Transformer Grammars (DTGs), a new class of Trans…

2024

Generative Pretrained Structured Transformers: Unsupervised Syntactic Language Models at Scale

ACL 2024long

A syntactic language model (SLM) incrementally generates a sentence with its syntactic tree in a left-to-right manner.We present Generative Pretrained Structured Transformers (GPST), an unsupervised SLM at scale capable of being pre-trained from scratch on raw texts with high parallelism. GPST circu…

2024

Improving Retrieval Augmented Open-Domain Question-Answering with Vectorized Contexts

ACL 2024findings

In the era of large language models, applying techniques such as Retrieval Augmented Generation can better address Open-Domain Question-Answering problems. Due to constraints including model sizes and computing resources, the length of context is often limited, and it becomes challenging to empower…

2024

SeqGPT: An Out-of-the-Box Large Language Model for Open Domain Sequence Understanding

AAAI 2024technical

Large language models (LLMs) have shown impressive abilities for open-domain NLP tasks. However, LLMs are sometimes too footloose for natural language understanding (NLU) tasks which always have restricted output and input format. Their performances on NLU tasks are highly related to prompts or demo…

2023

A Multi-Grained Self-Interpretable Symbolic-Neural Model For Single/Multi-Labeled Text Classification

ICLR 2023poster

Deep neural networks based on layer-stacking architectures have historically suffered from poor inherent interpretability. Meanwhile, symbolic probabilistic models function with clear interpretability, but how to combine them with neural networks to enhance their performance remains to be explored.…

Cited by 4SourcePDFScholar
2023

Conic10K: A Challenging Math Problem Understanding and Reasoning Dataset

EMNLP 2023long findings

Mathematical understanding and reasoning are crucial tasks for assessing the capabilities of artificial intelligence (AI). However, existing benchmarks either require just a few steps of reasoning, or only contain a small amount of data in one specific topic, making it hard to analyse AI's behaviour…

Cited by 0SourcecodeScholar
2023

Do PLMs Know and Understand Ontological Knowledge?

ACL 2023long

Ontological knowledge, which comprises classes and properties and their relationships, is integral to world knowledge. It is significant to explore whether Pretrained Language Models (PLMs) know and understand such knowledge. However, existing PLM-probing studies focus mainly on factual knowledge, l…

2023

Don’t Parse, Choose Spans! Continuous and Discontinuous Constituency Parsing via Autoregressive Span Selection

ACL 2023long

We present a simple and unified approach for both continuous and discontinuous constituency parsing via autoregressive span selection. Constituency parsing aims to produce a set of non-crossing spans so that they can form a constituency parse tree. We sort gold spans using a predefined order and lev…

Cited by 4SourcePDFScholar
2023

Improving Grammar-based Sequence-to-Sequence Modeling with Decomposition and Constraints

ACL 2023short

Neural QCFG is a grammar-based sequence-to-sequence model with strong inductive biases on hierarchical structures. It excels in interpretability and generalization but suffers from expensive inference. In this paper, we study two low-rank variants of Neural QCFG for faster inference with different t…

2023

Improving Low-resource Named Entity Recognition with Graph Propagated Data Augmentation

ACL 2023short

Data augmentation is an effective solution to improve model performance and robustness for low-resource named entity recognition (NER). However, synthetic data often suffer from poor diversity, which leads to performance limitations. In this paper, we propose a novel Graph Propagated Data Augmentati…

Cited by 8SourcePDFScholar
2023

Joint Entity and Relation Extraction with Span Pruning and Hypergraph Neural Networks

EMNLP 2023long main

Entity and Relation Extraction (ERE) is an important task in information extraction. Recent marker-based pipeline models achieve state-of-the-art performance, but still suffer from the error propagation issue. Also, most of current ERE models do not take into account higher-order interactions betwee…

Cited by 0SourcecodeScholar
2023

Modeling Instance Interactions for Joint Information Extraction with Neural High-Order Conditional Random Field

ACL 2023long

Prior works on joint Information Extraction (IE) typically model instance (e.g., event triggers, entities, roles, relations) interactions by representation enhancement, type dependencies scoring, or global decoding. We find that the previous models generally consider binary type dependency scoring o…

2023

Probabilistic Transformer: A Probabilistic Dependency Model for Contextual Word Representation

ACL 2023findings

Syntactic structures used to play a vital role in natural language processing (NLP), but since the deep learning revolution, NLP has been gradually dominated by neural models that do not consider syntactic structures in their design. One vastly successful class of neural models is transformers. When…

2023

Recall, Expand, and Multi-Candidate Cross-Encode: Fast and Accurate Ultra-Fine Entity Typing

ACL 2023long

Ultra-fine entity typing (UFET) predicts extremely free-formed types (e.g., president, politician) of a given entity mention (e.g., Joe Biden) in context. State-of-the-art (SOTA) methods use the cross-encoder (CE) based architecture. CE concatenates a mention (and its context) with each type and fee…

2023

Simple Hardware-Efficient PCFGs with Independent Left and Right Productions

EMNLP 2023short findings

Scaling dense PCFGs to thousands of nonterminals via low-rank parameterizations of the rule probability tensor has been shown to be beneficial for unsupervised parsing. However, PCFGs scaled this way still perform poorly as a language model, and even underperform similarly-sized HMMs. This work intr…

Cited by 0SourcecodeScholar
2023

Structured Mean-Field Variational Inference for Higher-Order Span-Based Semantic Role Labeling

ACL 2023findings

In this work, we enhance higher-order graph-based approaches for span-based semantic role labeling (SRL) by means of structured modeling. To decrease the complexity of higher-order modeling, we decompose the edge from predicate word to argument span into three different edges, predicate-to-head (P2H…

Cited by 2SourcePDFScholar
2022

Bottom-Up Constituency Parsing and Nested Named Entity Recognition with Pointer Networks

ACL 2022long

Constituency parsing and nested named entity recognition (NER) are similar tasks since they both aim to predict a collection of nested and non-crossing spans. In this work, we cast nested NER to constituency parsing and propose a novel pointing mechanism for bottom-up parsing to tackle both tasks. T…

2022

Combining (Second-Order) Graph-Based and Headed-Span-Based Projective Dependency Parsing

ACL 2022findings

Graph-based methods, which decompose the score of a dependency tree into scores of dependency arcs, are popular in dependency parsing for decades. Recently, (CITATION) propose a headed-span-based method that decomposes the score of a dependency tree into scores of headed spans. They show improvement…

2022

Dynamic Programming in Rank Space: Scaling Structured Inference with Low-Rank HMMs and PCFGs

NAACL 2022long

Hidden Markov Models (HMMs) and Probabilistic Context-Free Grammars (PCFGs) are widely used structured models, both of which can be represented as factor graph grammars (FGGs), a powerful formalism capable of describing a wide range of models. Recent research found it beneficial to use large state s…

2022

ITA: Image-Text Alignments for Multi-Modal Named Entity Recognition

NAACL 2022long

Recently, Multi-modal Named Entity Recognition (MNER) has attracted a lot of attention. Most of the work utilizes image information through region-level visual representations obtained from a pretrained object detector and relies on an attention mechanism to model the interactions between image and…

2022

Modeling Label Correlations for Ultra-Fine Entity Typing with Neural Pairwise Conditional Random Field

EMNLP 2022main

Ultra-fine entity typing (UFET) aims to predict a wide range of type phrases that correctly describe the categories of a given entity mention in a sentence. Most recent works infer each entity type independently, ignoring the correlations between types, e.g., when an entity is inferred as a presiden…

2022

Named Entity and Relation Extraction with Multi-Modal Retrieval

EMNLP 2022finding

Multi-modal named entity recognition (NER) and relation extraction (RE) aim to leverage relevant image information to improve the performance of NER and RE. Most existing efforts largely focused on directly extracting potentially useful information from images (such as pixel-level features, identifi…

2022

Nested Named Entity Recognition as Latent Lexicalized Constituency Parsing

ACL 2022long

Nested named entity recognition (NER) has been receiving increasing attention. Recently, Fu et al. (2020) adapt a span-based constituency parser to tackle nested NER. They treat nested entities as partially-observed constituency trees and propose the masked inside algorithm for partial marginalizati…

2022

SHARP: Search-Based Adversarial Attack for Structured Prediction

NAACL 2022findings

Adversarial attack of structured prediction models faces various challenges such as the difficulty of perturbing discrete words, the sentence quality issue, and the sensitivity of outputs to small perturbations. In this work, we introduce SHARP, a new attack method that formulates the black-box adve…

2022

Span-Based Semantic Role Labeling with Argument Pruning and Second-Order Inference

AAAI 2022technical

We study graph-based approaches to span-based semantic role labeling. This task is difficult due to the need to enumerate all possible predicate-argument pairs and the high degree of imbalance between positive and negative samples. Based on these difficulties, high-order inference that considers int…

2021

Adapting Unsupervised Syntactic Parsing Methodology for Discourse Dependency Parsing

ACL 2021long

One of the main bottlenecks in developing discourse dependency parsers is the lack of annotated training data. A potential solution is to utilize abundant unlabeled data by using unsupervised techniques, but there is so far little research in unsupervised discourse dependency parsing. Fortunately, u…

2021

Automated Concatenation of Embeddings for Structured Prediction

ACL 2021long

Pretrained contextualized embeddings are powerful word representations for structured prediction tasks. Recent work found that better word representations can be obtained by concatenating different types of embeddings. However, the selection of embeddings to form the best concatenated representation…

2021

Improving Named Entity Recognition by External Context Retrieving and Cooperative Learning

ACL 2021long

Recent advances in Named Entity Recognition (NER) show that document-level contexts can significantly improve model performance. In many application scenarios, however, such contexts are not available. In this paper, we propose to find external contexts of a sentence by retrieving and selecting a se…

2021

Multi-View Cross-Lingual Structured Prediction with Minimum Supervision

ACL 2021long

In structured prediction problems, cross-lingual transfer learning is an efficient way to train quality models for low-resource languages, and further improvement can be obtained by learning from multiple source languages. However, not all source models are created equal and some may hurt performanc…

Cited by 7SourcePDFScholar
2021

PCFGs Can Do Better: Inducing Probabilistic Context-Free Grammars with Many Symbols

NAACL 2021long

Probabilistic context-free grammars (PCFGs) with neural parameterization have been shown to be effective in unsupervised phrase-structure grammar induction. However, due to the cubic computational complexity of PCFG representation and parsing, previous approaches cannot scale up to a relatively larg…

2021

Risk Minimization for Zero-shot Sequence Labeling

ACL 2021long

Zero-shot sequence labeling aims to build a sequence labeler without human-annotated datasets. One straightforward approach is utilizing existing systems (source models) to generate pseudo-labeled datasets and train a target sequence labeler accordingly. However, due to the gap between the source an…

Cited by 3SourcePDFScholar
2021

Structural Knowledge Distillation: Tractably Distilling Information for Structured Predictor

ACL 2021long

Knowledge distillation is a critical technique to transfer knowledge between models, typically from a large model (the teacher) to a more fine-grained one (the student). The objective function of knowledge distillation is typically the cross-entropy between the teacher and the student’s output distr…