← Search

Ying Wen

50 accepted papers

2026

$G^2$-Reader: Dual Evolving Graphs for Multimodal Document QA

ICML 2026poster

Retrieval-augmented generation is a practical paradigm for question answering over long documents, but it remains brittle for multimodal reading where text, tables, and figures are interleaved across many pages. First, flat chunking breaks document-native structure and cross-modal alignment, yieldin…

Cited by 1SourceScholar
2026

DGS: Dual Gradient and Semantic-Shift Guided Low-Rank Adaptation for Class Incremental Learning

CVPR 2026

In Class-Incremental Learning (CIL), parameter efficient fine-tuning applied to Pre-trained Models (PTMs) remain vulnerable to catastrophic forgetting as they adapt to new tasks. The prevalent strategy to mitigate catastrophic forgetting is to constrain gradients within the orthogonal subspaces of p

Cited by 0SourceScholar
2026

Learning to Reason in Structured In-context Environments with Reinforcement Learning

ICLR 2026poster

Large language models (LLMs) have achieved significant advancements in reasoning capabilities through reinforcement learning (RL) via environmental exploration. As the intrinsic properties of the environment determine the abilities that LLMs can learn, the environment plays a important role in the R…

Cited by 0SourceScholar
2026

M-IDoL: Information Decomposition for Modality-Specific and Diverse Representation Learning in Medical Foundation Model

ICML 2026poster

Medical foundation models (MFMs) aim to learn universal representations from multimodal medical images that can generalize effectively to diverse downstream clinical tasks. However, most existing MFMs suffer from information ambiguity that blend multimodal representations in a single embedding space…

Cited by 0SourceScholar
2026

MAGIC: A Co-Evolving Attacker–Defender Adversarial Game for Robust LLM Safety

ICML 2026poster

Ensuring robust safety alignment is crucial for Large Language Models (LLMs), yet existing defenses often lag behind evolving adversarial attacks due to their \textbf{reliance on static, pre-collected data distributions}. In this paper, we introduce \textbf{MAGIC}, a novel multi-turn multi-agent rei…

Cited by 0SourceScholar
2026

SafeLab: An Interactive High-Fidelity Benchmark for Embodied Safety in Scientific Robotics

ICML 2026poster

Laboratory automation driven by scientific embodied agents represents a critical frontier in modern laboratories. Unlike conventional robotic domains, laboratory environments impose zero-tolerance constraints on manipulation precision and collision, as minor deviations can lead to irreversible chemi…

Cited by 0SourceScholar
2026

Towards Cold-Start Drafting and Continual Refining: A Value-Driven Memory Approach with Application to NPU Kernel Synthesis

ICML 2026poster

Deploying Large Language Models to data-scarce programming domains poses significant challenges, particularly for kernel synthesis on emerging Domain-Specific Architectures where a "Data Wall" limits available training data. While models excel on data-rich platforms like CUDA, they suffer catastroph…

Cited by 0SourceScholar
2026

When Token Pruning is Worse than Random: Understanding Visual Token Information in VLLMs

CVPR 2026

Vision Large Language Models (VLLMs) incur high computational costs due to their reliance on hundreds of visual tokens to represent images. While token pruning offers a promising solution for accelerating inference, this paper, however, identifies a key observation: in deeper layers (e.g., beyond th

Cited by 0SourcecodeScholar
2025

AFiRe: Anatomy-Driven Self-Supervised Learning for Fine-Grained Representation in Radiographic Images

AAAI 2025technical

Current self-supervised methods, such as contrastive learning, predominantly focus on global discrimination, neglecting the critical fine-grained anatomical details required for accurate radiographic analysis. To address this challenge, we propose the Anatomy-driven self-supervised framework for enh…

2025

AT-Drone: Benchmarking Adaptive Teaming in Multi-Drone Pursuit

CoRL 2025poster

Adaptive teaming—the capability of agents to effectively collaborate with unfamiliar teammates without prior coordination—is widely explored in virtual video games but overlooked in real-world multi-robot contexts. Yet, such adaptive collaboration is crucial for real-world applications, including bo…

Cited by 0SourceScholar
2025

CoSMIC: Continual Self-supervised Learning for Multi-Domain Medical Imaging via Conditional Mutual Information Maximization

ICCV 2025poster

Medical foundation models, pre-trained on diverse data sources, have shown significant potential for multi-domain medical imaging tasks.However, the domain shifts across different anatomical types significantly hinder their performance compared to domain-specific models.To address this challenge, we…

Cited by 0SourcePDFScholar
2025

Leveraging Dual Process Theory in Language Agent Framework for Real-time Simultaneous Human-AI Collaboration

ACL 2025long

Agents built on large language models (LLMs) have excelled in turn-by-turn human-AI collaboration but struggle with simultaneous tasks requiring real-time interaction. Latency issues and the challenge of inferring variable human strategies hinder their ability to make autonomous decisions without ex…

2025

M²RL-Net: Multi-View and Multi-Level Relation Learning Network for Weakly-Supervised Image Forgery Detection

AAAI 2025technical

As digital media manipulation becomes increasingly sophisticated, accurately detecting and localizing image forgeries with minimal supervision has become a critical challenge. Existing weakly supervised image forgery detection (W-IFD) methods often rely on convolutional neural networks (CNNs) and l…

Cited by 0SourcePDFScholar
2025

RAT: Adversarial Attacks on Deep Reinforcement Agents for Targeted Behaviors

AAAI 2025technical

Evaluating deep reinforcement learning (DRL) agents against targeted behavior attacks is critical for assessing their robustness. These attacks aim to manipulate the victim into specific behaviors that align with the attacker’s objectives, often bypassing traditional reward-based defenses. Prior met…

2025

ReMA: Learning to Meta-Think for LLMs with Multi-agent Reinforcement Learning

NeurIPS 2025poster

Recent research on Reasoning of Large Language Models (LLMs) has sought to further enhance their performance by integrating meta-thinking—enabling models to monitor, evaluate, and control their reasoning processes for more adaptive and effective problem-solving. However, current single-agent work la…

Cited by 0SourcecodeScholar
2025

Retrieval-Augmented Process Reward Model for Generalizable Mathematical Reasoning

ACL 2025finding

While large language models (LLMs) have significantly advanced mathematical reasoning, Process Reward Models (PRMs) have been developed to evaluate the logical validity of reasoning steps. However, PRMs still struggle with out-of-distribution (OOD) challenges. This paper identifies the OOD issues in…

2025

STAR: Efficient Preference-based Reinforcement Learning via Dual Regularization

NeurIPS 2025poster

Preference-based reinforcement learning (PbRL) bypasses complex reward engineering by learning from human feedback. However, due to the high cost of obtaining feedback, PbRL typically relies on a limited set of preference-labeled samples. This data scarcity introduces two key inefficiencies: (1) the…

Cited by 0SourceScholar
2025

ThinkBench: Dynamic Out-of-Distribution Evaluation for Robust LLM Reasoning

NeurIPS 2025poster

Evaluating large language models (LLMs) poses significant challenges, particularly due to issues of data contamination and the leakage of correct answers. To address these challenges, we introduce ThinkBench, a novel evaluation framework designed to robustly evaluate the reasoning capability of LLMs…

Cited by 0SourcecodeScholar
2024

Aligning Individual and Collective Objectives in Multi-Agent Cooperation

NeurIPS 2024poster

Among the research topics in multi-agent learning, mixed-motive cooperation is one of the most prominent challenges, primarily due to the mismatch between individual and collective goals. The cutting-edge research is focused on incorporating domain knowledge into rewards and introducing additional m…

Cited by 1SourcePDFScholar
2024

AlphaZero-Like Tree-Search can Guide Large Language Model Decoding and Training

ICML 2024poster

Recent works like Tree-of-Thought (ToT) and Reasoning via Planning (RAP) aim to augment the multi-step reasoning capabilities of LLMs by using tree-search algorithms. These methods rely on prompting a pre-trained model to serve as a value function and focus on problems with low search depth. As a re…

2024

Critic-Guided Decision Transformer for Offline Reinforcement Learning

AAAI 2024technical

Recent advancements in offline reinforcement learning (RL) have underscored the capabilities of Return-Conditioned Supervised Learning (RCSL), a paradigm that learns the action distribution based on target returns for each state in a supervised manner. However, prevailing RCSL methods largely focus…

2024

DS-Agent: Automated Data Science by Empowering Large Language Models with Case-Based Reasoning

ICML 2024poster

In this work, we investigate the potential of large language models (LLMs) based agents to automate data science tasks, with the goal of comprehending task requirements, then building and training the best-fit machine learning models. Despite their widespread success, existing LLM agents are hindere…

2024

EAN: An Efficient Attention Module Guided by Normalization for Deep Neural Networks

AAAI 2024technical

Deep neural networks (DNNs) have achieved remarkable success in various fields, and two powerful techniques, feature normalization and attention mechanisms, have been widely used to enhance model performance. However, they are usually considered as two separate approaches or combined in a simplistic…

Cited by 3SourcePDFScholar
2024

IIRP-Net: Iterative Inference Residual Pyramid Network for Enhanced Image Registration

CVPR 2024poster

Deep learning-based image registration (DLIR) methods have achieved remarkable success in deformable image registration. We observe that iterative inference can exploit the well-trained registration network to the fullest extent. In this work we propose a novel Iterative Inference Residual Pyramid N…

2024

Reinforcing LLM Agents via Policy Optimization with Action Decomposition

NeurIPS 2024poster

Language models as intelligent agents push the boundaries of sequential decision-making agents but struggle with limited knowledge of environmental dynamics and exponentially huge action space. Recent efforts like GLAM and TWOSOME manually constrain the action space to a restricted subset and employ…

2024

ZSC-Eval: An Evaluation Toolkit and Benchmark for Multi-agent Zero-shot Coordination

NeurIPS 2024poster

Zero-shot coordination (ZSC) is a new cooperative multi-agent reinforcement learning (MARL) challenge that aims to train an ego agent to work with diverse, unseen partners during deployment. The significant difference between the deployment-time partners' distribution and the training partners' dist…

2023

Cooperative Open-ended Learning Framework for Zero-Shot Coordination

ICML 2023poster

Zero-shot coordination in cooperative artificial intelligence (AI) remains a significant challenge, which means effectively coordinating with a wide range of unseen partners. Previous algorithms have attempted to address this challenge by optimizing fixed objectives within a population to improve st…

Cited by 30SourcePDFScholar
2023

GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models

ICML 2023poster

This paper introduces a distributed, GPU-centric experience replay system, GEAR, designed to perform scalable reinforcement learning (RL) with large sequence models (such as transformers). With such models, existing systems such as Reverb face considerable bottlenecks in memory, computation, and com…

2023

Order Matters: Agent-by-agent Policy Optimization

ICLR 2023poster

While multi-agent trust region algorithms have achieved great success empirically in solving coordination tasks, most of them, however, suffer from a non-stationarity problem since agents update their policies simultaneously. In contrast, a sequential scheme that updates policies agent-by-agent pro…

2022

Cross-Utterance Conditioned VAE for Non-Autoregressive Text-to-Speech

ACL 2022long

Modelling prosody variation is critical for synthesizing natural and expressive speech in end-to-end text-to-speech (TTS) systems. In this paper, a cross-utterance conditional VAE (CUC-VAE) is proposed to estimate a posterior probability distribution of the latent prosody features for each phoneme b…

2022

Greedy when Sure and Conservative when Uncertain about the Opponents

ICML 2022spotlight

We develop a new approach, named Greedy when Sure and Conservative when Uncertain (GSCU), to competing online against unknown and nonstationary opponents. GSCU improves in four aspects: 1) introduces a novel way of learning opponent policy embeddings offline; 2) trains offline a single best response…

2022

Multi-Agent Reinforcement Learning is a Sequence Modeling Problem

NeurIPS 2022accept

Large sequence models (SM) such as GPT series and BERT have displayed outstanding performance and generalization capabilities in natural language process, vision and recently reinforcement learning. A natural follow-up question is how to abstract multi-agent decision making also as an sequence model…

2022

Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning

ICLR 2022poster

Trust region methods rigorously enabled reinforcement learning (RL) agents to learn monotonically improving policies, leading to superior performance on a variety of tasks. Unfortunately, when it comes to multi-agent reinforcement learning (MARL), the property of monotonic improvement may not simpl…

Cited by 329SourcePDFScholar
2022

Unsupervised Hierarchical Translation-Based Model for Multi-Modal Medical Image Registration

ICASSP 2022accepted

Deformable registration of multi-modal medical images is a challenging task in medical image processing due to the differences in both appearance and structure. We propose an unsupervised hierarchical translation-based model to perform a coarse to fine registration of multi-modal medical images. The…

Cited by 0SourceScholar
2021

Learning in Nonzero-Sum Stochastic Games with Potentials

ICML 2021spotlight

Multi-agent reinforcement learning (MARL) has become effective in tackling discrete cooperative game scenarios. However, MARL has yet to penetrate settings beyond those modelled by team and zero-sum games, confining it to a small subset of multi-agent systems. In this paper, we introduce a new gener…

Cited by 59SourcePDFScholar
2021

Modelling Behavioural Diversity for Learning in Open-Ended Games

ICML 2021oral

Promoting behavioural diversity is critical for solving games with non-transitive dynamics where strategic cycles exist, and there is no consistent winner (e.g., Rock-Paper-Scissors). Yet, there is a lack of rigorous treatment for defining diversity and constructing diversity-aware learning dynamics…

Cited by 79SourcePDFScholar
2021

Neural Auto-Curricula in Two-Player Zero-Sum Games

NeurIPS 2021poster

When solving two-player zero-sum games, multi-agent reinforcement learning (MARL) algorithms often create populations of agents where, at each iteration, a new agent is discovered as the best response to a mixture over the opponent population. Within such a process, the update rules of "who to compe…

2021

Towards Unifying Behavioral and Response Diversity for Open-ended Learning in Zero-sum Games

NeurIPS 2021poster

Measuring and promoting policy diversity is critical for solving games with strong non-transitive dynamics where strategic cycles exist, and there is no consistent winner (e.g., Rock-Paper-Scissors). With that in mind, maintaining a pool of diverse policies via open-ended learning is an attractive s…

2020

Modelling Bounded Rationality in Multi-Agent Interactions by Generalized Recursive Reasoning

IJCAI 2020poster

Though limited in real-world decision making, most multi-agent reinforcement learning (MARL) models assume perfectly rational agents -- a property hardly met due to individual's cognitive limitation and/or the tractability of the decision problem. In this paper, we introduce generalized recursive…

2020

Multi-Agent Determinantal Q-Learning

ICML 2020poster

Centralized training with decentralized execution has become an important paradigm in multi-agent learning. Though practical, current methods rely on restrictive assumptions to decompose the centralized value function across agents for execution. In this paper, we eliminate this restriction by propo…

2020

SMARTS: An Open-Source Scalable Multi-Agent RL Training School for Autonomous Driving

CoRL 2020

Interaction is fundamental in autonomous driving (AD). Despite more than a decade of intensive R&D in AD, how to dynamically interact with diverse road users in various contexts still remains unsolved. Multi-agent learning has recently seen big breakthroughs and has much to offer towards solving rea

2019

A Novel Fractional Order Derivate Based Log-demons with Driving Force for High Accurate Image Registration

ICASSP 2019accepted

Image registration methods based on Thirion's demons method update displacement field by the image gradient obtained by integer order derivate. However, the fractional order derivate is superior to integral order derivate for computing image gradient under weak texture or smooth regions. To obtain h…

Cited by 0SourceScholar
2019

Probabilistic Recursive Reasoning for Multi-Agent Reinforcement Learning

ICLR 2019poster

Humans are capable of attributing latent mental contents such as beliefs, or intentions to others. The social skill is critical in everyday life to reason about the potential consequences of their behaviors so as to plan ahead. It is known that humans use this reasoning ability recursively, i.e. con…

Cited by 196SourcePDFScholar
2018

Sparse Low-Rank Component Coding for Face Recognition with Illumination And Corruption

ICASSP 2018accepted

Sparse representation-based classification shows a good performance for face recognition in recent years, but it can not be suitable for face recognition with illumination and corruption, which are often presented in the practical applications. To solve the problem, in this paper, we propose a novel…

Cited by 0SourceScholar
2017

A novel dictionary based SRC for face recognition

ICASSP 2017accepted

The sparse representation based classification (SRC) performs not very well for small sample data. A discriminative common vector dictionary based SRC is introduced in this paper to address this issue. The contribution of this paper is that the dictionary of the proposed method is constructed by the…

Cited by 0SourceScholar