← Search

Zhilin Yang

28 accepted papers

2026

Kimi-Dev: Agentless Training as Skill Prior for SWE-agents

ICLR 2026poster

Large Language Models (LLMs) are increasingly applied to software engineering (SWE), with SWE-bench as a key benchmark. Solutions are split into SWE-Agent frameworks with multi-turn interactions and workflow-based Agentless methods with single-turn verifiable steps. We argue these paradigms are not…

Cited by 0SourcecodeScholar
2025

Learning to Plan Before Answering: Self-Teaching LLMs to Learn Abstract Plans for Problem Solving

ICLR 2025poster

In the field of large language model (LLM) post-training, the effectiveness of utilizing synthetic data generated by the LLM itself has been well-presented. However, a key question remains unaddressed: what essential information should such self-generated data encapsulate? Existing approaches only p…

Cited by 0SourcePDFScholar
2025

MoBA: Mixture of Block Attention for Long-Context LLMs

NeurIPS 2025spotlight

Scaling the effective context length is essential for advancing large language models (LLMs) toward artificial general intelligence (AGI). However, the quadratic increase in computational complexity inherent in traditional attention mechanisms presents a prohibitive overhead. Existing approaches eit…

Cited by 0SourcecodeScholar
2025

OpenCUA: Open Foundations for Computer-Use Agents

NeurIPS 2025spotlight

Vision-language models have demonstrated impressive capabilities as computer-use agents (CUAs) capable of automating diverse computer tasks. As their commercial potential grows, critical details of the most capable CUA systems remain closed. As these agents will increasingly mediate digital interact…

Cited by 0SourceScholar
2023

A Universal Discriminator for Zero-Shot Generalization

ACL 2023long

Generative modeling has been the dominant approach for large-scale pretraining and zero-shot generalization. In this work, we challenge this convention by showing that discriminative approaches perform substantially better than generative ones on a large number of NLP tasks. Technically, we train a…

2023

Compositional Task Representations for Large Language Models

ICLR 2023poster

Large language models have shown a remarkable cross-task generalization ability. Most prior work assumed that prompts effectively extract knowledge from language models to facilitate generalization to new tasks. This perspective led to numerous studies on improving prompts. In contrast, we introduce…

Cited by 21SourcePDFScholar
2023

Not All Tasks Are Born Equal: Understanding Zero-Shot Generalization

ICLR 2023top-25%

Recent work has achieved remarkable zero-shot performance with multi-task prompted pretraining, but little has been understood. For the first time, we show that training on a small number of key tasks beats using all the training tasks, while removing these key tasks substantially hurts performance.…

Cited by 14SourcePDFScholar
2022

FewNLU: Benchmarking State-of-the-Art Methods for Few-Shot Natural Language Understanding

ACL 2022long

The few-shot natural language understanding (NLU) task has attracted much recent attention. However, prior methods have been evaluated under a disparate set of protocols, which hinders fair comparison and measuring the progress of the field. To address this issue, we introduce an evaluation framewor…

2022

FlipDA: Effective and Robust Data Augmentation for Few-Shot Learning

ACL 2022long

Most previous methods for text data augmentation are limited to simple tasks and weak baselines. We explore data augmentation on hard tasks (i.e., few-shot natural language understanding) and strong baselines (i.e., pretrained models with over one billion parameters). Under this setting, we reproduc…

2022

GLM: General Language Model Pretraining with Autoregressive Blank Infilling

ACL 2022long

There have been various types of pretraining architectures including autoencoding models (e.g., BERT), autoregressive models (e.g., GPT), and encoder-decoder models (e.g., T5). However, none of the pretraining frameworks performs the best for all tasks of three main categories including natural lang…

2022

GPS: Genetic Prompt Search for Efficient Few-Shot Learning

EMNLP 2022main

Prompt-based techniques have demostrated great potential for improving the few-shot generalization of pretrained language models. However, their performance heavily relies on the manual design of prompts and thus requiring a lot of human efforts. In this paper, we introduce Genetic Prompt Search (GP…

2022

Learning to Detect Noisy Labels Using Model-Based Features

EMNLP 2022finding

Label noise is ubiquitous in various machine learning scenarios such as self-labeling with model predictions and erroneous data annotation. Many existing approaches are based on heuristics such as sample losses, which might not be flexible enough to achieve optimal solutions. Meta learning based met…

2022

NLP From Scratch Without Large-Scale Pretraining: A Simple and Efficient Framework

ICML 2022spotlight

Pretrained language models have become the standard approach for many NLP tasks due to strong performance, but they are very expensive to train. We propose a simple and efficient learning framework, TLM, that does not rely on large-scale pretraining. Given some labeled task data and a large general…

2022

P-Tuning: Prompt Tuning Can Be Comparable to Fine-tuning Across Scales and Tasks

ACL 2022short

Prompt tuning, which only tunes continuous prompts with a frozen language model, substantially reduces per-task storage and memory usage at training. However, in the context of NLU, prior work reveals that prompt tuning does not perform well for normal-sized pretrained models. We also find that exis…

Cited by 1517SourcePDFScholar
2022

ZeroPrompt: Scaling Prompt-Based Pretraining to 1,000 Tasks Improves Zero-Shot Generalization

EMNLP 2022finding

We propose a multitask pretraining approach ZeroPrompt for zero-shot generalization, focusing on task scaling and zero-shot prompting.While previous models are trained on only a few dozen tasks, we scale to 1,000 tasks for the first time using real-world data. This leads to a crucial discovery that…

Cited by 28SourcePDFScholar
2019

Mixtape: Breaking the Softmax Bottleneck Efficiently

NeurIPS 2019poster

The softmax bottleneck has been shown to limit the expressiveness of neural lan- guage models. Mixture of Softmaxes (MoS) is an effective approach to address such a theoretical limitation, but are expensive compared to softmax in terms of both memory and time. We propose Mixtape, an output layer tha…

Cited by 16SourcePDFScholar
2019

XLNet: Generalized Autoregressive Pretraining for Language Understanding

NeurIPS 2019oral

With the capability of modeling bidirectional contexts, denoising autoencoding based pretraining like BERT achieves better performance than pretraining approaches based on autoregressive language modeling. However, relying on corrupting the input with masks, BERT neglects dependency between the mask…

2018

Breaking the Softmax Bottleneck: A High-Rank RNN Language Model

ICLR 2018oral

We formulate language modeling as a matrix factorization problem, and show that the expressiveness of Softmax-based models (including the majority of neural language models) is limited by a Softmax bottleneck. Given that natural language is highly context-dependent, this further implies that in prac…

2018

GLoMo: Unsupervised Learning of Transferable Relational Graphs

NeurIPS 2018poster

Modern deep transfer learning approaches have mainly focused on learning generic feature vectors from one task that are transferable to other tasks, such as word embeddings in language and pretrained convolutional features in vision. However, these approaches usually transfer unary features and larg…

Cited by 27SourcePDFScholar
2018

Mastering the Dungeon: Grounded Language Learning by Mechanical Turker Descent

ICLR 2018poster

Contrary to most natural language processing research, which makes use of static datasets, humans learn language interactively, grounded in an environment. In this work we propose an interactive learning procedure called Mechanical Turker Descent (MTD) that trains agents to execute natural language…

Cited by 32SourcePDFScholar
2017

Differentiable Learning of Logical Rules for Knowledge Base Reasoning

NeurIPS 2017poster

We study the problem of learning probabilistic first-order logical rules for knowledge base reasoning. This learning problem is difficult because it requires learning the parameters in a continuous space as well as the structure in a discrete space. We propose a framework, Neural Logic Programming,…

2017

Good Semi-supervised Learning That Requires a Bad GAN

NeurIPS 2017poster

Semi-supervised learning methods based on generative adversarial networks (GANs) obtained strong empirical results, but it is not clear 1) how the discriminator benefits from joint training with a generator, and 2) why good semi-supervised classification performance and a good generator cannot be ob…

2017

Transfer Learning for Sequence Tagging with Hierarchical Recurrent Networks

ICLR 2017poster

Recent papers have shown that neural networks obtain state-of-the-art performance on several different sequence tagging tasks. One appealing property of such systems is their generality, as excellent performance can be achieved with a unified architecture and without task-specific feature engineerin…

Cited by 432SourcecodeScholar
2017

Words or Characters? Fine-grained Gating for Reading Comprehension

ICLR 2017poster

Previous work combines word-level and character-level representations using concatenation or scalar weighting, which is suboptimal for high-level tasks like reading comprehension. We present a fine-grained gating mechanism to dynamically combine word-level and character-level representations based o…

Cited by 100SourcecodeScholar
2016

Review Networks for Caption Generation

NeurIPS 2016poster

We propose a novel extension of the encoder-decoder framework, called a review network. The review network is generic and can enhance any existing encoder- decoder model: in this paper, we consider RNN decoders with both CNN and RNN encoders. The review network performs a number of review steps with…

2016

Revisiting Semi-Supervised Learning with Graph Embeddings

ICML 2016poster

We present a semi-supervised learning framework based on graph embeddings. Given a graph between instances, we train an embedding for each instance to jointly predict the class label and the neighborhood context in the graph. We develop both transductive and inductive variants of our method. In the…