← Search

Wenhao Gao

10 accepted papers

2026

DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Models

ICLR 2026poster

We study inference-time scaling for diffusion models, where the goal is to adapt a pre-trained model to new target distributions without retraining. Existing guidance-based methods are simple but introduce bias, while particle-based corrections suffer from weight degeneracy and high computational co…

Cited by 0SourcecodeScholar
2026

EntroKV: Entropy-Guided Dynamic Budget Allocation for KV-Cache Compression

ICML 2026spotlight

The prohibitive memory footprint of the Key-Value (KV) cache imposes a critical bottleneck for efficient long-context LLM serving. Current compression techniques typically rely on static or uniform budget allocation, overlooking the significant heterogeneity in information density across attention h…

Cited by 0SourceScholar
2025

Efficient Evolutionary Search Over Chemical Space with Large Language Models

ICLR 2025poster

Molecular discovery, when formulated as an optimization problem, presents significant computational challenges because optimization objectives can be non-differentiable. Evolutionary Algorithms (EAs), often used to optimize black-box objectives in molecular discovery, traverse chemical space by perf…

2024

Double-Ended Synthesis Planning with Goal-Constrained Bidirectional Search

NeurIPS 2024spotlight

Computer-aided synthesis planning (CASP) algorithms have demonstrated expert-level abilities in planning retrosynthetic routes to molecules of low to moderate complexity. However, current search methods assume the sufficiency of reaching arbitrary building blocks, failing to address the common real-…

2024

Projecting Molecules into Synthesizable Chemical Spaces

ICML 2024poster

Discovering new drug molecules is a pivotal yet challenging process due to the near-infinitely large chemical space and notorious demands on time and resources. Numerous generative models have recently been introduced to accelerate the drug discovery process, but their progression to experimental va…

2022

Amortized Tree Generation for Bottom-up Synthesis Planning and Synthesizable Molecular Design

ICLR 2022spotlight

Molecular design and synthesis planning are two critical steps in the process of molecular discovery that we propose to formulate as a single shared task of conditional synthetic pathway generation. We report an amortized approach to generate synthetic pathways as a Markov decision process condition…

2022

Differentiable Scaffolding Tree for Molecule Optimization

ICLR 2022poster

The structural design of functional molecules, also called molecular optimization, is an essential chemical science and engineering task with important applications, such as drug discovery. Deep generative models and combinatorial optimization methods achieve initial success but still struggle with…

2022

Reinforced Genetic Algorithm for Structure-based Drug Design

NeurIPS 2022accept

Structure-based drug design (SBDD) aims to discover drug candidates by finding molecules (ligands) that bind tightly to a disease-related protein (targets), which is the primary approach to computer-aided drug discovery. Recently, applying deep generative models for three-dimensional (3D) molecular…

2022

Sample Efficiency Matters: A Benchmark for Practical Molecular Optimization

NeurIPS 2022accept

Molecular optimization is a fundamental goal in the chemical sciences and is of central interest to drug and material design. In recent years, significant progress has been made in solving challenging problems across various aspects of computational molecular optimizations, emphasizing high validity…

2021

Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Development

NeurIPS 2021poster

Therapeutics machine learning is an emerging field with incredible opportunities for innovation and impact. However, advancement in this field requires the formulation of meaningful tasks and careful curation of datasets. Here, we introduce Therapeutics Data Commons (TDC), the first unifying platfor…

Cited by 354SourcecodeScholar