← Search

Wengong Jin

19 accepted papers

2025

Boltzmann-Aligned Inverse Folding Model as a Predictor of Mutational Effects on Protein-Protein Interactions

ICLR 2025spotlight

Predicting the change in binding free energy ($\Delta \Delta G$) is crucial for understanding and modulating protein-protein interactions, which are critical in drug design. Due to the scarcity of experimental $\Delta\Delta G$ data, existing methods focus on pre-training, while neglecting the impo…

2025

Repurposing AlphaFold3-like Protein Folding Models for Antibody Sequence and Structure Co-design

NeurIPS 2025poster

Diffusion models hold great potential for accelerating antibody design, but their performance is so far limited by the number of antibody-antigen complexes used for model training. Meanwhile, AlphaFold3-like protein folding models, pre-trained on a large corpus of crystal structures, have acquired a…

Cited by 3SourceScholar
2025

Scalable Generation of Spatial Transcriptomics from Histology Images via Whole-Slide Flow Matching

ICML 2025spotlight

Spatial transcriptomics (ST) has emerged as a powerful technology for bridging histology imaging with gene expression profiling. However, its application has been limited by low throughput and the need for specialized experimental facilities. Prior works sought to predict ST from whole-slide histolo…

Cited by 0SourcePDFScholar
2024

Generative Enzyme Design Guided by Functionally Important Sites and Small-Molecule Substrates

ICML 2024poster

Enzymes are genetically encoded biocatalysts capable of accelerating chemical reactions. How can we automatically design functional enzymes? In this paper, we propose EnzyGen, an approach to learn a unified model to design enzymes across all functional families. Our key idea is to generate an enzyme…

2024

Protein-Nucleic Acid Complex Modeling with Frame Averaging Transformer

NeurIPS 2024poster

Nucleic acid-based drugs like aptamers have recently demonstrated great therapeutic potential. However, experimental platforms for aptamer screening are costly, and the scarcity of labeled data presents a challenge for supervised methods to learn protein-aptamer binding. To this end, we develop an u…

2024

RNAFlow: RNA Structure & Sequence Design via Inverse Folding-Based Flow Matching

ICML 2024poster

The growing significance of RNA engineering in diverse biological applications has spurred interest in developing AI methods for structure-based RNA design. While diffusion models have excelled in protein design, adapting them for RNA presents new challenges due to RNA's conformational flexibility a…

2023

Unsupervised Protein-Ligand Binding Energy Prediction via Neural Euler's Rotation Equation

NeurIPS 2023poster

Protein-ligand binding prediction is a fundamental problem in AI-driven drug discovery. Previous work focused on supervised learning methods for small molecules where binding affinity data is abundant, but it is hard to apply the same strategy to other ligand classes like antibodies where labelled d…

2022

Antibody-Antigen Docking and Design via Hierarchical Structure Refinement

ICML 2022spotlight

Computational antibody design seeks to automatically create an antibody that binds to an antigen. The binding affinity is governed by the 3D binding interface where antibody residues (paratope) closely interact with antigen residues (epitope). Thus, the key question of antibody design is how to pred…

2022

Iterative Refinement Graph Neural Network for Antibody Sequence-Structure Co-design

ICLR 2022spotlight

Antibodies are versatile proteins that bind to pathogens like viruses and stimulate the adaptive immune system. The specificity of antibody binding is determined by complementarity-determining regions (CDRs) at the tips of these Y-shaped proteins. In this paper, we propose a generative model to auto…

Cited by 160SourcePDFScholar
2021

Mol2Image: Improved Conditional Flow Models for Molecule to Image Synthesis

CVPR 2021poster

In this paper, we aim to synthesize cell microscopy images under different molecular interventions, motivated by practical applications to drug development. Building on the recent success of graph neural networks for learning molecular embeddings and flow-based models for image generation, we propos…

Cited by 14PDFScholar
2020

Hierarchical Generation of Molecular Graphs using Structural Motifs

ICML 2020poster

Graph generation techniques are increasingly being adopted for drug discovery. Previous graph generation approaches have utilized relatively small molecular building blocks such as atoms or simple cycles, limiting their effectiveness to smaller molecules. Indeed, as we demonstrate, their performance…

2020

Improving Molecular Design by Stochastic Iterative Target Augmentation

ICML 2020poster

Generative models in molecular design tend to be richly parameterized, data-hungry neural models, as they must create complex structured objects as outputs. Estimating such models from data may be challenging due to the lack of sufficient training data. In this paper, we propose a surprisingly effec…

2020

Multi-Objective Molecule Generation using Interpretable Substructures

ICML 2020poster

Drug discovery aims to find novel compounds with specified chemical property profiles. In terms of generative modeling, the goal is to learn to sample molecules in the intersection of multiple property constraints. This task becomes increasingly challenging when there are many property constraints.…

2019

Functional Transparency for Structured Data: a Game-Theoretic Approach

ICML 2019oral

We provide a new approach to training neural models to exhibit transparency in a well-defined, functional manner. Our approach naturally operates over structured data and tailors the predictor, functionally, towards a chosen family of (local) witnesses. The estimation problem is setup as a co-operat…

Cited by 23SourcePDFScholar
2019

Learning Multimodal Graph-to-Graph Translation for Molecule Optimization

ICLR 2019poster

We view molecule optimization as a graph-to-graph translation problem. The goal is to learn to map from one molecular graph to another with better properties based on an available corpus of paired molecules. Since molecules can be optimized in different ways, there are multiple viable translations f…

Cited by 327SourcePDFScholar
2018

Junction Tree Variational Autoencoder for Molecular Graph Generation

ICML 2018oral

We seek to automate the design of molecules based on specific chemical properties. In computational terms, this task involves continuous embedding and generation of molecular graphs. Our primary contribution is the direct realization of molecular graphs, a task previously approached by generating li…

2017

Deriving Neural Architectures from Sequence and Graph Kernels

ICML 2017poster

The design of neural architectures for structured objects is typically guided by experimental insights rather than a formal process. In this work, we appeal to kernels over combinatorial structures, such as sequences and graphs, to derive appropriate neural operations. We introduce a class of deep r…

2017

Predicting Organic Reaction Outcomes with Weisfeiler-Lehman Network

NeurIPS 2017poster

The prediction of organic reaction outcomes is a fundamental problem in computational chemistry. Since a reaction may involve hundreds of atoms, fully exploring the space of possible transformations is intractable. The current solution utilizes reaction templates to limit the space, but it suffers f…