← Search

Ya-Qin Zhang

38 accepted papers

2026

Beyond Euclidean Clipping: Overcoming Exploration Collapse in LLM RL via Riemannian Isometric Policy Optimization

ICML 2026poster

Reinforcement learning (RL) has become a dominant paradigm for enhancing LLMs' reasoning capabilities. However, RL algorithms with PPO-Clip are inherently limited by exploration collapse. Subsequent works remain primarily heuristic and fail to identify the essential cause of PPO-Clip’s failure. This…

Cited by 0SourceScholar
2026

DGGT: Feedforward 4D Reconstruction of Dynamic Driving Scenes using Unposed Images

CVPR 2026

Autonomous driving needs fast, scalable 4D reconstruction and re-simulation for training and evaluation, yet most methods for dynamic driving scenes still rely on per-scene optimization, known camera calibration, or short frame windows, making them slow and impractical. We revisit this problem from

Cited by 0SourcecodeScholar
2026

Drugging the Undruggable: Benchmarking and Modeling Fragment-Based Screening

ICLR 2026poster

A significant portion of disease-relevant proteins remain undruggable due to shallow, flexible, or otherwise ill-defined binding pockets that hinder conventional molecule screening. Fragment-based drug discovery (FBDD) offers a promising alternative, as small, low-complexity fragments can flexibly e…

Cited by 0SourceScholar
2026

Learning Protein–Ligand Binding in Hyperbolic Space

AAAI 2026technical

Protein-ligand binding prediction is central to virtual screening and affinity ranking, two fundamental tasks in drug discovery. While recent retrieval-based methods embed ligands and protein pockets into Euclidean space for similarity-based search, the geometry of Euclidean embeddings often fails t

Cited by 0SourcePDFScholar
2026

MemAgent: Reshaping Long-Context LLM with Multi-Conv RL-based Memory Agent

ICLR 2026oral

Despite improvements by length extrapolation, efficient attention and memory modules, handling infinitely long documents without performance degradation during extrapolation remains the ultimate challenge in long-text processing. To solve this problem, We introduce a novel agent workflow, \method, w…

Cited by 0SourceScholar
2026

PE-SGD: Differentially Private Deep Learning via Evolution of Gradient Subspace for Text

ICLR 2026poster

Differentially Private Stochastic Gradient Descent (DP-SGD) and its variants like DP-Adam ensure data privacy by injecting noise into per-sample gradients. Although effective with large private datasets, their performance degrades significantly when private training data is limited. Recent works lev…

Cited by 0SourcecodeScholar
2026

Self-Improving Robot Policy with Compositional World Model

RSS 2026poster

Despite the sustained scaling on model capacity and data acquisition, Vision–Language–Action (VLA) models remain brittle in contact-rich and dynamic manipulation tasks, where minor execution deviations can compound into failures. While reinforcement learning (RL) offers a principled path to robustne…

Cited by 0SourceScholar
2026

S²Drug: Bridging Protein Sequence and 3D Structure in Contrastive Representation Learning for Virtual Screening

AAAI 2026technical

Virtual screening (VS) is an essential task in drug discovery, focusing on the identification of small-molecule ligands that bind to specific protein pockets. Existing deep learning methods, from early regression models to recent contrastive learning approaches, primarily rely on structural data whi

Cited by 0SourcePDFScholar
2026

Vector Quantization using Gaussian Variational Autoencoder

ICML 2026poster

Vector-quantized variational autoencoders (VQ-VAEs) are discrete autoencoders that compress images into discrete tokens. However, they are difficult to train due to discretization. In this paper, we propose a simple yet effective technique dubbed __Gaussian Quant (GQ)__, which first trains a Gaussia…

Cited by 0SourceScholar
2026

X-VLA: Soft-Prompted Transformer as Scalable Cross-Embodiment Vision-Language-Action Model

ICLR 2026poster

Successful generalist Vision-Language-Action (VLA) models that rely on effective training across diverse robotic platforms with large-scale, cross-embodiment, heterogeneous datasets. To facilitate and leverage the heterogeneity in rich, diverse robotic data sources, we propose a novel Soft Prompt ap…

Cited by 0SourcecodeScholar
2025

AANet: Virtual Screening under Structural Uncertainty via Alignment and Aggregation

NeurIPS 2025poster

Virtual screening (VS) is a critical component of modern drug discovery, yet most existing methods—whether physics-based or deep learning-based—are developed around *holo* protein structures with known ligand-bound pockets. Consequently, their performance degrades significantly on *apo* or predicted…

Cited by 0SourcecodeScholar
2025

CIDD: Collaborative Intelligence for Structure-Based Drug Design Empowered by LLMs

NeurIPS 2025poster

Structure-guided molecular generation is pivotal in early-stage drug discovery, enabling the design of compounds tailored to specific protein targets. However, despite recent advances in 3D generative modeling, particularly in improving docking scores, these methods often produce rare and intrinsica…

Cited by 0SourceScholar
2025

Contrastive Private Data Synthesis via Weighted Multi-PLM Fusion

ICML 2025poster

Substantial quantity and high quality are the golden rules of making a good training dataset with sample privacy protection equally important. Generating synthetic samples that resemble high-quality private data while ensuring Differential Privacy (DP), a formal privacy guarantee, promises scalabili…

2025

CoopDETR: A Unified Cooperative Perception Framework for 3D Detection via Object Query

ICRA 2025

Cooperative perception enhances the individual perception capabilities of autonomous vehicles (AVs) by providing a comprehensive view of the environment. However, balancing perception performance and transmission costs remains a significant challenge. Current approaches that transmit regionlevel fea

Cited by 9SourceScholar
2025

DAPO: An Open-Source LLM Reinforcement Learning System at Scale

NeurIPS 2025poster

Inference scaling empowers LLMs with unprecedented reasoning ability, with reinforcement learning as the core technique to elicit complex reasoning. However, key technical details of state-of-the-art reasoning LLMs are concealed (such as in OpenAI o1 blog and DeepSeek R1 technical report), thus the…

Cited by 0SourceScholar
2025

Enhancing Indoor Occupancy Prediction via Sparse Query-Based Multi-Level Consistent Knowledge Distillation

RA-L 2025

Occupancy prediction provides critical geometric and semantic understanding for robotics but faces efficiency-accuracy trade-offs. Current dense methods suffer computational waste on empty voxels, while sparse query-based approaches lack robustness in diverse and complex indoor scenes. In this paper

Cited by 1SourceScholar
2025

IROAM: Improving Roadside Monocular 3D Object Detection Learning from Autonomous Vehicle Data Domain

ICRA 2025

In autonomous driving, The perception capabilities of the ego-vehicle can be improved with roadside sensors, which can provide a holistic view of the environment. However, existing monocular detection methods designed for vehicle cameras are not suitable for roadside cameras due to viewpoint domain

Cited by 0SourceScholar
2025

InvRGB+L: Inverse Rendering of Complex Scenes with Unified Color and LiDAR Reflectance Modeling

ICCV 2025poster

We present InvRGB+L, a novel inverse rendering model that reconstructs large, relightable, and dynamic scenes from a single RGB+LiDAR sequence. Conventional inverse graphics methods rely primarily on RGB observations and use LiDAR mainly for geometric information, often resulting in suboptimal mater…

Cited by 0SourcePDFScholar
2025

PICD: Versatile Perceptual Image Compression with Diffusion Rendering

CVPR 2025poster

Recently, perceptual image compression has achieved significant advancements, delivering high visual quality at low bitrates for natural images. However, for screen content, existing methods often produce noticeable artifacts when compressing text. To tackle this challenge, we propose versatile perc…

Cited by 0SourcePDFScholar
2025

Redefining the task of Bioactivity Prediction

ICLR 2025poster

Small molecules are vital to modern medicine, and accurately predicting their bioactivity against protein targets is crucial for therapeutic discovery and development. However, current machine learning models often rely on spurious features, leading to biased outcomes. Notably, a simple pocket-only…

Cited by 0SourcePDFScholar
2025

Reframing Structure-Based Drug Design Model Evaluation via Metrics Correlated to Practical Needs

ICLR 2025poster

Recent advances in structure-based drug design (SBDD) have produced surprising results, with models often generating molecules that achieve better Vina docking scores than actual ligands. However, these results are frequently overly optimistic due to the limitations of docking score accuracy and the…

Cited by 0SourcePDFScholar
2025

Rethinking Diffusion Posterior Sampling: From Conditional Score Estimator to Maximizing a Posterior

ICLR 2025poster

Recent advancements in diffusion models have been leveraged to address inverse problems without additional training, and Diffusion Posterior Sampling (DPS) (Chung et al., 2022a) is among the most popular approaches. Previous analyses suggest that DPS accomplishes posterior sampling by approximating…

2025

Robo-MUTUAL: Robotic Multimodal Task Specification via Unimodal Learning

ICRA 2025

Multimodal task specification is essential for enhanced robotic performance, where Cross-modality Alignment enables the robot to holistically understand complex task instructions. Directly annotating multimodal instructions for model training proves impractical, due to the sparsity of paired multimo

Cited by 5SourceScholar
2025

Universal Actions for Enhanced Embodied Foundation Models

CVPR 2025poster

Training on diverse, internet-scale data is a key factor in the success of recent large foundation models. Yet, using the same recipe for building embodied agents has faced noticeable difficulties. Despite the availability of many crowd-sourced embodied datasets, their action spaces often exhibit si…

2024

DecisionNCE: Embodied Multimodal Representations via Implicit Preference Learning

ICML 2024poster

Multimodal pretraining is an effective strategy for the trinity of goals of representation learning in autonomous robots: $1)$ extracting both local and global task progressions; $2)$ enforcing temporal consistency of visual representation; $3)$ capturing trajectory-level language grounding. Most ex…

2024

EMIFF: Enhanced Multi-scale Image Feature Fusion for Vehicle-Infrastructure Cooperative 3D Object Detection

ICRA 2024poster

In autonomous driving, cooperative perception makes use of multi-view cameras from both vehicles and infrastructure, providing a global vantage point with rich semantic context of road conditions beyond a single vehicle viewpoint. Currently, two major challenges persist in vehicle-infrastructure coo…

Cited by 7SourcecodeScholar
2024

FuseGen: PLM Fusion for Data-generation based Zero-shot Learning

EMNLP 2024main

Data-generation based zero-shot learning, although effective in training Small Task-specific Models (STMs) via synthetic datasets generated by Pre-trained Language Models (PLMs), is often limited by the low quality of such synthetic datasets. Previous solutions have primarily focused on single PLM s…

2024

Idempotence and Perceptual Image Compression

ICLR 2024spotlight

Idempotence is the stability of image codec to re-compression. At the first glance, it is unrelated to perceptual image compression. However, we find that theoretically: 1) Conditional generative model-based perceptual codec satisfies idempotence; 2) Unconditional generative model with idempotence c…

2024

Query-Policy Misalignment in Preference-Based Reinforcement Learning

ICLR 2024spotlight

Preference-based reinforcement learning (PbRL) provides a natural way to align RL agents’ behavior with human desired outcomes, but is often restrained by costly human feedback. To improve feedback efficiency, most existing PbRL methods focus on selecting queries to maximally improve the overall qua…

2024

VFLAIR: A Research Library and Benchmark for Vertical Federated Learning

ICLR 2024poster

Vertical Federated Learning (VFL) has emerged as a collaborative training paradigm that allows participants with different features of the same group of users to accomplish cooperative training without exposing their raw data or model parameters. VFL has gained significant attention for its research…

2023

Bit Allocation using Optimization

ICML 2023poster

In this paper, we consider the problem of bit allocation in Neural Video Compression (NVC). First, we reveal a fundamental relationship between bit allocation in NVC and Semi-Amortized Variational Inference (SAVI). Specifically, we show that SAVI with GoP (Group-of-Picture)-level likelihood is equiv…

2023

Idempotent Learned Image Compression with Right-Inverse

NeurIPS 2023poster

We consider the problem of idempotent learned image compression (LIC). The idempotence of codec refers to the stability of codec to re-compression. To achieve idempotence, previous codecs adopt invertible transforms such as DCT and normalizing flow. In this paper, we first identify that invertibilit…

Cited by 4SourcePDFScholar
2023

LODE: Locally Conditioned Eikonal Implicit Scene Completion from Sparse LiDAR

ICRA 2023poster

Scene completion refers to obtaining dense scene representation from an incomplete perception of complex 3D scenes. This helps robots detect multi-scale obstacles and analyse object occlusions in scenarios such as autonomous driving. Recent advances show that implicit representation learning can be…

Cited by 30SourcecodeScholar
2023

Mind the Gap: Offline Policy Optimization for Imperfect Rewards

ICLR 2023poster

Reward function is essential in reinforcement learning (RL), serving as the guiding signal to incentivize agents to solve given tasks, however, is also notoriously difficult to design. In many cases, only imperfect rewards are available, which inflicts substantial performance loss for RL agents. In…

2023

When Data Geometry Meets Deep Function: Generalizing Offline Reinforcement Learning

ICLR 2023poster

In offline reinforcement learning (RL), one detrimental issue to policy learning is the error accumulation of deep \textit{Q} function in out-of-distribution (OOD) areas. Unfortunately, existing offline RL methods are often over-conservative, inevitably hurting generalization performance outside dat…

2022

PQ-Transformer: Jointly Parsing 3D Objects and Layouts From Point Clouds

RA-L 2022

3D scene understanding from point clouds plays a vital role for various robotic applications. Unfortunately, current state-of-the-art methods use separate neural networks for different tasks like object detection or room layout estimation. Such a scheme has two limitations: 1) Storing and running se

Cited by 41SourcecodeScholar
2022

TOIST: Task Oriented Instance Segmentation Transformer with Noun-Pronoun Distillation

NeurIPS 2022accept

Current referring expression comprehension algorithms can effectively detect or segment objects indicated by nouns, but how to understand verb reference is still under-explored. As such, we study the challenging problem of task oriented detection, which aims to find objects that best afford an actio…