← Search

Jieyu Zhang

46 accepted papers

2026

CoAct-1: Computer-using Multi-agent System with Coding Actions

ICLR 2026poster

Autonomous agents that operate computers via Graphical User Interfaces (GUIs) often struggle with efficiency and reliability on complex, long-horizon tasks. While augmenting these agents with planners can improve task decomposition, they remain constrained by the inherent limitations of performing a…

Cited by 0SourcecodeScholar
2026

Generate Any Scene: Scene Graph Driven Data Synthesis for Visual Generation Training

ICLR 2026poster

Recent advances in text-to-vision generation excel in visual fidelity but struggle with compositional generalization and semantic alignment. Existing datasets are noisy and weakly compositional, limiting models' understanding of complex scenes, while scalable solutions for dense, high-quality annota…

Cited by 0SourcecodeScholar
2026

Molmo2: Open Weights and Data for Vision-Language Models with Video Understanding and Grounding

CVPR 2026

Today's strongest video-language models (VLMs) remain proprietary, and the strongest open-weight models often rely on synthetic data from proprietary VLMs and do not disclose their training data or recipe. As a result, the open-source community lacks the foundations needed to improve on the state-of

Cited by 0SourcecodeScholar
2026

MolmoAct: Action Reasoning Models That Can Reason in Space

ICRA 2026poster

Reasoning is essential for purposeful action, yet most robotic foundation models map perception and instructions directly to control, limiting adaptability, generalization, and semantic grounding. We introduce Action Reasoning Models (ARMs), which integrate perception, planning, and control through …

2026

Nemotron-Research-Tool-N1: Exploring Tool-Using Language Models with Reinforced Reasoning

ICLR 2026poster

Enabling large language models with external tools has become a pivotal strategy for extending their functionality beyond text space. To enhance LLMs' tool-calling abilities, previous approaches primarily rely on supervised fine-tuning (SFT) with trajectories distilled from stronger models, often re…

Cited by 0SourcecodeScholar
2026

Position: Digital Agents Require Unified Agent-Native Environments

ICML 2026poster

Large language models (LLMs) are increasingly deployed as digital agents that perform multi-step digital work on a computer, but the environments in which they operate remain fragmented and task-specific. Our position is that digital agents need Agent-Native Computer: interfaces that expose system c…

Cited by 0SourceScholar
2026

SAGE: Training Smart Any-Horizon Agents for Long Video Reasoning with Reinforcement Learning

CVPR 2026

As humans, we are natural any-horizon reasoners, i.e., we can decide whether to iteratively skim long videos or watch short ones in full when necessary for a given task. With this in mind, one would expect video reasoning models to reason flexibly across different durations. However, SOTA models are

Cited by 0SourcecodeScholar
2026

Synthetic Object Compositions for Scalable and Accurate Learning in Detection, Segmentation, and Grounding

CVPR 2026

Visual grouping--operationalized through tasks such as instance segmentation, visual grounding, and object detection--enables applications ranging from robotic perception to photo editing. These fundamental problems in computer vision are powered by large-scale, painstakingly annotated datasets. Des

Cited by 0SourceScholar
2026

Theory of Space: Can Foundation Models Construct Spatial Beliefs through Active Exploration?

ICLR 2026poster

Spatial embodied intelligence often operates under partial observability, where agents must act to acquire missing information rather than passively consume complete observations. In such settings, progress depends on actively selecting informative actions that reduce uncertainty and support the con…

Cited by 0SourcecodeScholar
2026

TrajTok: Learning Trajectory Tokens Enhances Video Understanding

CVPR 2026

Tokenization in video models, typically through patchification, generates an excessive and redundant number of tokens. This severely limits video efficiency and scalability. While the recent trajectory-based tokenizers offer a promising solution by decoupling video duration from token count, they re

Cited by 0SourcecodeScholar
2026

TrustGen: A Platform of Dynamic Benchmarking on the Trustworthiness of Generative Foundation Models

ICLR 2026poster

Generative foundation models (GenFMs), such as large language models and text-to-image systems, have demonstrated remarkable capabilities in various downstream applications. As they are increasingly deployed in high-stakes applications, assessing their trustworthiness has become both a critical nece…

Cited by 0SourceScholar
2026

Understanding VLMs Spatial Mental Modeling Capability from Limited Views

ICLR 2026poster

Can Vision Language Models (VLMs) imagine the full scene from just a few views, like humans do? Humans form spatial mental models, internal representations of unseen space, to reason about layout, perspective, and motion. Our new MindCube benchmark with 21,154 questions across 3,268 images exposes t…

Cited by 0SourcecodeScholar
2025

Explaining Length Bias in LLM-Based Preference Evaluations

EMNLP 2025

The use of large language models (LLMs) as judges, particularly in preference comparisons, has become widespread, but this reveals a notable bias towards longer responses, undermining the reliability of such evaluations. To better understand such bias, we propose to decompose the preference evaluati

Cited by 0SourcePDFScholar
2025

LATTE: Learning to Think with Vision Specialists

EMNLP 2025

While open-source vision-language models perform well on simple question-answering, they still struggle with complex questions that require both perceptual and reasoning capabilities. We propose LATTE, a family of vision-language models that have LeArned to Think wiTh vision spEcialists. By offloadi

2025

One Trajectory, One Token: Grounded Video Tokenization via Panoptic Sub-object Trajectory

ICCV 2025poster

Effective video tokenization is critical for scaling transformer models for long videos. Current approaches tokenize videos using space-time patches, leading to excessive tokens and computational inefficiencies. The best token reduction strategies degrade performance and barely reduce the number of…

Cited by 0SourcePDFScholar
2025

URDF-Anything: Constructing Articulated Objects with 3D Multimodal Language Model

NeurIPS 2025spotlight

Constructing accurate digital twins of articulated objects is essential for robotic simulation training and embodied AI world model building, yet historically requires painstaking manual modeling or multi-stage pipelines. In this work, we propose \textbf{URDF-Anything}, an end-to-end automatic recon…

Cited by 0SourceScholar
2025

Which Agent Causes Task Failures and When? On Automated Failure Attribution of LLM Multi-Agent Systems

ICML 2025spotlight

Failure attribution in LLM multi-agent systems—identifying the agent and step responsible for task failures—provides crucial clues for systems debugging but remains underexplored and labor-intensive. In this paper, we propose and formulate a new research area: automated failure attribution for LLM…

2024

Biomedical Visual Instruction Tuning with Clinician Preference Alignment

NeurIPS 2024poster

Recent advancements in multimodal foundation models have showcased impressive capabilities in understanding and reasoning with visual and textual information. Adapting these foundation models trained for general usage to specialized domains like biomedicine requires large-scale domain-specific instr…

2024

DataComp-LM: In search of the next generation of training sets for language models

NeurIPS 2024poster

We introduce DataComp for Language Models, a testbed for controlled dataset experiments with the goal of improving language models. As part of DCLM, we provide a standardized corpus of 240T tokens extracted from Common Crawl, effective pretraining recipes based on the OpenLM framework, and a broad s…

Cited by 64SourcePDFScholar
2024

EHRAgent: Code Empowers Large Language Models for Few-shot Complex Tabular Reasoning on Electronic Health Records

EMNLP 2024main

Clinicians often rely on data engineers to retrieve complex patient information from electronic health record (EHR) systems, a process that is both inefficient and time-consuming. We propose EHRAgent, a large language model (LLM) agent empowered with accumulative domain knowledge and robust coding c…

2024

Iterated Learning Improves Compositionality in Large Vision-Language Models

CVPR 2024poster

A fundamental characteristic common to both human vision and natural language is their compositional nature. Yet despite the performance gains contributed by large vision and language pretraining recent investigations find that most--if not all--our state-of-the-art vision-language models struggle a…

Cited by 18SourcePDFScholar
2024

Offline Training of Language Model Agents with Functions as Learnable Weights

ICML 2024poster

Researchers and practitioners have recently reframed powerful Large Language Models (LLMs) as *agents*, enabling them to automate complex tasks largely via the use of specialized functions. To facilitate the development of LLM agents, we present a novel paradigm of training LLM agents without modify…

Cited by 16SourcePDFScholar
2024

SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models

ICML 2024poster

Most existing Large Language Model (LLM) benchmarks on scientific problem reasoning focus on problems grounded in high-school subjects and are confined to elementary algebraic operations. To systematically examine the reasoning capabilities required for solving complex scientific problems, we introd…

2024

m&m’s: A Benchmark to Evaluate Tool-Use for multi-step multi-modal Tasks

ECCV 2024poster

"Real-world multi-modal problems are rarely solved by a single machine learning model, and often require multi-step computational plans that involve stitching several models. Tool-augmented LLMs hold tremendous promise for automating the generation of such computational plans. However, the lack of s…

2023

Characterizing the Impacts of Semi-supervised Learning for Weak Supervision

NeurIPS 2023poster

Labeling training data is a critical and expensive step in producing high accuracy ML models, whether training from scratch or fine-tuning. To make labeling more efficient, two major approaches are programmatic weak supervision (WS) and semi-supervised learning (SSL). More recent works have either…

Cited by 8SourcePDFScholar
2023

Cold-Start Data Selection for Better Few-shot Language Model Fine-tuning: A Prompt-based Uncertainty Propagation Approach

ACL 2023long

We present PATRON, a prompt-based data selection method for pre-trained language model fine-tuning under cold-start scenarios, i.e., no initial labeled data are available. In PATRON, we design (1) a prompt-based uncertainty propagation approach to estimate the importance of data points and (2) a par…

2023

DataComp: In search of the next generation of multimodal datasets

NeurIPS 2023oral

Multimodal datasets are a critical component in recent breakthroughs such as CLIP, Stable Diffusion and GPT-4, yet their design does not receive the same research attention as model architectures or training algorithms. To address this shortcoming in the machine learning ecosystem, we introduce Data…

2023

Frustratingly Easy Regularization on Representation Can Boost Deep Reinforcement Learning

CVPR 2023poster

Deep reinforcement learning (DRL) gives the promise that an agent learns good policy from high-dimensional information, whereas representation learning removes irrelevant and redundant information and retains pertinent information. In this work, we demonstrate that the learned representation of the…

2023

Large Language Model as Attributed Training Data Generator: A Tale of Diversity and Bias

NeurIPS 2023poster

Large language models (LLMs) have been recently leveraged as training data generators for various natural language processing (NLP) tasks. While previous research has explored different approaches to training models using generated data, they generally rely on simple class-conditional prompts, which…

2023

Learning Hyper Label Model for Programmatic Weak Supervision

ICLR 2023poster

To reduce the human annotation efforts, the programmatic weak supervision (PWS) paradigm abstracts weak supervision sources as labeling functions (LFs) and involves a label model to aggregate the output of multiple LFs to produce training labels. Most existing label models require a parameter learni…

2023

Leveraging Instance Features for Label Aggregation in Programmatic Weak Supervision

AISTATS 2023poster

Programmatic Weak Supervision (PWS) has emerged as a widespread paradigm to synthesize training labels efficiently. The core component of PWS is the label model, which infers true labels by aggregating the outputs of multiple noisy supervision sources abstracted as labeling functions (LFs). Existing…

2023

On the Trade-off of Intra-/Inter-class Diversity for Supervised Pre-training

NeurIPS 2023poster

Pre-training datasets are critical for building state-of-the-art machine learning models, motivating rigorous study on their impact on downstream tasks. In this work, we study the impact of the trade-off between the intra-class diversity (the number of samples per class) and the inter-class diversit…

Cited by 11SourcePDFScholar
2023

Subclass-balancing Contrastive Learning for Long-tailed Recognition

ICCV 2023poster

Long-tailed recognition with imbalanced class distribution naturally emerges in practical machine learning applications. Existing methods such as data reweighing, resampling, and supervised contrastive learning enforce the class balance with a price of introducing imbalance between instances of head…

Cited by 32PDFcodeScholar
2023

SugarCrepe: Fixing Hackable Benchmarks for Vision-Language Compositionality

NeurIPS 2023poster

In the last year alone, a surge of new benchmarks to measure $\textit{compositional}$ understanding of vision-language models have permeated the machine learning ecosystem. Given an image, these benchmarks probe a model's ability to identify its associated caption amongst a set of compositional dist…

2023

Uncovering Neural Scaling Laws in Molecular Representation Learning

NeurIPS 2023poster

Molecular Representation Learning (MRL) has emerged as a powerful tool for drug and materials discovery in a variety of tasks such as virtual screening and inverse design. While there has been a surge of interest in advancing model-centric techniques, the influence of both data quantity and quality…

Cited by 20SourcePDFScholar
2022

AcTune: Uncertainty-Based Active Self-Training for Active Fine-Tuning of Pretrained Language Models

NAACL 2022long

Although fine-tuning pre-trained language models (PLMs) renders strong performance in many NLP tasks, it relies on excessive labeled data. Recently, researchers have resorted to active fine-tuning for enhancing the label efficiency of PLM fine-tuning, but existing methods of this type usually ignore…

2022

Adaptive Ranking-based Sample Selection for Weakly Supervised Class-imbalanced Text Classification

EMNLP 2022finding

To obtain a large amount of training labels inexpensively, researchers have recently adopted the weak supervision (WS) paradigm, which leverages labeling rules to synthesize training labels rather than using individual annotations to achieve competitive results for natural language processing (NLP)…

2022

Creating Training Sets via Weak Indirect Supervision

ICLR 2022poster

Creating labeled training sets has become one of the major roadblocks in machine learning. To address this, recent Weak Supervision (WS) frameworks synthesize training labels from multiple potentially noisy supervision sources. However, existing frameworks are restricted to supervision sources that…

Cited by 19SourcePDFScholar
2022

Understanding Programmatic Weak Supervision via Source-aware Influence Function

NeurIPS 2022accept

Programmatic Weak Supervision (PWS) aggregates the source votes of multiple weak supervision sources into probabilistic training labels, which are in turn used to train an end model. With its increasing popularity, it is critical to have some tool for users to understand the influence of each compon…

Cited by 15SourcePDFScholar
2021

Optimizing Information-theoretical Generalization Bound via Anisotropic Noise of SGLD

NeurIPS 2021poster

Recently, the information-theoretical framework has been proven to be able to obtain non-vacuous generalization bounds for large models trained by Stochastic Gradient Langevin Dynamics (SGLD) with isotropic noise. In this paper, we optimize the information-theoretical generalization bound by manipu…

Cited by 8SourcePDFScholar
2021

TAXOGAN: Hierarchical Network Representation Learning via Taxonomy Guided Generative Adversarial Networks (Extended Abstract)

IJCAI 2021poster

Network representation learning aims at transferring node proximity in networks into distributed vectors, which can be leveraged in various downstream applications. Recent research has shown that nodes in a network can often be organized in latent hierarchical structures, but without a particular un…

Cited by 0SourcePDFScholar
2021

Taxonomy Completion via Triplet Matching Network

AAAI 2021technical

Automatically constructing taxonomy finds many applications in e-commerce and web search. One critical challenge is as data and business scope grow in real applications, new concepts are emerging and needed to be added to the existing taxonomy. Previous approaches focus on the taxonomy expansion, i.…

2021

WRENCH: A Comprehensive Benchmark for Weak Supervision

NeurIPS 2021poster

Recent Weak Supervision (WS) approaches have had widespread success in easing the bottleneck of labeling training data for machine learning by synthesizing labels from multiple potentially noisy supervision sources. However, proper measurement and analysis of these approaches remain a challenge. Fi…

Cited by 119SourcecodeScholar