← Search

Zichen Wang

28 accepted papers

2026

DreamSwapV: Mask-guided Subject Swapping for Any Customized Video Editing

ICLR 2026poster

With the rapid progress of video generation, demand for customized video editing is surging, where subject swapping constitutes a key component yet remains under-explored. Prevailing swapping approaches either specialize in narrow domains—such as human-body animation or hand-object interaction—or re…

Cited by 0SourceScholar
2026

SITA: A Framework for Structure-to-Instance Theorem Autoformalization

AAAI 2026technical

While large language models (LLMs) have shown progress in mathematical reasoning, they still face challenges in formalizing theorems that arise from instantiating abstract structures in concrete settings. With the goal of auto-formalizing mathematical results at the research level, we develop a fram

Cited by 0SourcePDFScholar
2026

VCWorld: A Biological World Model for Virtual Cell Simulation

ICLR 2026poster

Virtual cell modeling aims to predict cellular responses to perturbations. Existing virtual cell models rely heavily on large-scale single-cell datasets, learning explicit mappings between gene expression and perturbations. Although recent models attempt to incorporate multi-source biological inform…

Cited by 0SourcecodeScholar
2025

Accurate Differential Operators for Hybrid Neural Fields

CVPR 2025poster

Neural fields have become widely used in various fields, from shape representation to neural rendering, and for solving partial differential equations (PDEs). With the advent of hybrid neural field representations like Instant NGP that leverage small MLPs and explicit representations, these models t…

2025

AffinityFlow: Guided Flows for Antibody Affinity Maturation

ICML 2025poster

Antibodies are widely used as therapeutics, but their development requires costly affinity maturation, involving iterative mutations to enhance binding affinity. This paper explores a sequence-only scenario for affinity maturation, using solely antibody and antigen sequences. Recently AlphaFlow wrap…

Cited by 0SourcePDFScholar
2025

C2F-TP: A Coarse-to-Fine Denoising Framework for Uncertainty-Aware Trajectory Prediction

AAAI 2025technical

Accurately predicting the trajectory of vehicles is critically important for ensuring safety and reliability in autonomous driving. Although considerable research efforts have been made recently, the inherent trajectory uncertainty caused by various factors including the dynamic driving intends and…

2025

Can't Slow Me Down: Learning Robust and Hardware-Adaptive Object Detectors against Latency Attacks for Edge Devices

CVPR 2025poster

Object detection is a fundamental enabler for many real-time downstream applications such as autonomous driving, augmented reality and supply chain management. However, the algorithmic backbone of neural networks is brittle to imperceptible perturbations in the system inputs, which were generally kn…

2025

Design-Based Bandits Under Network Interference: Trade-Off Between Regret and Statistical Inference

NeurIPS 2025poster

In multi-armed bandits with network interference (MABNI), the action taken by one node can influence the rewards of others, creating complex interdependence. While existing research on MABNI largely concentrates on minimizing regret, it often overlooks the crucial concern that an excessive emphasis…

Cited by 0SourceScholar
2025

Embedding Domain Knowledge for Large Language Models via Reinforcement Learning from Augmented Generation

EMNLP 2025

Large language models (LLMs) often exhibit limited performance on domain-specific tasks due to the natural disproportionate representation of specialized information in their training data and the static nature of these datasets. Knowledge scarcity and temporal lag create knowledge gaps for domain a

2025

Fed-DFA: Federated Distillation for Heterogeneous Model Fusion Through the Adversarial Lens

AAAI 2025technical

Most of the federated learning techniques are limited to homogeneous model fusion. With the rapid growth of smart applications on resource-constrained edge devices, it becomes a barrier to accommodate their heterogeneous computing power and memory in the real world. Federated Distillation is a promi…

Cited by 1SourcePDFScholar
2025

Protein Structure Tokenization: Benchmarking and New Recipe

ICML 2025poster

Recent years have witnessed a surge in the development of protein structural tokenization methods, which chunk protein 3D structures into discrete or continuous representations. Structure tokenization enables the direct application of powerful techniques like language modeling for protein structures…

2025

Provably Efficient Algorithm for Best Scoring Rule Identification in Online Principal-Agent Information Acquisition

ICML 2025poster

We investigate the problem of identifying the optimal scoring rule within the principal-agent framework for online information acquisition problem. We focus on the principal's perspective, seeking to determine the desired scoring rule through interactions with the agent. To address this challenge, w…

Cited by 0SourcePDFScholar
2025

Pushing the Limits of All-Atom Geometric Graph Neural Networks: Pre-Training, Scaling, and Zero-Shot Transfer

ICLR 2025poster

The ability to construct transferable descriptors for molecular and biological systems has broad applications in drug discovery, molecular dynamics, and protein analysis. Geometric graph neural networks (Geom-GNNs) utilizing all-atom information have revolutionized atomistic simulations by enabling…

Cited by 3SourcePDFScholar
2025

Retrieval Augmented Diffusion Model for Structure-informed Antibody Design and Optimization

ICLR 2025poster

Antibodies are essential proteins responsible for immune responses in organisms, capable of specifically recognizing antigen molecules of pathogens. Recent advances in generative models have significantly enhanced rational antibody design. However, existing methods mainly create antibodies from scra…

Cited by 2SourcePDFScholar
2025

RiboFlow: Conditional De Novo RNA Co-Design via Synergistic Flow Matching

NeurIPS 2025poster

Ribonucleic acid (RNA) binds to molecules to achieve specific biological functions. While generative models are advancing biomolecule design, existing methods for designing RNA that target specific ligands face limitations in capturing RNA’s conformational flexibility, ensuring structural validity,…

Cited by 0SourceScholar
2025

Toward Engineering AGI: Benchmarking the Engineering Design Capabilities of LLMs

NeurIPS 2025poster

Modern engineering, spanning electrical, mechanical, aerospace, civil, and computer disciplines, stands as a cornerstone of human civilization and the foundation of our society. However, engineering design poses a fundamentally different challenge for large language models (LLMs) compared with tradi…

Cited by 0SourceScholar
2024

BioBridge: Bridging Biomedical Foundation Models via Knowledge Graphs

ICLR 2024poster

Foundation models (FMs) learn from large volumes of unlabeled data to demonstrate superior performance across a wide range of tasks. However, FMs developed for biomedical domains have largely remained unimodal, i.e., independently trained and used for tasks on protein sequences alone, small molecule…

2024

DiffusionPDE: Generative PDE-Solving under Partial Observation

NeurIPS 2024poster

We introduce a general framework for solving partial differential equations (PDEs) using generative diffusion models. In particular, we focus on the scenarios where we do not have the full knowledge of the scene necessary to apply classical solvers. Most existing forward or inverse PDE approaches pe…

2024

Fine-Grained Prototypes Distillation for Few-Shot Object Detection

AAAI 2024technical

Few-shot object detection (FSOD) aims at extending a generic detector for novel object detection with only a few training examples. It attracts great concerns recently due to the practical meanings. Meta-learning has been demonstrated to be an effective paradigm for this task. In general, methods ba…

2024

Graph Neural Prompting with Large Language Models

AAAI 2024technical

Large language models (LLMs) have shown remarkable generalization capability with exceptional performance in various language modeling tasks. However, they still exhibit inherent limitations in precisely capturing and returning grounded knowledge. While existing work has explored utilizing knowledge…

2024

Learn How to See: Collaborative Embodied Learning for Object Detection and Camera Adjusting

AAAI 2024technical

Passive object detectors, trained on large-scale static datasets, often overlook the feedback from object detection to image acquisition. Embodied vision and active detection mitigate this issue by interacting with the environment. Nevertheless, the materialization of activeness hinges on resource-i…

2024

Pure Exploration in Asynchronous Federated Bandits

UAI 2024poster

We study the federated pure exploration problem of multi-armed bandits and linear bandits, where $M$ agents cooperatively identify the best arm via communicating with the central server. To enhance the robustness against latency and unavailability of agents that are common in practice, we propose th…

Cited by 1SourcePDFScholar
2024

Simultaneous Optimization of Bid Shading and Internal Auction for Demand-Side Platforms

AAAI 2024technical

Online advertising has been one of the most important sources for industry's growth, where the demand-side platforms (DSP) play an important role via bidding to the ad exchanges on behalf of their advertiser clients. Since more and more ad exchanges have shifted from second to first price auctions,…

Cited by 5SourcePDFScholar
2024

UMG-CLIP: A Unified Multi-Granularity Vision Generalist for Open-World Understanding

ECCV 2024poster

"Vision-language foundation models, represented by Contras-tive Language-Image Pre-training (CLIP), have gained increasing attention for jointly understanding both vision and textual tasks. However, existing approaches primarily focus on training models to match global image representations with tex…

2023

Predicting Cellular Responses with Variational Causal Inference and Refined Relational Information

ICLR 2023poster

Predicting the responses of a cell under perturbations may bring important benefits to drug discovery and personalized therapeutics. In this work, we propose a novel graph variational Bayesian causal inference framework to predict a cell's gene expressions under counterfactual perturbations (perturb…

2022

Multi-Dimensional Proprioception and Stiffness Tuning for Soft Robotic Joints

ICRA 2022poster

Proprioception and variable stiffness are two trending topics in soft robotics research. The former could endow soft robots with the ability to perceive the environment as well as their internal states without the need of dedicated sensors, while the latter could strengthen the otherwise excessive c…

Cited by 5SourceScholar