← Search

Jingren Zhou

82 accepted papers

2026

AgentFold: Long-Horizon Web Agents with Proactive Context Folding

ICLR 2026poster

LLM-based web agents show immense promise for information seeking, yet their effectiveness on long-horizon tasks is hindered by a fundamental trade-off in context management. Prevailing ReAct-based agents suffer from context saturation as they accumulate noisy, raw histories, while methods that fixe…

Cited by 0SourceScholar
2026

ArenaRL: Scaling RL for Open-Ended Agents via Tournament-based Relative Ranking

ICML 2026poster

Reinforcement learning (RL) has advanced LLM agents on verifiable tasks but remains challenging for open-ended tasks with vast solution spaces (e.g., complex travel planning). Lacking objective ground truth, current RL algorithms rely on reward models assigning scalar scores to individual responses.…

Cited by 0SourceScholar
2026

BOTS: A Unified Framework for Bayesian Online Task Selection in LLM Reinforcement Finetuning

ICLR 2026poster

Reinforcement finetuning (RFT) is a key technique for aligning Large Language Models (LLMs) with human preferences and enhancing reasoning, yet its effectiveness is highly sensitive to which tasks are explored during training. Uniform task sampling is inefficient, wasting computation on tasks that a…

Cited by 0SourceScholar
2026

Clipping Bottleneck: Stabilizing RLVR via Stochastic Recovery of Near-Boundary Signals

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a central paradigm for scaling LLM reasoning, yet its optimization often suffers from training instability and suboptimal convergence. Through a systematic dissection of the GRPO-based objective, we reveal that the rigid clipping d…

Cited by 0SourceScholar
2026

Demystifying Deep Search: A Holistic Evaluation with Hint-free Multi-Hop Questions and Factorised Metrics

ICLR 2026poster

RAG (Retrieval-Augmented Generation) systems and web agents are increasingly evaluated on multi-hop deep search tasks, yet current practice suffers from two major limitations. First, most benchmarks leak the reasoning path in the question text, allowing models to follow surface cues rather than disc…

Cited by 0SourcecodeScholar
2026

Empowering Efficiency and Efficacy in WebAgent via Enabling Info-Rich Seeking

ICLR 2026poster

Large Language Model (LLM)-based agents have emerged as a transformative approach for open-ended problem solving, with information seeking (IS) being a core capability that enables autonomous reasoning and decision-making. While prior research has largely focused on improving retrieval depth, we ob…

Cited by 0SourcecodeScholar
2026

Expanding the Capability Frontier of LLM Agents with ZPD-Guided Data Synthesis

ICLR 2026poster

Unlocking advanced reasoning in large language model agents is hindered by a scarcity of training data situated at the very frontier of their capabilities. We address this with a novel data synthesis approach inspired by the educational theory of the Zone of Proximal Development (ZPD), which concept…

Cited by 0SourceScholar
2026

Experience Augmented Policy Optimization for LLM Reasoning

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is a powerful paradigm for improving the reasoning capabilities of large language models (LLMs). However, existing RLVR methods typically rely on on-policy optimization from scratch, resulting in high sampling costs and inefficient utilization of…

Cited by 0SourceScholar
2026

IterResearch: Rethinking Long-Horizon Agents via Markovian State Reconstruction

ICLR 2026poster

Recent advances in deep-research agents have shown promise for autonomous knowledge construction through dynamic reasoning over external sources. However, existing approaches rely on a mono-contextual paradigm that accumulates all information in a single, expanding context window, leading to context…

Cited by 0SourcecodeScholar
2026

On the Direction of RLVR Updates for LLM Reasoning: Identification and Exploitation

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has substantially improved the reasoning capabilities of large language models. While existing analyses identify that RLVR-induced changes are sparse, they primarily focus on the **magnitude** of these updates, largely overlooking their **direct…

Cited by 0SourcecodeScholar
2026

On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weighting

ICLR 2026poster

Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) are two prominent post-training paradigms for refining the capabilities and aligning the behavior of Large Language Models (LLMs). Existing approaches that integrate SFT and RL often face the risk of disrupting established response pattern…

Cited by 0SourcecodeScholar
2026

One-Way Policy Optimization for Self-Evolving LLMs

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has become a promising paradigm for scaling reasoning capabilities of Large Language Models (LLMs). However, the sparsity of binary verifier rewards often leads to low efficiency and optimization instability. To stabilize training, existing metho…

Cited by 0SourceScholar
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
2026

Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has significantly improved reasoning in large language models (LLMs), yet the token-level mechanisms through which they reshape model behavior remain unclear. We present a systematic empirical study of RLVR’s distributional effects across three c…

Cited by 0SourceScholar
2026

WebSailor-V2: Bridging the Chasm to Proprietary Agents via Synthetic Data and Scalable Reinforcement Learning

ICLR 2026poster

To significantly advance the capabilities of open-source web agents, we present WebSailor-V2, a complete post-training pipeline encompassing data construction, Supervised Fine-Tuning (SFT), and Reinforcement Learning (RL). Our methodology features two key innovations: (1) On the data front, we devel…

Cited by 0SourceScholar
2026

WebShaper: Agentically Data Synthesizing via Information-Seeking Formalization

ICLR 2026poster

The advent of Large Language Model (LLM)-powered agents has revolutionized artificial intelligence by enabling solutions to complex, open-ended tasks through web-based information-seeking (IS) capabilities. The scarcity of high-quality training data has limited the development of IS agents. Existin…

Cited by 0SourcecodeScholar
2026

WebWatcher: Breaking New Frontiers of Vision-Language Deep Research Agent

ICLR 2026poster

Web agents such as deep research have demonstrated superhuman cognitive abilities, capable of solving highly challenging information-seeking problems. However, most research remains largely text-centric, overlooking visual information in the real world. This makes multimodal deep research highly cha…

Cited by 0SourceScholar
2026

WebWeaver: Structuring Web-Scale Evidence with Dynamic Outlines for Open-Ended Deep Research

ICLR 2026poster

This paper tackles \textbf{open-ended deep research (OEDR)}, a complex challenge where AI agents must synthesize vast web-scale information into insightful reports. Current approaches are plagued by dual-fold limitations: static research pipelines that decouple planning from evidence acquisition and…

Cited by 0SourcecodeScholar
2025

ACE: All-round Creator and Editor Following Instructions via Diffusion Transformer

ICLR 2025poster

Diffusion models have emerged as a powerful generative technology and have been found to be applicable in various scenarios. Most existing foundational diffusion models are primarily designed for text-guided visual generation and do not support multi-modal conditions, which are essential for many vi…

Cited by 10SourcePDFScholar
2025

AsymKV: Enabling 1-Bit Quantization of KV Cache with Layer-Wise Asymmetric Quantization Configurations

COLING 2025main

Large language models have shown exceptional capabilities in a wide range of tasks, such as text generation and video generation, among others. However, due to their massive parameter count, these models often require substantial storage space, imposing significant constraints on the machines deploy…

Cited by 6SourcePDFScholar
2025

Benchmarking Multimodal Retrieval Augmented Generation with Dynamic VQA Dataset and Self-adaptive Planning Agent

ICLR 2025poster

Multimodal Retrieval Augmented Generation (mRAG) plays an important role in mitigating the “hallucination” issue inherent in multimodal large language models (MLLMs). Although promising, existing heuristic mRAGs typically predefined fixed retrieval processes, which causes two issues: (1) Non-adaptiv…

2025

Data-Juicer 2.0: Cloud-Scale Adaptive Data Processing for and with Foundation Models

NeurIPS 2025spotlight

Foundation models demand advanced data processing for their vast, multimodal datasets. However, traditional frameworks struggle with the unique complexities of multimodal data. In response, we present Data-Juicer 2.0, a data processing system backed by 100+ data processing operators spanning text, i…

Cited by 0SourcecodeScholar
2025

Data-Juicer Sandbox: A Feedback-Driven Suite for Multimodal Data-Model Co-development

ICML 2025spotlight

The emergence of multimodal large models has advanced artificial intelligence, introducing unprecedented levels of performance and functionality. However, optimizing these models remains challenging due to historically isolated paths of model-centric and data-centric developments, leading to subopti…

Cited by 0SourcePDFScholar
2025

Demons in the Detail: On Implementing Load Balancing Loss for Training Specialized Mixture-of-Expert Models

ACL 2025long

This paper revisits the implementation of Load-Balancing-Loss (LBL) when training Mixture-of-Experts (MoEs) models. Specifically, LBL for MoEs is defined as NE ∑i=1NE fipi, where NE is the total number of experts, fi represents the frequency of expert i being selected, and pi denotes the average gat…

2025

FPE2M2: Approaching Lossless and Efficient Quantization with Native Floating Point

ACL 2025finding

Auto-regressive decoding is a memory-bound job, meaning decoding inference performance is limited by the bandwidth rather than the computational capabilities of the GPU. Weight-only quantization is a promising method to address the memory-bound limitations. Previous studies have followed one of two…

Cited by 0SourcePDFScholar
2025

Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free

NeurIPS 2025oral

Gating mechanisms have been widely utilized, from early models like LSTMs and Highway Networks to recent state space models, linear attention, and also softmax attention. Yet, existing literature rarely examines the specific effects of gating. In this work, we conduct comprehensive experiments to sy…

Cited by 0SourcecodeScholar
2025

GenSim: A General Social Simulation Platform with Large Language Model based Agents

NAACL 2025system demonstrations

With the rapid advancement of large language models (LLMs), recent years have witnessed many promising studies on leveraging LLM-based agents to simulate human social behavior. While prior work has demonstrated significant potential across various domains, much of it has focused on specific scenario…

2025

P-MMEval: A Parallel Multilingual Multitask Benchmark for Consistent Evaluation of LLMs

EMNLP 2025

Recent advancements in large language models (LLMs) showcase varied multilingual capabilities across tasks like translation, code generation, and reasoning. Previous assessments often limited their scope to fundamental natural language processing (NLP) or isolated capability-specific tasks. To allev

2025

PolyMath: Evaluating Mathematical Reasoning in Multilingual Contexts

NeurIPS 2025poster

In this paper, we introduce **PolyMath**, a multilingual mathematical reasoning benchmark covering 18 languages and 4 easy-to-hard difficulty levels. Our benchmark ensures difficulty comprehensiveness, language diversity, and high-quality translation, making it a highly discriminative multilingual m…

Cited by 0SourceScholar
2025

ProcessBench: Identifying Process Errors in Mathematical Reasoning

ACL 2025long

As language models regularly make mistakes when solving math problems, automated identification of errors in the reasoning process becomes increasingly significant for their scalable oversight. In this paper, we introduce ProcessBench for measuring the ability to identify erroneous steps in mathemat…

2025

Provable Scaling Laws for the Test-Time Compute of Large Language Models

NeurIPS 2025poster

We propose two simple, principled and practical algorithms that enjoy provable scaling laws for the test-time compute of large language models (LLMs). The first one is a two-stage knockout-style algorithm: given an input problem, it first generates multiple candidate solutions, and then aggregate th…

Cited by 0SourceScholar
2025

RMTBench: Benchmarking LLMs Through Multi-Turn User-Centric Role-Playing

EMNLP 2025

Recent advancements in Large Language Models (LLMs) have shown outstanding potential for role-playing applications. Evaluating these capabilities is becoming crucial yet remains challenging. Existing benchmarks mostly adopt a character-centric approach, simplify user-character interactions to isolat

Cited by 0SourcePDFScholar
2025

Rotated Runtime Smooth: Training-Free Activation Smoother for accurate INT4 inference

ICLR 2025poster

Large language models have demonstrated promising capabilities upon scaling up parameters. However, serving large language models incurs substantial computation and memory movement costs due to their large scale. Quantization methods have been employed to reduce service costs and latency. Neverthele…

Cited by 0SourcePDFScholar
2025

Self-Steering Optimization: Autonomous Preference Optimization for Large Language Models

ACL 2025finding

The key to effective alignment lies in high-quality preference data. Recent research has focused on automated alignment, which involves developing alignment systems with minimal human intervention. However, prior research has predominantly focused on developing data generation methods, while insuffi…

Cited by 0SourcePDFScholar
2025

Self-play with Execution Feedback: Improving Instruction-following Capabilities of Large Language Models

ICLR 2025spotlight

One core capability of large language models~(LLMs) is to follow natural language instructions. However, the issue of automatically constructing high-quality training data to enhance the complex instruction-following abilities of LLMs without manual annotation remains unresolved. In this paper, we i…

2025

The Lessons of Developing Process Reward Models in Mathematical Reasoning

ACL 2025finding

Process Reward Models (PRMs) aim to identify and mitigate intermediate errors in the reasoning processes in mathematical reasoning of Large Language Models (LLMs).However, the development of effective PRMs faces significant challenges, particularly in data annotation and evaluation methodologies.In…

2025

WebDancer: Towards Autonomous Information Seeking Agency

NeurIPS 2025poster

Addressing intricate real-world problems necessitates in-depth information seeking and multi-step reasoning. Recent progress in agentic systems, exemplified by Deep Research, underscores the potential for autonomous multi-step research. In this work, we present a cohesive paradigm for building end…

Cited by 0SourcecodeScholar
2025

mPLUG-DocOwl2: High-resolution Compressing for OCR-free Multi-page Document Understanding

ACL 2025long

Multimodel Large Language Models(MLLMs) have achieved promising OCR-free Document Understanding performance by increasing the supported resolution of document images. However, this comes at the cost of generating thousands of visual tokens for a single document image, leading to excessive GPU memory…

2025

mPLUG-Owl3: Towards Long Image-Sequence Understanding in Multi-Modal Large Language Models

ICLR 2025poster

Multi-modal Large Language Models have demonstrated remarkable capabilities in executing instructions for a variety of single-image tasks. Despite this progress, significant challenges remain in modeling long image sequences. In this work, we introduce the versatile multi-modal large language model,…

2024

#InsTag: Instruction Tagging for Analyzing Supervised Fine-tuning of Large Language Models

ICLR 2024poster

Pre-trained large language models (LLMs) can understand and align with human instructions by supervised fine-tuning (SFT). It is commonly believed that diverse and complex SFT data are of the essence to enable good instruction-following abilities. However, such diversity and complexity are obscure a…

2024

AIR-Bench: Benchmarking Large Audio-Language Models via Generative Comprehension

ACL 2024long

Recently, instruction-following audio-language models have received broad attention for human-audio interaction. However, the absence of benchmarks capable of evaluating audio-centric interaction capabilities has impeded advancements in this field. Previous models primarily focus on assessing differ…

2024

DreamVideo: Composing Your Dream Videos with Customized Subject and Motion

CVPR 2024poster

Customized generation using diffusion models has made impressive progress in image generation but remains unsatisfactory in the challenging video generation task as it requires the controllability of both subjects and motions. To that end we present DreamVideo a novel approach to generating personal…

2024

EE-LLM: Large-Scale Training and Inference of Early-Exit Large Language Models with 3D Parallelism

ICML 2024poster

We present EE-LLM, a framework for large-scale training and inference of early-exit large language models (LLMs). While recent works have shown preliminary evidence for the efficacy of early exiting in accelerating LLM inference, EE-LLM makes a foundational step towards scaling up early-exit LLMs by…

2024

How Abilities in Large Language Models are Affected by Supervised Fine-tuning Data Composition

ACL 2024long

Large language models (LLMs) with enormous pre-training tokens and parameters emerge diverse abilities, including math reasoning, codegeneration, and instruction following. These abilities are further enhanced by supervised fine-tuning (SFT). While the open-source community has explored ad-hoc SFT f…

2024

Large Language Models are Superpositions of All Characters: Attaining Arbitrary Role-play via Self-Alignment

ACL 2024long

Considerable efforts have been invested in augmenting the role-playing proficiency of open-source large language models (LLMs) by emulating proprietary counterparts. Nevertheless, we posit that LLMs inherently harbor role-play capabilities, owing to the extensive knowledge of characters and potentia…

2024

Lipschitz Singularities in Diffusion Models

ICLR 2024oral

Diffusion models, which employ stochastic differential equations to sample images through integrals, have emerged as a dominant class of generative models. However, the rationality of the diffusion process itself receives limited attention, leaving the question of whether the problem is well-posed a…

Cited by 10SourcePDFScholar
2024

Predicting Rewards Alongside Tokens: Non-disruptive Parameter Insertion for Efficient Inference Intervention in Large Language Model

EMNLP 2024main

Transformer-based large language models (LLMs) exhibit limitations such as generating unsafe responses, unreliable reasoning, etc. Existing inference intervention approaches attempt to mitigate these issues by finetuning additional models to produce calibration signals (such as rewards) that guide t…

2024

Ranni: Taming Text-to-Image Diffusion for Accurate Instruction Following

CVPR 2024poster

Existing text-to-image (T2I) diffusion models usually struggle in interpreting complex prompts especially those with quantity object-attribute binding and multi-subject descriptions. In this work we introduce a semantic panel as the middleware in decoding texts to images supporting the generator to…

Cited by 48SourcePDFScholar
2024

Routing to the Expert: Efficient Reward-guided Ensemble of Large Language Models

NAACL 2024long

The complementary potential of Large Language Models (LLM) assumes off-the-shelf LLMs have heterogeneous expertise in a wide range of domains and tasks so that an ensemble of LLMs can achieve consistently better performance. Existing ensemble methods for LLMs mainly focus on reward model ranking of…

Cited by 83SourcePDFScholar
2024

mPLUG-DocOwl 1.5: Unified Structure Learning for OCR-free Document Understanding

EMNLP 2024finding

Structure information is critical for understanding the semantics of text-rich images, such as documents, tables, and charts. Existing Multimodal Large Language Models (MLLMs) for Visual Document Understanding are equipped with text recognition ability but lack general structure understanding abilit…

2023

CancerUniT: Towards a Single Unified Model for Effective Detection, Segmentation, and Diagnosis of Eight Major Cancers Using a Large Collection of CT Scans

ICCV 2023poster

Human readers or radiologists routinely perform full-body multi-organ multi-disease detection and diagnosis in clinical practice, while most medical AI systems are built to focus on single organs with a narrow list of a few diseases. This might severely limit AI's clinical adoption. A certain number…

Cited by 12PDFScholar
2023

Composer: Creative and Controllable Image Synthesis with Composable Conditions

ICML 2023poster

Recent large-scale generative models learned on big data are capable of synthesizing incredible images yet suffer from limited controllability. This work offers a new generation paradigm that allows flexible control of the output image, such as spatial layout and palette, while maintaining the synth…

2023

Cones: Concept Neurons in Diffusion Models for Customized Generation

ICML 2023oral

Human brains respond to semantic features of presented stimuli with different neurons. This raises the question of whether deep neural networks admit a similar behavior pattern. To investigate this phenomenon, this paper identifies a small cluster of neurons associated with a specific subject in a d…

Cited by 119SourcePDFScholar
2023

Customizable Image Synthesis with Multiple Subjects

NeurIPS 2023poster

Synthesizing images with user-specified subjects has received growing attention due to its practical applications. Despite the recent success in single subject customization, existing algorithms suffer from high training cost and low success rate along with increased number of subjects. Towards cont…

Cited by 84SourcePDFScholar
2023

Devil Is in the Queries: Advancing Mask Transformers for Real-World Medical Image Segmentation and Out-of-Distribution Localization

CVPR 2023highlight

Real-world medical image segmentation has tremendous long-tailed complexity of objects, among which tail conditions correlate with relatively rare diseases and are clinically significant. A trustworthy medical AI algorithm should demonstrate its effectiveness on tail conditions to avoid clinically d…

Cited by 28SourcePDFScholar
2023

Dimensionality-Varying Diffusion Process

CVPR 2023poster

Diffusion models, which learn to reverse a signal destruction process to generate new data, typically require the signal at each step to have the same dimension. We argue that, considering the spatial redundancy in image signals, there is no need to maintain a high dimensionality in the evolution pr…

2023

FaceComposer: A Unified Model for Versatile Facial Content Creation

NeurIPS 2023poster

This work presents FaceComposer, a unified generative model that accomplishes a variety of facial content creation tasks, including text-conditioned face synthesis, text-guided face editing, face animation etc. Based on the latent diffusion framework, FaceComposer follows the paradigm of composition…

Cited by 8SourcePDFScholar
2023

LipFormer: High-Fidelity and Generalizable Talking Face Generation With a Pre-Learned Facial Codebook

CVPR 2023poster

Generating a talking face video from the input audio sequence is a practical yet challenging task. Most existing methods either fail to capture fine facial details or need to train a specific model for each identity. We argue that a codebook pre-learned on high-quality face images can serve as a use…

Cited by 36SourcePDFScholar
2023

Neural Dependencies Emerging From Learning Massive Categories

CVPR 2023poster

This work presents two astonishing findings on neural networks learned for large-scale image classification. 1) Given a well-trained model, the logits predicted for some category can be directly obtained by linearly combining the predictions of a few other categories, which we call neural dependency…

2023

PASS: Patch Automatic Skip Scheme for Efficient Real-Time Video Perception on Edge Devices

AAAI 2023technical

Real-time video perception tasks are often challenging over the resource-constrained edge devices due to the concerns of accuracy drop and hardware overhead, where saving computations is the key to performance improvement. Existing methods either rely on domain-specific neural chips or priorly searc…

Cited by 2SourcePDFScholar
2023

RA-CLIP: Retrieval Augmented Contrastive Language-Image Pre-Training

CVPR 2023poster

Contrastive Language-Image Pre-training (CLIP) is attracting increasing attention for its impressive zero-shot recognition performance on different down-stream tasks. However, training CLIP is data-hungry and requires lots of image-text pairs to memorize various semantic concepts. In this paper, we…

Cited by 38SourcePDFScholar
2023

RLEG: Vision-Language Representation Learning with Diffusion-based Embedding Generation

ICML 2023poster

Vision-language representation learning models (e.g., CLIP) have achieved state-of-the-art performance on various downstream tasks, which usually need large-scale training data to learn discriminative representation. Recent progress on generative diffusion models (e.g., DALL-E 2) has demonstrated th…

Cited by 12SourcePDFScholar
2023

Res-Tuning: A Flexible and Efficient Tuning Paradigm via Unbinding Tuner from Backbone

NeurIPS 2023poster

Parameter-efficient tuning has become a trend in transferring large-scale foundation models to downstream applications. Existing methods typically embed some light-weight tuners into the backbone, where both the design and the learning of the tuners are highly dependent on the base model. This work…

2023

ViM: Vision Middleware for Unified Downstream Transferring

ICCV 2023poster

Foundation models are pre-trained on massive data and transferred to downstream tasks via fine-tuning. This work presents Vision Middleware (ViM), a new learning paradigm that targets unified transferring from a single foundation model to a variety of downstream tasks. ViM consists of a zoo of light…

Cited by 1PDFScholar
2023

VideoComposer: Compositional Video Synthesis with Motion Controllability

NeurIPS 2023poster

The pursuit of controllability as a higher standard of visual content creation has yielded remarkable progress in customizable image synthesis. However, achieving controllable video synthesis remains challenging due to the large variation of temporal dynamics and the requirement of cross-frame tempo…

2023

mPLUG-2: A Modularized Multi-modal Foundation Model Across Text, Image and Video

ICML 2023poster

Recent years have witnessed a big convergence of language, vision, and multi-modal pretraining. In this work, we present mPLUG-2, a new unified paradigm with modularized design for multi-modal pretraining, which can benefit from modality collaboration while addressing the problem of modality entangl…

2022

OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework

ICML 2022spotlight

In this work, we pursue a unified paradigm for multimodal pretraining to break the shackles of complex task/modality-specific customization. We propose OFA, a Task-Agnostic and Modality-Agnostic framework that supports Task Comprehensiveness. OFA unifies a diverse set of cross-modal and unimodal tas…

2022

Principled Knowledge Extrapolation with GANs

ICML 2022spotlight

Human can extrapolate well, generalize daily knowledge into unseen scenarios, raise and answer counterfactual questions. To imitate this ability via generative models, previous works have extensively studied explicitly encoding Structural Causal Models (SCMs) into architectures of generator networks…

2022

Reliable Adversarial Distillation with Unreliable Teachers

ICLR 2022poster

In ordinary distillation, student networks are trained with soft labels (SLs) given by pretrained teacher networks, and students are expected to improve upon teachers since SLs are stronger supervision than the original hard labels. However, when considering adversarial robustness, teachers may beco…

2022

iFlood: A Stable and Effective Regularizer

ICLR 2022poster

Various regularization methods have been designed to prevent overfitting of machine learning models. Among them, a surprisingly simple yet effective one, called Flooding, is proposed recently, which directly constrains the training loss on average to stay at a given level. However, our further studi…

Cited by 5SourcePDFScholar
2022

mPLUG: Effective and Efficient Vision-Language Learning by Cross-modal Skip-connections

EMNLP 2022main

Large-scale pre-trained foundation models have been an emerging paradigm for building artificial intelligence (AI) systems, which can be quickly adapted to a wide range of downstream tasks. This paper presents mPLUG, a new vision-language foundation model for both cross-modal understanding and gener…

2021

Dynamic Memory based Attention Network for Sequential Recommendation

AAAI 2021technical

Sequential recommendation has become increasingly essential in various online services. It aims to model the dynamic preferences of users from their historical interactions and predict their next items. The accumulated user behavior records on real systems could be very long. This rich data brings o…

Cited by 77SourcePDFScholar
2021

Learning Relation Alignment for Calibrated Cross-modal Retrieval

ACL 2021long

Despite the achievements of large-scale multimodal pre-training approaches, cross-modal retrieval, e.g., image-text retrieval, remains a challenging task. To bridge the semantic gap between the two modalities, previous studies mainly focus on word-region alignment at the object level, lacking the ma…

2021

Learning to Rehearse in Long Sequence Memorization

ICML 2021spotlight

Existing reasoning tasks often have an important assumption that the input contents can be always accessed while reasoning, requiring unlimited storage resources and suffering from severe time delay on long sequences. To achieve efficient reasoning on long sequences with limited storage resources, m…

Cited by 12SourcePDFScholar
2021

Low-Rank Subspaces in GANs

NeurIPS 2021poster

The latent space of a Generative Adversarial Network (GAN) has been shown to encode rich semantics within some subspaces. To identify these subspaces, researchers typically analyze the statistical information from a collection of synthesized data, and the identified subspaces tend to control image a…

2021

UFC-BERT: Unifying Multi-Modal Controls for Conditional Image Synthesis

NeurIPS 2021poster

Conditional image synthesis aims to create an image according to some multi-modal guidance in the forms of textual descriptions, reference images, and image blocks to preserve, as well as their combinations. In this paper, instead of investigating these control signals separately, we propose a new t…

Cited by 77SourcePDFScholar
2021

Uncertainty Principles of Encoding GANs

ICML 2021spotlight

The compelling synthesis results of Generative Adversarial Networks (GANs) demonstrate rich semantic knowledge in their latent codes. To obtain this knowledge for downstream applications, encoding GANs has been proposed to learn encoders, such that real world data can be encoded to latent codes, whi…

Cited by 8SourcePDFScholar
2020

AdaBERT: Task-Adaptive BERT Compression with Differentiable Neural Architecture Search

IJCAI 2020poster

Large pre-trained language models such as BERT have shown their effectiveness in various natural language processing tasks. However, the huge parameter size makes them difficult to be deployed in real-time applications that require quick inference with limited resources. Existing methods compress BE…

Cited by 0SourcePDFScholar
2020

Learning Efficient Parameter Server Synchronization Policies for Distributed SGD

ICLR 2020poster

We apply a reinforcement learning (RL) based approach to learning optimal synchronization policies used for Parameter Server-based distributed training of machine learning models with Stochastic Gradient Descent (SGD). Utilizing a formal synchronization policy description in the PS-setting, we are a…

Cited by 10SourceScholar
2020

Learning to Mutate with Hypergradient Guided Population

NeurIPS 2020poster

Computing the gradient of model hyperparameters, i.e., hypergradient, enables a promising and natural way to solve the hyperparameter optimization task. However, gradient-based methods could lead to suboptimal solutions due to the non-convex nature of optimization in a complex hyperparameter space.…