← Search

Yang Gao

228 accepted papers

2026

AGiLe: Learning Robust Long-Horizon Manipulation via Affordance-Grounded Bidirectional Latent Planning

CVPR 2026

The robust execution of long-horizon manipulation tasks remains a central challenge in embodied intelligence, necessitating both coherent high-level planning and reliable low-level control. Existing approaches often encounter two critical limitations: the accumulation of prediction errors in subgoal

Cited by 0SourcecodeScholar
2026

ARTIFREE: DETECTING AND REDUCING GENERATIVE ARTIFACTS IN DIFFUSION-BASED SPEECH ENHANCEMENT

ICASSP 2026poster

Diffusion-based speech enhancement (SE) achieves natural-sounding speech and strong generalization, yet suffers from key limitations like generative artifacts and high inference latency. In this work, we systematically study artifact prediction and reduction in diffusion-based SE. We show that varia…

Cited by 0SourcePDFScholar
2026

Best of Sim and Real: Decoupled Visuomotor Manipulation Via Learning Control in Simulation and Perception in Real

ICRA 2026poster

Sim-to-real transfer remains a fundamental challenge in robot manipulation due to the entanglement of perception and control in end-to-end learning. We present a decoupled framework that learns each component where it is most reliable: control policies are trained in simulation with privileged state…

2026

Beyond Text-to-SQL: Can LLMs Really Debug Enterprise ETL SQL?

ICML 2026poster

SQL is central to enterprise data engineering, yet generating fully correct SQL code in a single attempt remains difficult—even for experienced developers and advanced \ttsql LLMs—often requiring multiple debugging iterations. We introduce \textbf{\ourbench}, the first benchmark for enterprise-level…

Cited by 0SourceScholar
2026

CatalystBench: A Comprehensive Multi-Task Benchmark for Advancing Language Models in Catalysis Science

ICLR 2026poster

The discovery of novel catalytic materials is a cornerstone of chemical engineering and sustainable energy, yet it remains a complex, knowledge-intensive process. While Large Language Models (LLMs) have demonstrated remarkable potential in various scientific domains, their application to catalysis i…

Cited by 0SourceScholar
2026

Causality-Aware Efficient Exploration for Cooperative Multi-Agent Reinforcement Learning

AAAI 2026technical

Exploration is critical for cooperative multi agent reinforcement learning (MARL) to improve sample efficiency. However, existing intrinsic motivation based exploration strategies in MARL overlook the causal relationships among agents, global states, and rewards, suffering from interference by irrel

Cited by 0SourcePDFScholar
2026

DART: Navigating Last-Mile Heterogeneity in Instant Delivery via Distribution-Adaptive Splines

IJCAI 2026

On-demand delivery platforms rely on Travel Time Estimation (TTE) to balance courier earnings and overdue risks. In collaboration with one of China's largest platforms, we address a critical "Fairness Gap" in TTE: current systems fail to capture complex delivery patterns in GNSS-denied environments,

Cited by 0Scholar
2026

DeCo: Task Decomposition and Skill Composition for Zero-Shot Generalization in Long-Horizon 3D Manipulation

RA-L 2026

Generalizing language-conditioned multi-task imitation learning (IL) models to novel long-horizon 3D manipulation tasks is challenging. To address this, we propose <bold xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">DeCo</b> (<italic xmlns:mml="http://www.

Cited by 11SourcecodeScholar
2026

Deformable Gaussian Occupancy: Decoupling Rigid and Nonrigid Motion with Factorized Distillation

CVPR 2026

Understanding dynamic 3D environments is essential for safe autonomous driving, particularly when reasoning about human-centric, nonrigid agents. However, existing weakly supervised occupancy prediction frameworks predominantly assume rigid-body motion and rely on simple frame-to-frame offsets, limi

Cited by 0SourcecodeScholar
2026

Don't Forget Why You Started: Tackling Dual Forgetting in Vision-Language Continual Learning

ICML 2026poster

Continual learning of Vision-Language Model (VLM) aspires to empower foundation models with new expertise without compromising their universal zero-shot capabilities. However, this pursuit faces a critical ''dual-forgetting'' challenge: the catastrophic forgetting of newly acquired classes (Incremen…

Cited by 0SourceScholar
2026

Faster Parameter-Free Regret Matching Algorithms

ICLR 2026poster

Regret Matching (RM) and its variants are widely employed to learn a Nash equilibrium (NE) in large-scale games. However, most existing research only establishes a theoretical convergence rate of $O(1/\sqrt{T})$ for these algorithms in learning an NE. Recent studies have shown that smooth RM$^+$ var…

Cited by 0SourceScholar
2026

Identifying and Analyzing Performance-Critical Tokens in Large Language Models

AAAI 2026technical

In-context learning (ICL) has emerged as an effective solution for few-shot learning with large language models (LLMs). However, how LLMs leverage demonstrations to specify a task and learn a corresponding computational function through ICL is underexplored. Drawing from the way humans learn from c

Cited by 0SourcePDFScholar
2026

ImgCoT: Compressing Long Chain of Thought into Compact Visual Tokens for Efficient Reasoning of Large Language Model

ICML 2026poster

Compressing long chains of thought (CoT) into compact latent tokens is crucial for efficient reasoning with large language models (LLMs). Recent studies employ autoencoders to achieve this by reconstructing textual CoT from latent tokens, thus encoding CoT semantics. However, treating textual CoT as…

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

LaViRA: Language-Vision-Robot Actions Translation for Zero-Shot Vision Language Navigation in Continuous Environments

ICRA 2026poster

Zero-shot Vision-and-Language Navigation in Continuous Environments (VLN-CE) requires an agent to navigate unseen environments based on natural language instructions without any prior training. Current methods face a critical trade-off: either rely on environment-specific waypoint predictors that li…

2026

Learning Brain Representation with Hierarchical Visual Embeddings

ICLR 2026poster

Decoding visual representations from brain signals has attracted significant attention in both neuroscience and artificial intelligence. However, the degree to which brain signals truly encode visual information remains unclear. Current visual decoding approaches explore various brain–image alignmen…

Cited by 0SourceScholar
2026

Learning Native Continuation for Action Chunking Flow Policies

RSS 2026poster

Action chunking enables Vision Language Action (VLA) models to run in real time, but naive chunked execution often exhibits discontinuities at chunk boundaries. Real-Time Chunking (RTC) alleviates this issue but is external to the policy, leading to spurious multimodal switching and trajectories tha…

Cited by 0SourceScholar
2026

Learning to Reconfigure: Co-designing Reconfigurable robots for Heterogeneous Locomotion

ICML 2026poster

Traditional robot co-design approaches typically converge to \textit{one} configuration, which do not explore the flexibility from reconfiguration on heterogeneous environments. On the other hand, existing designs for reconfigurable robots require human-designed configurations. We present Learning t…

Cited by 0SourceScholar
2026

Lightweight and Generalizable Acoustic Scene Representations via Contrastive Fine-Tuning and Distillation

ICASSP 2026poster

Acoustic scene classification (ASC) models on edge devices typically operate under fixed class assumptions, lacking the transferability needed for real-world applications that require adaptation to new or refined acoustic categories. We propose ContrastASC, which learns generalizable acoustic scene…

Cited by 0SourcePDFScholar
2026

ManiLong-Shot: Interaction-Aware One-Shot Imitation Learning for Long-Horizon Manipulation

AAAI 2026technical

One-shot imitation learning (OSIL) offers a promising way to teach robots new skills without large-scale data collection. However, current OSIL methods are primarily limited to short-horizon tasks, thus limiting their applicability to complex, long-horizon manipulations. To address this limitation,

Cited by 0SourcePDFScholar
2026

MotionTrans: Human VR Data Enable Motion-Level Learning for Robotic Manipulation Policies

ICRA 2026poster

Scaling real robot data is a key bottleneck in imitation learning, leading to the use of auxiliary data for policy training. While other aspects of robotic manipulation such as image or language understanding may be learned from internet-based datasets, acquiring motion knowledge remains challenging…

2026

OPRIDE: Efficient Offline Preference-based Reinforcement Learning via In-Dataset Exploration

ICLR 2026poster

Preference-based reinforcement learning (PbRL) can help avoid sophisticated reward designs and align better with human intentions, showing great promise in various real-world applications. However, obtaining human feedback for preferences can be expensive and time-consuming, which forms a strong bar…

Cited by 0SourceScholar
2026

OneTwoVLA: A Unified Vision-Language-Action Model with Adaptive Reasoning

ICLR 2026poster

General-purpose robots capable of performing diverse tasks require synergistic reasoning and acting capabilities. However, recent dual-system approaches, which separate high-level reasoning from low-level acting, often suffer from challenges such as limited mutual understanding of capabilities betwe…

Cited by 0SourcecodeScholar
2026

PhenoBrain: Phenotype-Conditioned Long-Range Communication for Multi-Modal Brain Network Analysis

ICML 2026oral

Multi-modal brain network analysis aims to predict neuropsychiatric status from functional connectomes with heterogeneous phenotypes. However, most existing methods treat phenotypes as auxiliary features and perform late fusion, implicitly assuming that the connectome representation should be learne…

Cited by 0SourceScholar
2026

QPrompt-R1: Real-Time Reasoning for Domain-Generalized Semantic Segmentation via Group-Relative Query Alignment

ICLR 2026poster

Deploying semantic segmentation in driving and robotics requires both real-time inference and robustness to domain shifts, formalized as Real-Time Domain-Generalized Semantic Segmentation (RT-DGSS), which has not been fully addressed. Existing methods often treat real-time(RT) inference and domain g…

Cited by 0SourceScholar
2026

RAP: 3D Rasterization Augmented End-to-End Planning

ICLR 2026poster

Imitation learning for end-to-end driving trains policies only on expert demonstrations. Once deployed in a closed loop, such policies lack recovery data: small mistakes cannot be corrected and quickly compound into failures. A promising direction is to generate alternative viewpoints and trajectori…

Cited by 0SourcecodeScholar
2026

Retain and Adapt: Auto-Balanced Model Editing for Open-Vocabulary Object Detection under Domain Shifts

ICLR 2026poster

Recent advances in Open Vocabulary Object Detection (OVOD) have shown strong performance on standard benchmarks, but performance drops sharply under out-of-distribution (OOD) shifts. Continual learning offers a potential remedy by sequentially integrating new tasks, yet existing methods often strugg…

Cited by 0SourceScholar
2026

RoTri-Diff: A Spatial Robot–Object Triadic Interaction-Guided Diffusion Model for Bimanual Manipulation

ICRA 2026poster

Bimanual manipulation is a fundamental robotic skill that requires continuous and precise coordination between two arms. While imitation learning (IL) is the dominant paradigm for acquiring this capability, existing approaches, whether robot-centric or object-centric, often overlook the dynamic geom…

2026

SAME: Sparse and Anchored Model Editing for Heterogeneous Incremental Learning under Limited Data

CVPR 2026

Existing Incremental Learning (IL) methods are primarily evaluated under either a single-domain class-incremental setting, or a multi-domain task-incremental setting with known task identifiers. However, these assumptions often fail to hold in real-world applications. To bridge this gap, we introduc

Cited by 0SourceScholar
2026

SEPT: Standard-Definition Map Enhanced Scene Perception and Topology Reasoning for Autonomous Driving

ICRA 2026poster

Online scene perception and topology reasoning are critical for autonomous vehicles to understand their driving environment, particularly for mapless driving systems that endeavor to reduce reliance on costly High-Definition (HD) maps. However, recent advances in online scene understanding still fac…

2026

SafeDialBench: A Fine-Grained Safety Evaluation Benchmark for Large Language Models in Multi-Turn Dialogues with Diverse Jailbreak Attacks

ICLR 2026poster

With the rapid advancement of Large Language Models (LLMs), the safety of LLMs has been a critical concern requiring precise assessment. Current benchmarks primarily concentrate on single-turn dialogues or a single jailbreak attack method to assess the safety. Additionally, these benchmarks have not…

Cited by 0SourcecodeScholar
2026

Selective, Regularized, and Calibrated: Harnessing Vision Foundation Models for Cross-Domain Few-Shot Semantic Segmentation

CVPR 2026

Vision foundation models (VFMs) have achieved strong performance across various vision tasks. However, it still remains challenging to apply VFMs for cross-domain few-shot segmentation (CD-FSS), which segments objects of novel classes under domain shifts using only a few labeled exemplars. The chall

Cited by 0SourceScholar
2026

Simulated Rewards, Skewed Strategies: Tracing the Acquired Preference Bias in LLM-Based Dialogue Planners

AAAI 2026technical

Large language models have enabled sophisticated dialogue planning policy, but their reliance on LLM-generated simulation and feedback for policy optimization may introduce systematic preference bias. We present the first comprehensive analysis of preference bias in LLM-based dialogue planners, eval

Cited by 0SourcePDFScholar
2026

SoLoPO: Unlocking Long-Context Capabilities in LLMs via Short-to-Long Preference Optimization

ICLR 2026poster

Despite advances in pretraining with extended context sizes, large language models (LLMs) still face challenges in effectively utilizing real-world long-context information, primarily due to insufficient long-context alignment caused by data quality issues, training inefficiencies, and the lack of w…

Cited by 0SourcecodeScholar
2026

Spatially-Anchored Tactile Awareness for Robust Dexterous Manipulation

ICRA 2026poster

Abstract— Dexterous manipulation requires precise geometric reasoning, yet existing visuo-tactile learning methods struggle with sub-millimeter precision tasks that are routine for traditional model-based approaches. We identify a key limitation: while tactile sensors provide rich contact informatio…

2026

Stable Video Infinity: Infinite-Length Video Generation with Error Recycling

ICLR 2026oral

We propose **Stable Video Infinity (SVI)** that can generate non-looping, ultra-long videos with stable visual quality, while supporting per-clip prompt control and multi-modal conditioning. While existing long-video methods attempt to _**mitigate accumulated errors**_ via handcrafted anti-drifting…

Cited by 0SourcecodeScholar
2026

TRCoRSurg: Temporal-Relational Co-Reasoning for Surgical Video Triplet Recognition

CVPR 2026

Understanding complex surgical scenes requires recognizing multiple interdependent entities--such as instruments, actions, and targets--and maintaining their relational consistency across time. Existing surgical triplet recognition methods struggle to jointly model intra-frame label dependencies and

Cited by 0SourceScholar
2026

The Shape of Addition: Geometric Structures of Arithmetic in Large Language Models

ICML 2026poster

Large Language Models exhibit paradoxical fragility in fundamental arithmetic, implying a disconnect between internal computation and discrete output. By analyzing the residual stream geometry during multi-operand addition, we identify the **Iso-Raw-Sum Trajectory (IRST)**, a topological manifold wh…

Cited by 0SourceScholar
2026

TimeRewarder: Learning Dense Reward from Passive Videos via Frame-wise Temporal Distance

ICML 2026spotlight

Designing dense rewards is crucial for reinforcement learning (RL), yet in robotics it often demands extensive manual effort and lacks scalability. One promising solution is to view task progress as a dense reward signal, as it quantifies the degree to which actions advance the system toward task co…

Cited by 6SourceScholar
2026

Translating Flow to Policy via Hindsight Online Imitation

ICLR 2026poster

Recent advances in hierarchical robot systems leverage a high-level planner to propose task plans and a low-level policy to generate robot actions. This design allows training the planner on action-free or even non-robot data sources (e.g., videos), providing transferable high-level guidance. Nevert…

Cited by 0SourceScholar
2026

UHR-BAT: Budget-Aware Token Compression Vision-Language model for Ultra-High-Resolution Remote Sensing

ICML 2026poster

Ultra-high-resolution (UHR) remote sensing imagery couples kilometer-scale context with query-critical evidence that may occupy only a few pixels. Such vast spatial scale leads to a quadratic explosion of visual tokens and hinders the extraction of information from small objects. Previous works util…

Cited by 0SourceScholar
2026

Understanding and Enforcing Weight Disentanglement in Task Arithmetic

CVPR 2026

Task arithmetic provides an efficient, training-free way to edit pre-trained models, yet lacks a fundamental theoretical explanation for its success. The existing concept of "weight disentanglement" describes the ideal outcome of non-interfering task composition but does not reveal its underlying ca

Cited by 0SourcecodeScholar
2026

UniDex: A Robot Foundation Suite for Universal Dexterous Hand Control from Egocentric Human Videos

CVPR 2026

Dexterous manipulation remains challenging due to the cost of collecting real-robot teleoperation data, the heterogeneity of hand embodiments, and the high dimensionality of control. We present UniDex, a robot foundation suite that couples a large-scale robot-centric dataset with a unified vision-la

Cited by 0SourcecodeScholar
2025

A Conditional KAN Diffusion Network for Human Activity Recognition with Missing Sensor Signal Series

ICASSP 2025accepted

Human Activity Recognition (HAR) is crucial for applications like urban traffic management and health monitoring but faces challenges in handling complex patterns and missing sensor data. In this work, we propose a conditional Kolmogorov-Arnold network diffusion (CKAD) framework for HAR, which separ…

Cited by 0SourceScholar
2025

A Self-Evolving Framework for Multi-Agent Medical Consultation Based on Large Language Models

ICASSP 2025accepted

We propose a multi-agent approach (SeM-Agents) based on large language models for medical consultations. This framework incorporates various doctor roles and auxiliary roles, with agents communicating through natural language. Using a residual structure, the system conducts multi-round medical consu…

Cited by 0SourceScholar
2025

Adapting In-Domain Few-Shot Segmentation to New Domains without Source Domain Retraining

ICCV 2025poster

Cross-domain few-shot segmentation (CD-FSS) aims to segment objects of novel classes in new domains, which is often challenging due to the diverse characteristics of target domains and the limited availability of support data. Most CD-FSS methods redesign and retrain in-domain FSS models using abund…

2025

Association-Focused Path Aggregation for Graph Fraud Detection

NeurIPS 2025poster

Fraudulent activities have caused substantial negative social impacts and are exhibiting emerging characteristics such as intelligence and industrialization, posing challenges of high-order interactions, intricate dependencies, and the sparse yet concealed nature of fraudulent entities. Existing gra…

Cited by 0SourcecodeScholar
2025

Beyond Mandatory Federations: Balancing Egoism, Utilitarianism and Egalitarianism in Mixed-Motive Games

AAAI 2025technical

In the field of mixed-motive games, extensive multi-agent learning studies have explored the balance between egoism (individual interest), utilitarianism (collective interest), and egalitarianism (fairness). Traditional approaches often rely on manually designed reward functions, social norms, and a…

2025

Causal Information Prioritization for Efficient Reinforcement Learning

ICLR 2025poster

Current Reinforcement Learning (RL) methods often suffer from sample-inefficiency, resulting from blind exploration strategies that neglect causal relationships among states, actions, and rewards. Although recent causal approaches aim to address this problem, they lack grounded modeling of reward-gu…

Cited by 0SourcePDFScholar
2025

CoT-VTM: Visual-to-Music Generation with Chain-of-Thought Reasoning

ACL 2025finding

The application of visual-to-music generation (VTM) is rapidly growing. However, current VTM methods struggle with capturing the relationship between visuals and music in open-domain settings, mainly due to two challenges: the lack of large-scale, high-quality visual-music paired datasets and the ab…

2025

Combinatorial Ski Rental Problem: Robust and Learning-Augmented Algorithms

NeurIPS 2025poster

We introduce and study the Combinatorial Ski Rental (CSR) problem, which involves multiple items that can be rented or purchased, either individually or in combination. At each time step, a decision-maker must make an irrevocable buy-or-rent decision for items that have not yet been purchased, witho…

Cited by 0SourceScholar
2025

DON’T NEED RETRAINING: A Mixture of DETR and Vision Foundation Models for Cross-Domain Few-Shot Object Detection

NeurIPS 2025poster

Cross-Domain Few-Shot Object Detection (CD-FSOD) aims to generalize to unseen domains by leveraging a few annotated samples of the target domain, requiring models to exhibit both strong generalization and localization capabilities. However, existing well-trained detectors typically have strong loca…

Cited by 0SourceScholar
2025

Data Scaling Laws in Imitation Learning for Robotic Manipulation

ICLR 2025oral

Data scaling has revolutionized fields like natural language processing and computer vision, providing models with remarkable generalization capabilities. In this paper, we investigate whether similar data scaling laws exist in robotics, particularly in robotic manipulation, and whether appropriate…

2025

Dynamic Multi-Layer Null Space Projection for Vision-Language Continual Learning

ICCV 2025poster

Vision-Language Models (VLM) have emerged as a highly promising approach for Continual Learning (CL) due to their powerful generalized features. While adapter-based VLM can exploit both task-specific and task-agnostic features, current CL methods have largely overlooked the distinct and evolving par…

Cited by 0SourcePDFScholar
2025

Efficient Last-Iterate Convergence in Solving Extensive-Form Games

NeurIPS 2025poster

To establish last-iterate convergence for Counterfactual Regret Minimization (CFR) algorithms in learning a Nash equilibrium (NE) of extensive-form games (EFGs), recent studies reformulate learning an NE of the original EFG as learning the NEs of a sequence of (perturbed) regularized EFGs. Hence, pr…

Cited by 0SourcecodeScholar
2025

Enhancing Few-Shot Class-Incremental Learning via Training-Free Bi-Level Modality Calibration

CVPR 2025poster

Few-shot Class-Incremental Learning (FSCIL) challenges models to adapt to new classes with limited samples, presenting greater difficulties than traditional classincremental learning. While existing approaches rely heavily on visual models and require additional training during base or incremental p…

2025

FACET: Force-Adaptive Control via Impedance Reference Tracking for Legged Robots

CoRL 2025oral

Reinforcement learning (RL) has made significant strides in legged robot control, enabling locomotion across diverse terrains and complex loco-manipulation capabilities. However, the commonly used position or velocity tracking-based objectives are agnostic to forces experienced by the robot, leading…

Cited by 0SourceScholar
2025

Federated Multi-armed Bandits with Efficient Bit-Level Communications

NeurIPS 2025poster

In this work, we study the federated multi-armed bandit (FMAB) problem, where a set of distributed agents collaboratively aim to minimize cumulative regret while interacting with a shared set of arms. Unlike traditional centralized bandit models, agents in FMAB settings are connected via a communica…

Cited by 0SourceScholar
2025

Fine-Tuning Hard-to-Simulate Objectives for Quadruped Locomotion: A Case Study on Total Power Saving

ICRA 2025

Legged locomotion is not just about mobility; it also encompasses crucial objectives such as energy efficiency, safety, and user experience, which are vital for real-world applications. However, key factors such as battery power consumption and stepping noise are often inaccurately modeled or missin

Cited by 2SourcecodeScholar
2025

Firewall Routing: Blocking Leads to Better Hybrid Inference for LLMs

EMNLP 2025

The rapid advancement of Large Language Models (LLMs) has significantly enhanced performance across various natural language processing (NLP) tasks, yet the high computational costs and latency associated with deploying such models continue to pose critical bottlenecks, limiting their broader applic

Cited by 0SourcePDFScholar
2025

GravMAD: Grounded Spatial Value Maps Guided Action Diffusion for Generalized 3D Manipulation

ICLR 2025poster

Robots' ability to follow language instructions and execute diverse 3D manipulation tasks is vital in robot learning. Traditional imitation learning-based methods perform well on seen tasks but struggle with novel, unseen ones due to variability. Recent approaches leverage large foundation models to…

2025

Joint Modeling of fMRI and EEG Imaging Using Ordinary Differential Equation-Based Hypergraph Neural Networks

NeurIPS 2025poster

Fusing multimodal brain imaging has been a hot topic since different modalities of brain imaging can provide complementary information. However, due to the size of simultaneous recorded fMRI-EEG dataset being limited and the substantial discrepancy between hemodynamic responses of fMRI and neural os…

Cited by 0SourceScholar
2025

KineDex: Learning Tactile-Informed Visuomotor Policies via Kinesthetic Teaching for Dexterous Manipulation

CoRL 2025poster

Collecting demonstrations enriched with fine-grained tactile information is critical for dexterous manipulation, particularly in contact-rich tasks that require precise force control and physical interaction. While prior works primarily focus on teleoperation or video-based retargeting, they often s…

Cited by 0SourceScholar
2025

Large Language Models Enhanced Personalized Graph Neural Architecture Search in Federated Learning

AAAI 2025technical

Personalized federated learning (PFL) on graphs is an emerging field focusing on the collaborative development of architectures across multiple clients, each with distinct graph data distributions while adhering to strict privacy standards. This area often requires extensive expert intervention in m…

2025

Last-Iterate Convergence of Smooth Regret Matching$^+$ Variants in Learning Nash Equilibria

NeurIPS 2025poster

Regret Matching$^+$ (RM$^+$) variants are widely used to build superhuman Poker AIs, yet few studies investigate their last-iterate convergence in learning a Nash equilibrium (NE). Although their last-iterate convergence is established for games satisfying the Minty Variational Inequality (MVI), no…

Cited by 0SourcecodeScholar
2025

Learning Manipulation Skills through Robot Chain-of-Thought with Sparse Failure Guidance

IROS 2025

Reward engineering for policy learning has been a long-standing challenge in robotics. Recently, to avoid manual reward designs, vision-language models (VLMs) have shown promise in defining rewards for teaching robots manipulation skills. However, existing work often provides reward guidance that is

Cited by 10SourceScholar
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

Leveraging Flatness to Improve Information-Theoretic Generalization Bounds for SGD

ICLR 2025poster

Information-theoretic (IT) generalization bounds have been used to study the generalization of learning algorithms. These bounds are intrinsically data- and algorithm-dependent so that one can exploit the properties of data and algorithm to derive tighter bounds. However, we observe that although th…

2025

Multi-Agent Reinforcement Learning with Communication-Constrained Priors

NeurIPS 2025poster

Communication is one of the effective means to improve the learning of cooperative policy in multi-agent systems. However, in most real-world scenarios, lossy communication is a prevalent issue. Existing multi-agent reinforcement learning with communication, due to their limited scalability and robu…

Cited by 0SourceScholar
2025

NATRA: Noise-Agnostic Framework for Trajectory Prediction with Noisy Observations

ICCV 2025poster

Trajectory prediction aims to forecast an agent's future trajectories based on its historical observed trajectories, which is a critical task for various applications such as autonomous driving, robotics, and surveillance systems. Most existing trajectory prediction methods assume that the observed…

Cited by 0SourcePDFScholar
2025

Near-Optimal Regret Bounds for Federated Multi-armed Bandits with Fully Distributed Communication

UAI 2025

In this paper, we focus on the research of federated multi-armed bandit (FMAB) problems where agents can only communicate with their neighbors. All agents aim to solve a common multi-armed bandit (MAB) problem to minimize individual regrets, while group regret can also be minimized. In a federated b

Cited by 0SourcePDFScholar
2025

Node-Centric Meta Structure Search in Heterogeneous Graphs

ICASSP 2025accepted

Heterogeneous graphs are increasingly used to represent complex real-world scenarios with diverse entities and interactions by meta structures. Recently, the search of meta structures is combined with graph neural architecture search to automatically extract the semantic knowledge for various tasks…

Cited by 0SourceScholar
2025

OpenCoder: The Open Cookbook for Top-Tier Code Large Language Models

ACL 2025long

Code LLMs have been widely used in various domains, including code generation, logical reasoning, and agent systems. However, open-access code LLMs mostly only release weights, lacking key features such as reproducible data pipelines and transparent training protocols, which are crucial for advancin…

2025

RRM: Robust Reward Model Training Mitigates Reward Hacking

ICLR 2025poster

Reward models (RMs) play a pivotal role in aligning large language models (LLMs) with human preferences. However, traditional RM training, which relies on response pairs tied to specific prompts, struggles to disentangle prompt-driven preferences from prompt-independent artifacts, such as response l…

Cited by 0SourcePDFScholar
2025

Reducing Variance of Stochastic Optimization for Approximating Nash Equilibria in Normal-Form Games

ICML 2025spotlight

Nash equilibrium (NE) plays an important role in game theory. How to efficiently compute an NE in NFGs is challenging due to its complexity and non-convex optimization property. Machine Learning (ML), the cornerstone of modern artificial intelligence, has demonstrated remarkable empirical performanc…

Cited by 0SourcePDFScholar
2025

RoboEngine: Plug-and-Play Robot Data Augmentation with Semantic Robot Segmentation and Background Generation

IROS 2025

Visual augmentation has become a crucial technique for enhancing the visual robustness of imitation learning. However, existing methods are often limited by prerequisites such as camera calibration or the need for controlled environments (e.g., green screen setups). In this work, we introduce RoboEn

Cited by 36SourcecodeScholar
2025

SEPT: Standard-Definition Map Enhanced Scene Perception and Topology Reasoning for Autonomous Driving

RA-L 2025

Online scene perception and topology reasoning are critical for autonomous vehicles to understand their driving environments, particularly for mapless driving systems that endeavor to reduce reliance on costly High-Definition (HD) maps. However, recent advances in online scene understanding still fa

Cited by 6SourceScholar
2025

SKIL: Semantic Keypoint Imitation Learning for Generalizable Data-efficient Manipulation

RSS 2025poster

Real-world tasks such as garment manipulation and table rearrangement demand robots to perform generalizable, highly precise, and long-horizon actions. Although imitation learning has proven to be an effective approach for teaching robots new skills, large amounts of expert demonstration data are st…

Cited by 2PDFScholar
2025

SOLAMI: Social Vision-Language-Action Modeling for Immersive Interaction with 3D Autonomous Characters

CVPR 2025poster

Human beings are social animals. How to equip 3D autonomous characters with similar social intelligence that can perceive, understand and interact with humans remains an open yet foundamental problem. In this paper, we introduce SOLAMI, the first end-to-end Social vision-Language-Action (VLA) Modeli…

Cited by 2SourcePDFScholar
2025

SRA-MCTS: Self-driven Reasoning Augmentation with Monte Carlo Tree Search for Code Generation

IJCAI 2025

Large language models exhibit remarkable performance in simple code generation tasks. However, they encounter significant challenges when addressing complex problems that require reasoning and question decomposition. To tackle this, we propose a self-driven reasoning augmentation process, SRA-MCTS,

2025

Sharpness-aware Zeroth-order Optimization for Graph Transformers

IJCAI 2025

Graph Transformers (GTs) have emerged as powerful tools for handling graph-structured data through global attention mechanisms. While GTs can effectively capture long-range dependencies, they introduce difficulties in optimization due to their complex, non-differentiable operators, which cannot be d

2025

SuperGPQA: Scaling LLM Evaluation across 285 Graduate Disciplines

NeurIPS 2025poster

Large language models (LLMs) have demonstrated remarkable proficiency in mainstream academic disciplines such as mathematics, physics, and computer science. However, human knowledge encompasses over 200 specialized disciplines, far exceeding the scope of existing benchmarks. The capabilities of LLMs…

Cited by 215SourceScholar
2025

Towards Empowerment Gain through Causal Structure Learning in Model-Based Reinforcement Learning

ICLR 2025poster

In Model-Based Reinforcement Learning (MBRL), incorporating causal structures into dynamics models provides agents with a structured understanding of the environments, enabling efficient decision. Empowerment as an intrinsic motivation enhances the ability of agents to actively control their enviro…

Cited by 0SourcePDFScholar
2025

Unified Human Localization and Trajectory Prediction with Monocular Vision

ICRA 2025

Conventional human trajectory prediction models rely on clean curated data, requiring specialized equipment or manual labeling, which is often impractical for robotic applications. The existing predictors tend to overfit to clean observation affecting their robustness when used with noisy inputs. In

Cited by 2SourcecodeScholar
2025

Unveiling and Addressing Pseudo Forgetting in Large Language Models

ACL 2025finding

Although substantial efforts have been made to mitigate catastrophic forgetting in continual learning, the intrinsic mechanisms are not well understood. In this work, we demonstrate the existence of “pseudo forgetting”: the performance degradation in previous tasks is not attributed to a loss of cap…

Cited by 0SourcePDFScholar
2024

Angle Robustness Unmanned Aerial Vehicle Navigation in GNSS-Denied Scenarios

AAAI 2024technical

Due to the inability to receive signals from the Global Navigation Satellite System (GNSS) in extreme conditions, achieving accurate and robust navigation for Unmanned Aerial Vehicles (UAVs) is a challenging task. Recently emerged, vision-based navigation has been a promising and feasible alternativ…

2024

Any-point Trajectory Modeling for Policy Learning

RSS 2024poster

Learning from demonstration is a powerful method for teaching robots new skills, and having more demonstration data often improves policy learning. However, the high cost of collecting demonstration data is a significant bottleneck. Videos, as a rich data source, contain knowledge of behaviors, phys…

Cited by 102SourcePDFScholar
2024

CItruS: Chunked Instruction-aware State Eviction for Long Sequence Modeling

EMNLP 2024main

Long sequence modeling has gained broad interest as large language models (LLMs) continue to advance. Recent research has identified that a large portion of hidden states within the key-value caches of Transformer models can be discarded (also termed evicted) withoutaffecting the perplexity performa…

2024

CoPa: General Robotic Manipulation through Spatial Constraints of Parts with Foundation Models

IROS 2024poster

Foundation models pre-trained on web-scale data are shown to encapsulate extensive world knowledge beneficial for robotic manipulation in the form of task planning. However, the actual physical implementation of these plans often relies on task-specific learning methods, which require significant da…

Cited by 52SourcecodeScholar
2024

Constructing and Exploring Intermediate Domains in Mixed Domain Semi-supervised Medical Image Segmentation

CVPR 2024poster

Both limited annotation and domain shift are prevalent challenges in medical image segmentation. Traditional semi-supervised segmentation and unsupervised domain adaptation methods address one of these issues separately. However the coexistence of limited annotation and domain shift is quite common…

2024

DGA-GNN: Dynamic Grouping Aggregation GNN for Fraud Detection

AAAI 2024technical

Fraud detection has increasingly become a prominent research field due to the dramatically increased incidents of fraud. The complex connections involving thousands, or even millions of nodes, present challenges for fraud detection tasks. Many researchers have developed various graph-based methods t…

2024

DexCatch: Learning to Catch Arbitrary Objects with Dexterous Hands

CoRL 2024poster

Achieving human-like dexterous manipulation remains a crucial area of research in robotics. Current research focuses on improving the success rate of pick-and-place tasks. Compared with pick-and-place, throwing-catching behavior has the potential to increase the speed of transporting objects to thei…

Cited by 4SourceScholar
2024

Digital Life Project: Autonomous 3D Characters with Social Intelligence

CVPR 2024poster

In this work we present Digital Life Project a framework utilizing language as the universal medium to build autonomous 3D characters who are capable of engaging in social interactions and expressing with articulated body motions thereby simulating life in a digital environment. Our framework compri…

Cited by 30SourcePDFScholar
2024

Discriminative Feature Decoupling Enhancement for Speech Forgery Detection

IJCAI 2024poster

The emergence of AIGC has brought attention to the issue of generating realistic deceptive content. While AIGC has the potential to revolutionize content creation, it also facilitates criminal activities. Specifically, the manipulation of speech has been exploited in tele-fraud and financial fraud s…

Cited by 0SourcePDFScholar
2024

Dynamic Replay Training for Class-Incremental Learning

ICASSP 2024accepted

Replay-based methods for Class-Incremental Learning (CIL) typically employ new classes and a limited subset of old classes stored in memory to facilitate the model training. However, these methods often lead to class imbalance and catastrophic forgetting, where the model forgets previously learned t…

Cited by 0SourceScholar
2024

EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data

ICML 2024spotlight

Sample efficiency remains a crucial challenge in applying Reinforcement Learning (RL) to real-world tasks. While recent algorithms have made significant strides in improving sample efficiency, none have achieved consistently superior performance across diverse domains. In this paper, we introduce Ef…

2024

Exploiting Inter-sample and Inter-feature Relations in Dataset Distillation

CVPR 2024poster

Dataset distillation has emerged as a promising approach in deep learning enabling efficient training with small synthetic datasets derived from larger real ones. Particularly distribution matching-based distillation methods attract attention thanks to its effectiveness and low computational cost. H…

2024

Fundamental Capabilities of Large Language Models and their Applications in Domain Scenarios: A Survey

ACL 2024long

Large Language Models (LLMs) demonstrate significant value in domain-specific applications, benefiting from their fundamental capabilities. Nevertheless, it is still unclear which fundamental capabilities contribute to success in specific domains. Moreover, the existing benchmark-based evaluation ca…

Cited by 4SourcePDFScholar
2024

HOIAnimator: Generating Text-prompt Human-object Animations using Novel Perceptive Diffusion Models

CVPR 2024poster

To date the quest to rapidly and effectively produce human-object interaction (HOI) animations directly from textual descriptions stands at the forefront of computer vision research. The underlying challenge demands both a discriminating interpretation of language and a comprehensive physics-centric…

Cited by 11SourcePDFScholar
2024

How Far Can In-Context Alignment Go? Exploring the State of In-Context Alignment

EMNLP 2024finding

Recent studies have demonstrated that In-Context Learning (ICL), through the use of specific demonstrations, can align Large Language Models (LLMs) with human preferences known as In-Context Alignment (ICA), indicating that models can comprehend human instructions without requiring parameter adjustm…

2024

Hybrid Sharing for Multi-Label Image Classification

ICLR 2024poster

Existing multi-label classification methods have long suffered from label heterogeneity, where learning a label obscures another. By modeling multi-label classification as a multi-task problem, this issue can be regarded as a negative transfer, which indicates challenges to achieve simultaneously sa…

2024

Imitation Learning from Observation with Automatic Discount Scheduling

ICLR 2024poster

Humans often acquire new skills through observation and imitation. For robotic agents, learning from the plethora of unlabeled video demonstration data available on the Internet necessitates imitating the expert without access to its action, presenting a challenge known as Imitation Learning from Ob…

2024

InsertNeRF: Instilling Generalizability into NeRF with HyperNet Modules

ICLR 2024poster

Generalizing Neural Radiance Fields (NeRF) to new scenes is a significant challenge that existing approaches struggle to address without extensive modifications to vanilla NeRF framework. We introduce **InsertNeRF**, a method for **INS**tilling g**E**ne**R**alizabili**T**y into **NeRF**. By utilizin…

2024

InternLM-XComposer2-4KHD: A Pioneering Large Vision-Language Model Handling Resolutions from 336 Pixels to 4K HD

NeurIPS 2024poster

The Large Vision-Language Model (LVLM) field has seen significant advancements, yet its progression has been hindered by challenges in comprehending fine-grained visual content due to limited resolution. Recent efforts have aimed to enhance the high-resolution understanding capabilities of LVLMs, ye…

2024

Learn to Preserve and Diversify: Parameter-Efficient Group with Orthogonal Regularization for Domain Generalization

ECCV 2024poster

"Domain generalization (DG) aims to avoid the performance degradation of the model when the distribution shift between the limited training data and unseen test data occurs. Recently, foundation models with enormous parameters have been pre-trained with huge datasets, demonstrating strong generaliza…

2024

Leveraging Locality to Boost Sample Efficiency in Robotic Manipulation

CoRL 2024poster

Given the high cost of collecting robotic data in the real world, sample efficiency is a consistently compelling pursuit in robotics. In this paper, we introduce SGRv2, an imitation learning framework that enhances sample efficiency through improved visual and action representations. Central to the…

Cited by 8SourcecodeScholar
2024

MQE: Unleashing the Power of Interaction with Multi-agent Quadruped Environment

IROS 2024poster

The advent of deep reinforcement learning (DRL) has significantly advanced the field of robotics, particularly in the control and coordination of quadruped robots. However, the complexity of real-world tasks often necessitates the deployment of multi-robot systems capable of sophisticated interactio…

Cited by 4SourcecodeScholar
2024

Meta Structure Search for Link Weight Prediction in Heterogeneous Graphs

ICASSP 2024accepted

Recently link weight prediction has attracted an increasing research interest due to its merits in quantifying the strength between nodes within a graph. Nonetheless, current link weight prediction methods focus solely on graph topology, disregarding node feature information embedded in graphs. In r…

Cited by 0SourceScholar
2024

Model LEGO: Creating Models Like Disassembling and Assembling Building Blocks

NeurIPS 2024poster

With the rapid development of deep learning, the increasing complexity and scale of parameters make training a new model increasingly resource-intensive. In this paper, we start from the classic convolutional neural network (CNN) and explore a paradigm that does not require training to obtain new mo…

2024

Multi-Agent Exploration via Self-Learning and Social Learning

ICASSP 2024accepted

Self-learning and social learning stand as two pivotal constituents in multi-agent exploration. Inspired by the fact that animals and humans explore unfamiliar environments to learn survival skills by training themselves using unlabeled data and replicating others’ successful experiences, we propose…

Cited by 0SourceScholar
2024

Multi-Agent Sparse Interaction Modeling is an Anomaly Detection Problem

ICASSP 2024accepted

Most real-world multi-agent tasks exhibit the characteristic of sparse interaction, wherein agents interact with each other in a limited number of crucial states while largely acting independently. Effectively modeling the sparse interaction and leveraging the learned interaction structure to instru…

Cited by 0SourceScholar
2024

OpenMSD: Towards Multilingual Scientific Documents Similarity Measurement

COLING 2024main

We develop and evaluate multilingual scientific documents similarity measurement models in this work. Such models can be used to find related papers in different languages, which can help multilingual researchers find and explore papers more efficiently. We propose the first multilingual scientific…

2024

Optimistic Value Instructors for Cooperative Multi-Agent Reinforcement Learning

AAAI 2024technical

In cooperative multi-agent reinforcement learning, decentralized agents hold the promise of overcoming the combinatorial explosion of joint action space and enabling greater scalability. However, they are susceptible to a game-theoretic pathology called relative overgeneralization that shadows the o…

Cited by 2SourcePDFScholar
2024

PG-LBO: Enhancing High-Dimensional Bayesian Optimization with Pseudo-Label and Gaussian Process Guidance

AAAI 2024technical

Variational Autoencoder based Bayesian Optimization (VAE-BO) has demonstrated its excellent performance in addressing high-dimensional structured optimization problems. However, current mainstream methods overlook the potential of utilizing a pool of unlabeled data to construct the latent space, whi…

2024

PSST: A Benchmark for Evaluation-driven Text Public-Speaking Style Transfer

EMNLP 2024finding

Language style is necessary for AI systems to accurately understand and generate diverse human language. However, previous text style transfer primarily focused on sentence-level data-driven approaches, limiting exploration of potential problems in large language models (LLMs) and the ability to mee…

2024

Reinforcement Learning with Foundation Priors: Let Embodied Agent Efficiently Learn on Its Own

CoRL 2024poster

Reinforcement learning (RL) is a promising approach for solving robotic manipulation tasks. However, it is challenging to apply the RL algorithms directly in the real world. For one thing, RL is data-intensive and typically requires millions of interactions with environments, which are impractical i…

Cited by 25SourceScholar
2024

Revisiting Disentanglement in Downstream Tasks: A Study on Its Necessity for Abstract Visual Reasoning

AAAI 2024technical

In representation learning, a disentangled representation is highly desirable as it encodes generative factors of data in a separable and compact pattern. Researchers have advocated leveraging disentangled representations to complete downstream tasks with encouraging empirical evidence. This paper f…

2024

SCaR: Refining Skill Chaining for Long-Horizon Robotic Manipulation via Dual Regularization

NeurIPS 2024poster

Long-horizon robotic manipulation tasks typically involve a series of interrelated sub-tasks spanning multiple execution stages. Skill chaining offers a feasible solution for these tasks by pre-training the skills for each sub-task and linking them sequentially. However, imperfections in skill learn…

Cited by 2SourcePDFScholar
2024

STAR: Spatio-Temporal State Compression for Multi-Agent Tasks with Rich Observations

IJCAI 2024poster

This paper focuses on the problem of learning compressed state representations for multi-agent tasks. Under the assumption of rich observation, we pinpoint that the state representations should be compressed both spatially and temporally to enable efficient prioritization of task-relevant features,…

Cited by 0SourcePDFScholar
2024

START: A Generalized State Space Model with Saliency-Driven Token-Aware Transformation

NeurIPS 2024poster

Domain Generalization (DG) aims to enable models to generalize to unseen target domains by learning from multiple source domains. Existing DG methods primarily rely on convolutional neural networks (CNNs), which inherently learn texture biases due to their limited receptive fields, making them prone…

2024

Safe and Robust Subgame Exploitation in Imperfect Information Games

ICML 2024poster

Opponent exploitation is an important task for players to exploit the weaknesses of others in games. Existing approaches mainly focus on balancing between exploitation and exploitability but are often vulnerable to modeling errors and deceptive adversaries. To address this problem, our paper offers…

Cited by 2SourcePDFScholar
2024

Scalable and Domain-General Abstractive Proposition Segmentation

EMNLP 2024finding

Segmenting text into fine-grained units of meaning is important to a wide range of NLP applications. The default approach of segmenting text into sentences is often insufficient, especially since sentences are usually complex enough to include multiple units of meaning that merit separate treatment…

Cited by 0SourcePDFScholar
2024

Seer: Language Instructed Video Prediction with Latent Diffusion Models

ICLR 2024poster

Imagining the future trajectory is the key for robots to make sound planning and successfully reach their goals. Therefore, text-conditioned video prediction (TVP) is an essential task to facilitate general robot policy learning. To tackle this task and empower robots with the ability to foresee the…

2024

Social-Transmotion: Promptable Human Trajectory Prediction

ICLR 2024poster

Accurate human trajectory prediction is crucial for applications such as autonomous vehicles, robotics, and surveillance systems. Yet, existing models often fail to fully leverage the non-verbal social cues human subconsciously communicate when navigating the space. To address this, we introduce *So…

2024

TD-GPT: Target Protein-Specific Drug Molecule Generation GPT

ICASSP 2024accepted

Drug discovery faces challenges due to the vast chemical space and complex drug-target interactions. This paper proposes a novel deep learning framework TD-GPT for targeted drug molecule generation. TD-GPT comprises a linear Transformer for drug-target affinity prediction, an affinity-enhanced prote…

Cited by 0SourceScholar
2024

The Devil is in the Statistics: Mitigating and Exploiting Statistics Difference for Generalizable Semi-supervised Medical Image Segmentation

ECCV 2024poster

"Despite the recent success of domain generalization in medical image segmentation, voxel-wise annotation for all source domains remains a huge burden. Semi-supervised domain generalization has been proposed very recently to combat this challenge by leveraging limited labeled data along with abundan…

2024

Transformer Doctor: Diagnosing and Treating Vision Transformers

NeurIPS 2024poster

Due to its powerful representational capabilities, Transformers have gradually become the mainstream model in the field of machine vision. However, the vast and complex parameters of Transformers impede researchers from gaining a deep understanding of their internal mechanisms, especially error mech…

Cited by 0SourcePDFScholar
2024

Uni-O4: Unifying Online and Offline Deep Reinforcement Learning with Multi-Step On-Policy Optimization

ICLR 2024poster

Combining offline and online reinforcement learning (RL) is crucial for efficient and safe learning. However, previous approaches treat offline and online learning as separate procedures, resulting in redundant designs and limited performance. We ask: *Can we achieve straightforward yet effective of…

2024

ViT-Calibrator: Decision Stream Calibration for Vision Transformer

AAAI 2024technical

A surge of interest has emerged in utilizing Transformers in diverse vision tasks owing to its formidable performance. However, existing approaches primarily focus on optimizing internal model architecture designs that often entail significant trial and error with high burdens. In this work, we prop…

2024

Weakly Supervised Multimodal Affordance Grounding for Egocentric Images

AAAI 2024technical

To enhance the interaction between intelligent systems and the environment, locating the affordance regions of objects is crucial. These regions correspond to specific areas that provide distinct functionalities. Humans often acquire the ability to identify these regions through action demonstration…

2024

Word Matters: What Influences Domain Adaptation in Summarization?

ACL 2024long

Domain adaptation aims to enable Large Language Models (LLMs) to generalize domain datasets unseen effectively during the training phase. However, factors such as the size of the model parameters and the scale of training data are general influencers and do not reflect the nuances of domain adaptati…

2023

A Policy Optimization Method Towards Optimal-time Stability

CoRL 2023poster

In current model-free reinforcement learning (RL) algorithms, stability criteria based on sampling methods are commonly utilized to guide policy optimization. However, these criteria only guarantee the infinite-time convergence of the system's state to an equilibrium point, which leads to sub-optima…

Cited by 3SourceScholar
2023

A Universal Semantic-Geometric Representation for Robotic Manipulation

CoRL 2023poster

Robots rely heavily on sensors, especially RGB and depth cameras, to perceive and interact with the world. RGB cameras record 2D images with rich semantic information while missing precise spatial information. On the other side, depth cameras offer critical 3D geometry data but capture limited seman…

Cited by 22SourcecodeScholar
2023

An Efficient Deep Reinforcement Learning Algorithm for Solving Imperfect Information Extensive-Form Games

AAAI 2023technical

One of the most popular methods for learning Nash equilibrium (NE) in large-scale imperfect information extensive-form games (IIEFGs) is the neural variants of counterfactual regret minimization (CFR). CFR is a special case of Follow-The-Regularized-Leader (FTRL). At each iteration, the neural varia…

2023

Become a Proficient Player with Limited Data through Watching Pure Videos

ICLR 2023poster

Recently, RL has shown its strong ability for visually complex tasks. However, it suffers from the low sample efficiency and poor generalization ability, which prevent RL from being useful in real-world scenarios. Inspired by the huge success of unsupervised pre-training methods on language and visi…

2023

Convergence Analysis of Graphical Game-Based Nash Q-Learning using the Interaction Detection Signal of N-Step Return

ICASSP 2023accepted

The graphical game provides an effective method for modeling different kinds of sparse interactions in multi-agent reinforcement learning. Most previous work on game abstraction lacks theoretical guarantees of convergence. In this paper, we adopt the ${\mathcal{N}}$-step return signal to detect inte…

Cited by 0SourceScholar
2023

DNA-Rendering: A Diverse Neural Actor Repository for High-Fidelity Human-Centric Rendering

ICCV 2023poster

Realistic human-centric rendering plays a key role in both computer vision and computer graphics. Rapid progress has been made in the algorithm aspect over the years, yet existing human-centric rendering datasets and benchmarks are rather impoverished in terms of diversity (e.g., outfit's fabric/mat…

Cited by 61PDFcodeScholar
2023

Efficient Exploration Using Extra Safety Budget in Constrained Policy Optimization

IROS 2023poster

Reinforcement learning (RL) has achieved promising results on most robotic control tasks. Safety of learning-based controllers is an essential notion of ensuring the effectiveness of the controllers. Current methods adopt whole consistency constraints during the training, thus resulting in inefficie…

Cited by 2SourceScholar
2023

Enhanced Tensor Low-Rank and Sparse Representation Recovery for Incomplete Multi-View Clustering

AAAI 2023technical

Incomplete multi-view clustering (IMVC) has attracted remarkable attention due to the emergence of multi-view data with missing views in real applications. Recent methods attempt to recover the missing information to address the IMVC problem. However, they generally cannot fully explore the underlyi…

Cited by 71SourcePDFScholar
2023

Entity-Agnostic Representation Learning for Parameter-Efficient Knowledge Graph Embedding

AAAI 2023technical

We propose an entity-agnostic representation learning method for handling the problem of inefficient parameter storage costs brought by embedding knowledge graphs. Conventional knowledge graph embedding methods map elements in a knowledge graph, including entities and relations, into continuous vect…

2023

For Pre-Trained Vision Models in Motor Control, Not All Policy Learning Methods are Created Equal

ICML 2023poster

In recent years, increasing attention has been directed to leveraging pre-trained vision models for motor control. While existing works mainly emphasize the importance of this pre-training phase, the arguably equally important role played by downstream policy learning during control-specific fine-tu…

Cited by 26SourcePDFScholar
2023

Graph vs. Sequence: An Empirical Study on Knowledge Forms for Knowledge-Grounded Dialogue

EMNLP 2023long main

Knowledge-grounded dialogue is a task of gener- ating an informative response based on both the dialogue history and external knowledge source. In general, there are two forms of knowledge: manu- ally annotated knowledge graphs and knowledge text from website. From various evaluation viewpoints, eac…

Cited by 0SourceScholar
2023

IOMatch: Simplifying Open-Set Semi-Supervised Learning with Joint Inliers and Outliers Utilization

ICCV 2023oral

Semi-supervised learning (SSL) aims to leverage massive unlabeled data when labels are expensive to obtain. Unfortunately, in many real-world applications, the collected unlabeled data will inevitably contain unseen-class outliers not belonging to any of the labeled classes. To deal with the challen…

Cited by 32PDFcodeScholar
2023

Learning Explicit Credit Assignment for Cooperative Multi-Agent Reinforcement Learning via Polarization Policy Gradient

AAAI 2023technical

Cooperative multi-agent policy gradient (MAPG) algorithms have recently attracted wide attention and are regarded as a general scheme for the multi-agent system. Credit assignment plays an important role in MAPG and can induce cooperation among multiple agents. However, most MAPG algorithms cannot a…

Cited by 10SourcePDFScholar
2023

Maximum Entropy Population-Based Training for Zero-Shot Human-AI Coordination

AAAI 2023technical

We study the problem of training a Reinforcement Learning (RL) agent that is collaborative with humans without using human data. Although such agents can be obtained through self-play training, they can suffer significantly from the distributional shift when paired with unencountered partners, such…

2023

Modeling Inter-Class and Intra-Class Constraints in Novel Class Discovery

CVPR 2023poster

Novel class discovery (NCD) aims at learning a model that transfers the common knowledge from a class-disjoint labelled dataset to another unlabelled dataset and discovers new classes (clusters) within it. Many methods, as well as elaborate training pipelines and appropriate objectives, have been pr…

2023

Modified Retrace for Off-Policy Temporal Difference Learning

UAI 2023poster

Off-policy learning is a key to extend reinforcement learning as it allows to learn a target policy from a different behavior policy that generates the data. However, it is well known as “the deadly triad” when combined with bootstrapping and function approximation. Retrace is an efficient and con…

Cited by 4SourcePDFScholar
2023

Orthogonal Annotation Benefits Barely-Supervised Medical Image Segmentation

CVPR 2023poster

Recent trends in semi-supervised learning have significantly boosted the performance of 3D semi-supervised medical image segmentation. Compared with 2D images, 3D medical volumes involve information from different directions, e.g., transverse, sagittal, and coronal planes, so as to naturally provide…

2023

Predictive Inference with Feature Conformal Prediction

ICLR 2023poster

Conformal prediction is a distribution-free technique for establishing valid prediction intervals. Although conventionally people conduct conformal prediction in the output space, this is not the only possibility. In this paper, we propose feature conformal prediction, which extends the scope of con…

2023

Programmatically Grounded, Compositionally Generalizable Robotic Manipulation

ICLR 2023top-25%

Robots operating in the real world require both rich manipulation skills as well as the ability to semantically reason about when to apply those skills. Towards this goal, recent works have integrated semantic representations from large-scale pretrained vision-language (VL) models into manipulation…

2023

SpeedyZero: Mastering Atari with Limited Data and Time

ICLR 2023poster

Many recent breakthroughs of deep reinforcement learning (RL) are mainly built upon large-scale distributed training of model-free methods using millions to billions of samples. On the other hand, state-of-the-art model-based RL methods can achieve human-level sample efficiency but often take a much…

Cited by 5SourcePDFScholar
2023

TemplateGEC: Improving Grammatical Error Correction with Detection Template

ACL 2023long

Grammatical error correction (GEC) can be divided into sequence-to-edit (Seq2Edit) and sequence-to-sequence (Seq2Seq) frameworks, both of which have their pros and cons. To utilize the strengths and make up for the shortcomings of these frameworks, this paper proposes a novel method, TemplateGEC, wh…

2023

USEEK: Unsupervised SE(3)-Equivariant 3D Keypoints for Generalizable Manipulation

ICRA 2023poster

Can a robot manipulate intra-category unseen objects in arbitrary poses with the help of a mere demonstration of grasping pose on a single object instance? In this paper, we try to address this intriguing challenge by using USEEK, an unsupervised SE(3)-equivariant keypoints method that enjoys alignm…

Cited by 29SourceScholar
2022

An Empirical Study on Disentanglement of Negative-free Contrastive Learning

NeurIPS 2022accept

Negative-free contrastive learning methods have attracted a lot of attention with simplicity and impressive performances for large-scale pretraining. However, its disentanglement property remains unexplored. In this paper, we examine negative-free contrastive learning methods to study the disentangl…

2022

CYBORGS: Contrastively Bootstrapping Object Representations by Grounding in Segmentation

ECCV 2022poster

"Many recent approaches in contrastive learning have worked to close the gap between pretraining on iconic images like ImageNet and pretraining on complex scenes like COCO. This gap exists largely because commonly used random crop augmentations obtain semantically inconsistent content in crowded sce…

2022

EleGANt: Exquisite and Locally Editable GAN for Makeup Transfer

ECCV 2022poster

"Most existing methods view makeup transfer as transferring color distributions of different facial regions and ignore details such as eye shadows and blushes. Besides, they only achieve controllable transfer within predefined fixed regions. This paper emphasizes the transfer of makeup details and s…

2022

Fighting Fire with Fire: Avoiding DNN Shortcuts through Priming

ICML 2022spotlight

Across applications spanning supervised classification and sequential control, deep learning has been reported to find “shortcut” solutions that fail catastrophically under minor changes in the data distribution. In this paper, we show empirically that DNNs can be coaxed to avoid poor shortcuts by p…

Cited by 21SourcePDFScholar
2022

HuMMan: Multi-modal 4D Human Dataset for Versatile Sensing and Modeling

ECCV 2022poster

"4D human sensing and modeling are fundamental tasks in vision and graphics with numerous applications. With the advances of new sensors and algorithms, there is an increasing demand for more versatile datasets. In this work, we contribute HuMMan, a large-scale multi-modal 4D human dataset with 1000…

Cited by 125SourcePDFScholar
2022

Individual Reward Assisted Multi-Agent Reinforcement Learning

ICML 2022spotlight

In many real-world multi-agent systems, the sparsity of team rewards often makes it difficult for an algorithm to successfully learn a cooperative team policy. At present, the common way for solving this problem is to design some dense individual rewards for the agents to guide the cooperation. Howe…

2022

LaSSL: Label-Guided Self-Training for Semi-supervised Learning

AAAI 2022technical

The key to semi-supervised learning (SSL) is to explore adequate information to leverage the unlabeled data. Current dominant approaches aim to generate pseudo-labels on weakly augmented instances and train models on their corresponding strongly augmented variants with high-confidence results. Howev…

2022

MVDG: A Unified Multi-View Framework for Domain Generalization

ECCV 2022poster

"Aiming to generalize the model trained in source domains to unseen target domains, domain generalization (DG) has attracted lots of attention recently. Since target domains can not be involved in training, overfitting to source domains is inevitable. As a popular regularization technique, the meta-…

2022

PSP: Pre-trained Soft Prompts for Few-Shot Abstractive Summarization

COLING 2022main

Few-shot abstractive summarization has become a challenging task in natural language generation. To support it, we developed a novel soft prompts architecture coupled with a prompt pre-training plus prompt fine-tuning paradigm, which is effective and tunes only extremely light parameters. To meet th…

Cited by 26SourcePDFScholar
2022

Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learning

NeurIPS 2022accept

Learning generalizable policies that can adapt to unseen environments remains challenging in visual Reinforcement Learning (RL). Existing approaches try to acquire a robust representation via diversifying the appearances of in-domain observations for better generalization. Limited by the specific ob…

Cited by 85SourcePDFScholar
2022

Resolving Copycat Problems in Visual Imitation Learning via Residual Action Prediction

ECCV 2022poster

"Imitation learning is a widely used policy learning method that enables intelligent agents to acquire complex skills from expert demonstrations. The input to the imitation learning algorithm is usually composed of both the current observation and historical observations since the most recent observ…

Cited by 12SourcePDFScholar
2022

ST++: Make Self-Training Work Better for Semi-Supervised Semantic Segmentation

CVPR 2022poster

Self-training via pseudo labeling is a conventional, simple, and popular pipeline to leverage unlabeled data. In this work, we first construct a strong baseline of self-training (namely ST) for semi-supervised semantic segmentation via injecting strong data augmentations (SDA) on unlabeled images to…

Cited by 477PDFcodeScholar
2022

Spending Thinking Time Wisely: Accelerating MCTS with Virtual Expansions

NeurIPS 2022accept

One of the most important AI research questions is to trade off computation versus performance since ``perfect rationality" exists in theory but is impossible to achieve in practice. Recently, Monte-Carlo tree search (MCTS) has attracted considerable attention due to the significant performance impr…

2022

Stage-wise Stylistic Headline Generation: Style Generation and Summarized Content Insertion

IJCAI 2022poster

A quality headline with a high click-rate should not only summarize the content of an article, but also reflect a style that attracts users. Such demand has drawn rising attention to the task of stylistic headline generation (SHG). An intuitive method is to first generate plain headlines leveraged b…

2021

Conv1D Energy-Aware Path Planner for Mobile Robots in Unstructured Environments

ICRA 2021poster

Driving energy consumption plays a major role in the navigation of mobile robots in challenging environments, especially if they are left to operate unattended under limited on-board power. This paper reports on first results of an energy-aware path planner, which can provide estimates of the drivin…

Cited by 18SourceScholar
2021

Discovering Non-monotonic Autoregressive Orderings with Variational Inference

ICLR 2021poster

The predominant approach for language modeling is to encode a sequence of tokens from left to right, but this eliminates a source of information: the order by which the sequence was naturally generated. One strategy to recover this information is to decode both the content and ordering of tokens. So…

2021

Episodic Multi-agent Reinforcement Learning with Curiosity-driven Exploration

NeurIPS 2021poster

Efficient exploration in deep cooperative multi-agent reinforcement learning (MARL) still remains challenging in complex coordination problems. In this paper, we introduce a novel Episodic Multi-agent reinforcement learning with Curiosity-driven exploration, called EMC. We leverage an insight of pop…

Cited by 101SourcePDFScholar
2021

Exploring Explainable Selection to Control Abstractive Summarization

AAAI 2021technical

Like humans, document summarization models can interpret a document’s contents in a number of ways. Unfortunately, the neural models of today are largely black boxes that provide little explanation of how or why they generated a summary in the way they did. Therefore, to begin prying open the black…

2021

LoFGAN: Fusing Local Representations for Few-Shot Image Generation

ICCV 2021poster

Given only a few available images for a novel unseen category, few-shot image generation aims to generate more data for this category. Previous works attempt to globally fuse these images by using adjustable weighted coefficients. However, there is a serious semantic misalignment between different i…

Cited by 79PDFcodeScholar
2021

Manifold Alignment for Semantically Aligned Style Transfer

ICCV 2021poster

Most existing style transfer methods follow the assumption that styles can be represented with global statistics (e.g., Gram matrices or covariance matrices), and thus address the problem by forcing the output and style images to have similar global statistics. An alternative is the assumption of lo…

Cited by 60PDFcodeScholar
2021

Mastering Atari Games with Limited Data

NeurIPS 2021poster

Reinforcement learning has achieved great success in many applications. However, sample efficiency remains a key challenge, with prominent methods requiring millions (or even billions) of environment steps to train. Recently, there has been significant progress in sample efficient image-based RL al…

2021

NDT-Transformer: Large-Scale 3D Point Cloud Localisation using the Normal Distribution Transform Representation

ICRA 2021poster

3D point cloud-based place recognition is highly demanded by autonomous driving in GPS-challenged environments and serves as an essential component (i.e. loop-closure detection) in lidar-based SLAM systems. This paper proposes a novel approach, named NDT-Transformer, for real-time and large-scale pl…

Cited by 117SourcecodeScholar
2021

Prototypical Cross-Domain Self-Supervised Learning for Few-Shot Unsupervised Domain Adaptation

CVPR 2021poster

Unsupervised Domain Adaptation (UDA) transfers predictive models from a fully-labeled source domain to an unlabeled target domain. In some applications, however, it is expensive even to collect labels in the source domain, making most previous works impractical. To cope with this problem, recent wor…

Cited by 206PDFcodeScholar
2021

Reinforcement Learning with Latent Flow

NeurIPS 2021poster

Temporal information is essential to learning effective policies with Reinforcement Learning (RL). However, current state-of-the-art RL algorithms either assume that such information is given as part of the state space or, when learning from pixels, use the simple heuristic of frame-stacking to imp…

Cited by 29SourcePDFScholar