← Search

Yang Wei

29 accepted papers

2026

Clear Nights Ahead: Towards Multi-Weather Nighttime Image Restoration

AAAI 2026technical

Restoring nighttime images affected by multiple adverse weather conditions is a practical yet under-explored research problem, as multiple weather degradations usually coexist in the real world alongside various lighting effects at night. This paper first explores the challenging multi-weather night

Cited by 0SourcePDFScholar
2026

Inter-Client Dependency Recovery with Hidden Global Components for Federated Traffic Prediction

AAAI 2026technical

Traffic prediction plays an important role in urban management. However, existing methods rely on centralized traffic data, which may raise privacy concerns. Federated traffic prediction offers a promising solution for clients (e.g., traffic management administrations) in different regions to collab

Cited by 0SourcePDFScholar
2026

Well Begun, Half Done: Reinforcement Learning with Prefix Optimization for LLM Reasoning

AAAI 2026technical

Reinforcement Learning with Verifiable Rewards (RLVR) significantly enhances the reasoning capability of Large Language Models (LLMs). Current RLVR approaches typically conduct training across all generated tokens, but neglect to explore which tokens (e.g., prefix tokens) actually contribute to reas

Cited by 0SourcePDFScholar
2025

Boosting Consistency in Story Visualization with Rich-Contextual Conditional Diffusion Models

AAAI 2025technical

Recent research showcases the considerable potential of conditional diffusion models for generating consistent stories. However, current methods, which primarily generate stories in a caption-dependent manner, often overlook the importance of contextual consistency and the relevance of frames durin…

2025

Breaking Grid Constraints: Dynamic Graph Reconstruction Network for Multi-organ Segmentation

ICCV 2025poster

Morphological differences and dense spatial relations of organs make multi-organ segmentation challenging. Current segmentation networks, primarily based on CNNs and Transformers, represent organs by aggregating information within fixed regions. However, aggregated representations often fail to accu…

2025

Covert and Potent: A Weather-Camouflaged Backdoor Attacks on Self-Supervised Learning

ICASSP 2025accepted

Self-supervised learning is widely applied across various domains due to its advantage of learning data representations without the need for labels. However, recent research shows that backdoor attacks on self-supervised learning are achievable by coupling benign features with trigger features witho…

Cited by 0SourceScholar
2025

Diverse Policies Recovering via Pointwise Mutual Information Weighted Imitation Learning

ICLR 2025poster

Recovering a spectrum of diverse policies from a set of expert trajectories is an important research topic in imitation learning. After determining a latent style for a trajectory, previous diverse polices recovering methods usually employ a vanilla behavioral cloning learning objective conditioned…

Cited by 0SourcePDFScholar
2025

Power of Diversity: Enhancing Data-Free Black-Box Attack with Domain-Augmented Learning

AAAI 2025technical

Substitute training-based data-free black-box attacks pose a significant threat to enterprise-deployed models. These attacks use a generator to synthesize data and query APIs, then train a substitute model to approximate the target model's decision boundary based on the returned results. However, ex…

Cited by 0SourcePDFScholar
2025

Who Controls the Authorization? Invertible Networks for Copyright Protection in Text-to-Image Synthesis

ICCV 2025poster

To defend against personalized generation, a new form of infringement that is more concealed and destructive, the existing copyright protection methods is to add adversarial perturbations in images. However, these methods focus solely on countering illegal personalization, neglecting the requirement…

Cited by 0SourcePDFScholar
2024

Advancing Pose-Guided Image Synthesis with Progressive Conditional Diffusion Models

ICLR 2024poster

Recent work has showcased the significant potential of diffusion models in pose-guided person image synthesis. However, owing to the inconsistency in pose between the source and target images, synthesizing an image with a distinct pose, relying exclusively on the source image and target pose informa…

2024

Enhancing Human Experience in Human-Agent Collaboration: A Human-Centered Modeling Approach Based on Positive Human Gain

ICLR 2024poster

Existing game AI research mainly focuses on enhancing agents' abilities to win games, but this does not inherently make humans have a better experience when collaborating with these agents. For example, agents may dominate the collaboration and exhibit unintended or detrimental behaviors, leading to…

Cited by 2SourcePDFScholar
2024

Make Pixels Dance: High-Dynamic Video Generation

CVPR 2024poster

Creating high-dynamic videos such as motion-rich actions and sophisticated visual effects poses a significant challenge in the field of artificial intelligence. Unfortunately current state-of-the-art video generation methods primarily focusing on text-to-video generation tend to produce video clips…

Cited by 102SourcePDFScholar
2024

SoftDedup: an Efficient Data Reweighting Method for Speeding Up Language Model Pre-training

ACL 2024long

The effectiveness of large language models (LLMs) is often hindered by duplicated data in their extensive pre-training datasets. Current approaches primarily focus on detecting and removing duplicates, which risks the loss of valuable information and neglects the varying degrees of duplication. To a…

Cited by 2SourcePDFScholar
2024

VersVideo: Leveraging Enhanced Temporal Diffusion Models for Versatile Video Generation

ICLR 2024poster

Creating stable, controllable videos is a complex task due to the need for significant variation in temporal dynamics and cross-frame temporal consistency. To address this, we enhance the spatial-temporal capability and introduce a versatile video generation model, VersVideo, which leverages textual…

2024

What Matters in Training a GPT4-Style Language Model with Multimodal Inputs?

NAACL 2024long

Recent advancements in GPT-4V have displayed remarkable multi-modal capabilities in processing image inputs and following open-ended instructions. Despite these advancements, there is considerable scope for enhancing open-source multi-modal LLMs, especially in terms of multi-modal understanding accu…

2023

A Robust and Opponent-Aware League Training Method for StarCraft II

NeurIPS 2023spotlight

It is extremely difficult to train a superhuman Artificial Intelligence (AI) for games of similar size to StarCraft II. AlphaStar is the first AI that beat human professionals in the full game of StarCraft II, using a league training framework that is inspired by a game-theoretic approach. In this p…

Cited by 12SourcePDFScholar
2023

Act As You Wish: Fine-Grained Control of Motion Diffusion Model with Hierarchical Semantic Graphs

NeurIPS 2023poster

Most text-driven human motion generation methods employ sequential modeling approaches, e.g., transformer, to extract sentence-level text representations automatically and implicitly for human motion synthesis. However, these compact text representations may overemphasize the action names at the exp…

2023

Future-conditioned Unsupervised Pretraining for Decision Transformer

ICML 2023poster

Recent research in offline reinforcement learning (RL) has demonstrated that return-conditioned supervised learning is a powerful paradigm for decision-making problems. While promising, return conditioning is limited to training data labeled with rewards and therefore faces challenges in learning fr…

2023

Hokoff: Real Game Dataset from Honor of Kings and its Offline Reinforcement Learning Benchmarks

NeurIPS 2023poster

The advancement of Offline Reinforcement Learning (RL) and Offline Multi-Agent Reinforcement Learning (MARL) critically depends on the availability of high-quality, pre-collected offline datasets that represent real-world complexities and practical applications. However, existing datasets often fall…

2023

Policy Space Diversity for Non-Transitive Games

NeurIPS 2023poster

Policy-Space Response Oracles (PSRO) is an influential algorithm framework for approximating a Nash Equilibrium (NE) in multi-agent non-transitive games. Many previous studies have been trying to promote policy diversity in PSRO. A major weakness with existing diversity metrics is that a more divers…

Cited by 18SourcePDFScholar
2023

Quality-Similar Diversity via Population Based Reinforcement Learning

ICLR 2023poster

Diversity is a growing research topic in Reinforcement Learning (RL). Previous research on diversity has mainly focused on promoting diversity to encourage exploration and thereby improve quality (the cumulative reward), maximizing diversity subject to quality constraints, or jointly maximizing qual…

Cited by 22SourcePDFScholar
2023

Towards Effective and Interpretable Human-Agent Collaboration in MOBA Games: A Communication Perspective

ICLR 2023top-25%

MOBA games, e.g., Dota2 and Honor of Kings, have been actively used as the testbed for the recent AI research on games, and various AI systems have been developed at the human level so far. However, these AI systems mainly focus on how to compete with humans, less on exploring how to collaborate wit…

Cited by 9SourcePDFScholar
2022

Actor-Critic Policy Optimization in a Large-Scale Imperfect-Information Game

ICLR 2022poster

The deep policy gradient method has demonstrated promising results in many large-scale games, where the agent learns purely from its own experience. Yet, policy gradient methods with self-play suffer convergence problems to a Nash Equilibrium (NE) in multi-agent situations. Counterfactual regret min…

Cited by 33SourcePDFScholar
2022

Honor of Kings Arena: an Environment for Generalization in Competitive Reinforcement Learning

NeurIPS 2022accept

This paper introduces Honor of Kings Arena, a reinforcement learning (RL) environment based on the Honor of Kings, one of the world’s most popular games at present. Compared to other environments studied in most previous work, ours presents new generalization challenges for competitive reinforcement…

2022

SCL-WC: Cross-Slide Contrastive Learning for Weakly-Supervised Whole-Slide Image Classification

NeurIPS 2022accept

Weakly-supervised whole-slide image (WSI) classification (WSWC) is a challenging task where a large number of unlabeled patches (instances) exist within each WSI (bag) while only a slide label is given. Despite recent progress for the multiple instance learning (MIL)-based WSI analysis, the major l…

2021

Learning Diverse Policies in MOBA Games via Macro-Goals

NeurIPS 2021poster

Recently, many researchers have made successful progress in building the AI systems for MOBA-game-playing with deep reinforcement learning, such as on Dota 2 and Honor of Kings. Even though these AI systems have achieved or even exceeded human-level performance, they still suffer from the lack of po…

Cited by 12SourcePDFScholar
2021

LightSeq: A High Performance Inference Library for Transformers

NAACL 2021industry

Transformer and its variants have achieved great success in natural language processing. Since Transformer models are huge in size, serving these models is a challenge for real industrial applications. In this paper, we propose , a highly efficient inference library for models in the Transformer fam…