← Search

Jianzhu Ma

42 accepted papers

2026

Can Language Models Discover Scaling Laws?

ICLR 2026poster

Discovering scaling laws for predicting model performance at scale is a fundamental and open-ended challenge, mostly reliant on slow, case specific human experimentation. To investigate the potential for LLMs to automate this process, we collect over 5,000 experiments from existing literature and cu…

Cited by 0SourcecodeScholar
2026

HAIC: Humanoid Agile Object Interaction Control via Dynamics-Aware World Model

RSS 2026poster

Humanoid robots exhibit significant potential for executing complex whole-body interaction tasks in unstructured environments. While recent advancements in Human-Object Interaction (HOI) have been substantial, prevailing methodologies predominantly address the manipulation of fully actuated objects,…

Cited by 0SourceScholar
2026

Inference-time scaling of diffusion models through classical search

ICLR 2026poster

Classical search algorithms have long underpinned modern artificial intelligence. In this work, we tackle the challenge of inference-time control in diffusion models—adapting generated outputs to meet diverse test-time objectives—using principles from classical search. We propose a general framework…

Cited by 0SourcecodeScholar
2025

Designing Cyclic Peptides via Harmonic SDE with Atom-Bond Modeling

ICML 2025poster

Cyclic peptides offer inherent advantages in pharmaceuticals. For example, cyclic peptides are more resistant to enzymatic hydrolysis compared to linear peptides and usually exhibit excellent stability and affinity. Although deep generative models have achieved great success in linear peptide design…

Cited by 0SourcePDFScholar
2025

Group Ligands Docking to Protein Pockets

ICLR 2025poster

Molecular docking is a key task in computational biology that has attracted increasing interest from the machine learning community. While existing methods have achieved success, they generally treat each protein-ligand pair in isolation. Inspired by the biochemical observation that ligands binding…

Cited by 1SourcePDFScholar
2025

Hotspot-Driven Peptide Design via Multi-Fragment Autoregressive Extension

ICLR 2025poster

Peptides, short chains of amino acids, interact with target proteins, making them a unique class of protein-based therapeutics for treating human diseases. Recently, deep generative models have shown great promise in peptide generation. However, several challenges remain in designing effective pepti…

2025

Integrating Protein Dynamics into Structure-Based Drug Design via Full-Atom Stochastic Flows

ICLR 2025poster

The dynamic nature of proteins, influenced by ligand interactions, is essential for comprehending protein function and progressing drug discovery. Traditional structure-based drug design (SBDD) approaches typically target binding sites with rigid structures, limiting their practical application in d…

Cited by 0SourcePDFScholar
2025

Morpheus: A Neural-driven Animatronic Face with Hybrid Actuation and Diverse Emotion Control

RSS 2025poster

Previous animatronic faces struggle to effectively express emotions due to both hardware and software limitations. On the hardware side, earlier approaches either used rigid-driven mechanisms, which provide precise control but are difficult to design within constrained spaces, or tendon-driven mecha…

Cited by 0PDFScholar
2025

PUGS: Zero-Shot Physical Understanding with Gaussian Splatting

ICRA 2025

Current robotic systems can understand the categories and poses of objects well. But understanding physical properties like mass, friction, and hardness, in the wild, remains challenging. We propose a new method that reconstructs 3D objects using the Gaussian splatting representation and predicts va

Cited by 11SourcecodeScholar
2025

Safety-Polarized and Prioritized Reinforcement Learning

ICML 2025poster

Motivated by the first priority of safety in many real-world applications, we propose \textsc{MaxSafe}, a chance-constrained bi-level optimization framework for safe reinforcement learning. \textsc{MaxSafe} first minimizes the unsafe probability and then maximizes the return among the safest policie…

Cited by 0SourcePDFScholar
2025

TFG-Flow: Training-free Guidance in Multimodal Generative Flow

ICLR 2025poster

Given an unconditional generative model and a predictor for a target property (e.g., a classifier), the goal of training-free guidance is to generate samples with desirable target properties without additional training. As a highly efficient technique for steering generative models toward flexible o…

2025

UniMoMo: Unified Generative Modeling of 3D Molecules for De Novo Binder Design

ICML 2025poster

The design of target-specific molecules such as small molecules, peptides, and antibodies is vital for biological research and drug discovery. Existing generative methods are restricted to single-domain molecules, failing to address versatile therapeutic needs or utilize cross-domain transferability…

2025

Zero-Shot Cyclic Peptide Design via Composable Geometric Constraints

ICML 2025poster

Cyclic peptides, characterized by geometric constraints absent in linear peptides, offer enhanced biochemical properties, presenting new opportunities to address unmet medical needs. However, designing target-specific cyclic peptides remains underexplored due to limited training data. To bridge the…

Cited by 0SourcePDFScholar
2024

Enhancing Protein Mutation Effect Prediction through a Retrieval-Augmented Framework

NeurIPS 2024poster

Predicting the effects of protein mutations is crucial for analyzing protein functions and understanding genetic diseases. However, existing models struggle to effectively extract mutation-related local structure motifs from protein databases, which hinders their predictive accuracy and robustness.…

Cited by 1SourcePDFScholar
2024

FAFE: Immune Complex Modeling with Geodesic Distance Loss on Noisy Group Frames

ICML 2024spotlight

Despite the striking success of general protein folding models such as AlphaFold2 (AF2), the accurate computational modeling of antibody-antigen complexes remains a challenging task. In this paper, we first analyze AF2's primary loss function, known as the Frame Aligned Point Error (FAPE), and raise…

Cited by 1SourcePDFScholar
2024

Full-Atom Peptide Design based on Multi-modal Flow Matching

ICML 2024poster

Peptides, short chains of amino acid residues, play a vital role in numerous biological processes by interacting with other target molecules, offering substantial potential in drug discovery. In this work, we present *PepFlow*, the first multi-modal deep generative model grounded in the flow-matchin…

2024

InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generation

ICLR 2024poster

Diffusion models have revolutionized text-to-image generation with its exceptional quality and creativity. However, its multi-step sampling process is known to be slow, often requiring tens of inference steps to obtain satisfactory results. Previous attempts to improve its sampling speed and reduce…

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…

2024

Reprogramming Pretrained Target-Specific Diffusion Models for Dual-Target Drug Design

NeurIPS 2024poster

Dual-target therapeutic strategies have become a compelling approach and attracted significant attention due to various benefits, such as their potential in overcoming drug resistance in cancer therapy. Considering the tremendous success that deep generative models have achieved in structure-based d…

2024

Selecting Large Language Model to Fine-tune via Rectified Scaling Law

ICML 2024poster

The ever-growing ecosystem of LLMs has posed a challenge in selecting the most appropriate pre-trained model to fine-tune amidst a sea of options. Given constrained resources, fine-tuning all models and making selections afterward is unrealistic. In this work, we formulate this resource-constrained…

2024

TFG: Unified Training-Free Guidance for Diffusion Models

NeurIPS 2024spotlight

Given an unconditional diffusion model and a predictor for a target property of interest (e.g., a classifier), the goal of training-free guidance is to generate samples with desirable target properties without additional training. Existing methods, though effective in various individual applications…

2023

3D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction

ICLR 2023poster

Rich data and powerful machine learning models allow us to design drugs for a specific protein target <em>in silico</em>. Recently, the inclusion of 3D structures during targeted drug design shows superior performance to other target-free models as the atomic interaction in the 3D space is explicitl…

2023

Boosting the Cycle Counting Power of Graph Neural Networks with I$^2$-GNNs

ICLR 2023poster

Message Passing Neural Networks (MPNNs) are a widely used class of Graph Neural Networks (GNNs). The limited representational power of MPNNs inspires the study of provably powerful GNN architectures. However, knowing one model is more powerful than another gives little insight about what functions t…

2023

DecompDiff: Diffusion Models with Decomposed Priors for Structure-Based Drug Design

ICML 2023poster

Designing 3D ligands within a target binding site is a fundamental task in drug discovery. Existing structured-based drug design methods treat all ligand atoms equally, which ignores different roles of atoms in the ligand for drug design and can be less efficient for exploring the large drug-like mo…

2023

Learning Sparse Group Models Through Boolean Relaxation

ICLR 2023top-25%

We introduce an efficient algorithmic framework for learning sparse group models formulated as the natural convex relaxation of a cardinality-constrained program with Boolean variables. We provide theoretical techniques to characterize the equivalent condition when the relaxation achieves the exact…

Cited by 0SourcePDFScholar
2023

LinkerNet: Fragment Poses and Linker Co-Design with 3D Equivariant Diffusion

NeurIPS 2023spotlight

Targeted protein degradation techniques, such as PROteolysis TArgeting Chimeras (PROTACs), have emerged as powerful tools for selectively removing disease-causing proteins. One challenging problem in this field is designing a linker to connect different molecular fragments to form a stable drug-cand…

2023

MolDiff: Addressing the Atom-Bond Inconsistency Problem in 3D Molecule Diffusion Generation

ICML 2023poster

Deep generative models have recently achieved superior performance in 3D molecule generation. Most of them first generate atoms and then add chemical bonds based on the generated atoms in a post-processing manner. However, there might be no corresponding bond solution for the temporally generated at…

2023

ProBio: A Protocol-guided Multimodal Dataset for Molecular Biology Lab

NeurIPS 2023poster

The challenge of replicating research results has posed a significant impediment to the field of molecular biology. The advent of modern intelligent systems has led to notable progress in various domains. Consequently, we embarked on an investigation of intelligent monitoring systems as a means of t…

Cited by 3SourcePDFScholar
2023

Rotamer Density Estimator is an Unsupervised Learner of the Effect of Mutations on Protein-Protein Interaction

ICLR 2023poster

Protein-protein interactions are crucial to many biological processes, and predicting the effect of amino acid mutations on binding is important for protein engineering. While data-driven approaches using deep learning have shown promise, the scarcity of annotated experimental data remains a major c…

Cited by 18SourcePDFScholar
2023

xTrimoGene: An Efficient and Scalable Representation Learner for Single-Cell RNA-Seq Data

NeurIPS 2023poster

Advances in high-throughput sequencing technology have led to significant progress in measuring gene expressions at the single-cell level. The amount of publicly available single-cell RNA-seq (scRNA-seq) data is already surpassing 50M records for humans with each record measuring 20,000 genes. This…

Cited by 28SourcePDFScholar
2022

3DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker Design

ICML 2022oral

Deep learning has achieved tremendous success in designing novel chemical compounds with desirable pharmaceutical properties. In this work, we focus on a new type of drug design problem — generating a small “linker” to physically attach two independent molecules with their distinct functions. The ma…

2022

Antigen-Specific Antibody Design and Optimization with Diffusion-Based Generative Models for Protein Structures

NeurIPS 2022accept

Antibodies are immune system proteins that protect the host by binding to specific antigens such as viruses and bacteria. The binding between antibodies and antigens is mainly determined by the complementarity-determining regions (CDR) of the antibodies. In this work, we develop a deep generative mo…

Cited by 244SourcePDFScholar
2022

Efficient Meta Reinforcement Learning for Preference-based Fast Adaptation

NeurIPS 2022accept

Learning new task-specific skills from a few trials is a fundamental challenge for artificial intelligence. Meta reinforcement learning (meta-RL) tackles this problem by learning transferable policies that support few-shot adaptation to unseen tasks. Despite recent advances in meta-RL, most existing…

2022

Energy-Inspired Molecular Conformation Optimization

ICLR 2022poster

This paper studies an important problem in computational chemistry: predicting a molecule's spatial atom arrangements, or a molecular conformation. We propose a neural energy minimization formulation that casts the prediction problem into an unrolled optimization process, where a neural network is p…

Cited by 24SourcePDFScholar
2022

Equivariant Point Cloud Analysis via Learning Orientations for Message Passing

CVPR 2022oral

Equivariance has been a long-standing concern in various fields ranging from computer vision to physical modeling. Most previous methods struggle with generality, simplicity, and expressiveness --- some are designed ad hoc for specific data types, some are too complex to be accessible, and some sacr…

Cited by 45PDFcodeScholar
2022

Pocket2Mol: Efficient Molecular Sampling Based on 3D Protein Pockets

ICML 2022spotlight

Deep generative models have achieved tremendous success in designing novel drug molecules in recent years. A new thread of works have shown potential in advancing the specificity and success rate of in silico drug design by considering the structure of protein pockets. This setting posts fundamental…

2022

Proximal Exploration for Model-guided Protein Sequence Design

ICML 2022spotlight

Designing protein sequences with a particular biological function is a long-lasting challenge for protein engineering. Recent advances in machine-learning-guided approaches focus on building a surrogate sequence-function model to reduce the burden of expensive in-lab experiments. In this paper, we s…

Cited by 54SourcePDFScholar
2021

A 3D Generative Model for Structure-Based Drug Design

NeurIPS 2021poster

We study a fundamental problem in structure-based drug design --- generating molecules that bind to specific protein binding sites. While we have witnessed the great success of deep generative models in drug design, the existing methods are mostly string-based or graph-based. They are limited by the…

2021

Fast Projection onto the Capped Simplex with Applications to Sparse Regression in Bioinformatics

NeurIPS 2021poster

We consider the problem of projecting a vector onto the so-called k-capped simplex, which is a hyper-cube cut by a hyperplane. For an n-dimensional input vector with bounded elements, we found that a simple algorithm based on Newton's method is able to solve the projection problem to high precision…

Cited by 11SourcePDFScholar
2021

PALM: Probabilistic area loss Minimization for Protein Sequence Alignment

UAI 2021poster

Protein sequence alignment is a fundamental problem in computational structure biology and popular for protein 3D structural prediction and protein homology detection. Most of the developed programs for detecting protein sequence alignments are based upon the likelihood information of amino acids an…