← Search

Wei-Ying Ma

45 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

DCFold: Efficient Protein Structure Generation with Single Forward Pass

ICLR 2026oral

AlphaFold3 introduces a diffusion-based architecture that elevates protein structure prediction to all-atom resolution with improved accuracy. This state-of-the-art performance has established AlphaFold3 as a foundation model for diverse generation and design tasks. However, its iterative design sub…

Cited by 0SourceScholar
2026

Demystifying Multimodal Biomolecular Co-design With Intrinsic Geodesic Coupling

ICML 2026poster

Biomolecules such as proteins and small-molecule ligands play a central role in biological systems, arising from the tight interplay between sequence and three-dimensional structure. Recent generative models for biomolecular co-design aim to capture this interplay by jointly modeling coupled modalit…

Cited by 0SourceScholar
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

MoMa: A Simple Modular Learning Framework for Material Property Prediction

ICLR 2026poster

Deep learning methods for material property prediction have been widely explored to advance materials discovery. However, the prevailing pre-train paradigm often fails to address the inherent diversity and disparity of material tasks. To overcome these challenges, we introduce MoMa, a simple Modular…

Cited by 0SourceScholar
2026

MolAlign3D: Enhancing Fixed-Dimensional E(3)-Equivariant Latent Space for High-Fidelity 3D Molecular Reconstruction and Editing

ICML 2026poster

Recent advances in 3D molecular modeling have achieved high-fidelity structural synthesis, yet these models often lack an explicit and manipulable representation space. To address this, MolFLAE introduced a fixed-dimensional, E(3)-equivariant latent space, providing a novel framework for molecular e…

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
2025

A Periodic Bayesian Flow for Material Generation

ICLR 2025spotlight

Generative modeling of crystal data distribution is an important yet challenging task due to the unique periodic physical symmetry of crystals. Diffusion-based methods have shown early promise in modeling crystal distribution. More recently, Bayesian Flow Networks were introduced to aggregate noisy…

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

Accelerating 3D Molecule Generative Models with Trajectory Diagnosis

NeurIPS 2025poster

Geometric molecule generative models have found expanding applications across various scientific domains, but their generation inefficiency has become a critical bottleneck. Through a systematic investigation of the generative trajectory, we discover a unique challenge for molecule geometric graph g…

Cited by 0SourceScholar
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

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

Empower Structure-Based Molecule Optimization with Gradient Guided Bayesian Flow Networks

ICML 2025poster

Structure-based molecule optimization (SBMO) aims to optimize molecules with both continuous coordinates and discrete types against protein targets. A promising direction is to exert gradient guidance on generative models given its remarkable success in images, but it is challenging to guide discret…

2025

FIGRDock: Fast Interaction-Guided Regression for Flexible Docking

NeurIPS 2025poster

Flexible docking, which predicts the binding conformations of both proteins and small molecules by modeling their structural flexibility, plays a vital role in structure-based drug design. Although recent generative approaches, particularly diffusion-based models, have shown promising results, they…

Cited by 0SourceScholar
2025

MOF-BFN: Metal-Organic Frameworks Structure Prediction via Bayesian Flow Networks

NeurIPS 2025poster

Metal-Organic Frameworks (MOFs) have attracted considerable attention due to their unique properties including high surface area and tunable porosity, and promising applications in catalysis, gas storage, and drug delivery. Structure prediction for MOFs is a challenging task, as these frameworks are…

Cited by 0SourceScholar
2025

Manipulating 3D Molecules in a Fixed-Dimensional E(3)-Equivariant Latent Space

NeurIPS 2025poster

Medicinal chemists often optimize drugs considering their 3D structures and designing structurally distinct molecules that retain key features, such as shapes, pharmacophores, or chemical properties. Previous deep learning approaches address this through supervised tasks like molecule inpainting or…

Cited by 0SourceScholar
2025

Piloting Structure-Based Drug Design via Modality-Specific Optimal Schedule

ICML 2025poster

Structure-Based Drug Design (SBDD) is crucial for identifying bioactive molecules. Recent deep generative models are faced with challenges in geometric structure modeling. A major bottleneck lies in the twisted probability path of multi-modalities—continuous 3D positions and discrete 2D topologies—w…

2025

Rationalized All-Atom Protein Design with Unified Multi-Modal Bayesian Flow

NeurIPS 2025poster

Designing functional proteins is a critical yet challenging problem due to the intricate interplay between backbone structures, sequences, and side-chains. Current approaches often decompose protein design into separate tasks, which can lead to accumulated errors, while recent efforts increasingly f…

Cited by 0SourceScholar
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

Retro-R1: LLM-based Agentic Retrosynthesis

NeurIPS 2025poster

Retrosynthetic planning is a fundamental task in chemical discovery. Due to the vast combinatorial search space, identifying viable synthetic routes remains a significant challenge--even for expert chemists. Recent advances in Large Language Models (LLMs), particularly equipped with reinforcement le…

Cited by 0SourceScholar
2025

RetroDiff: Retrosynthesis as Multi-stage Distribution Interpolation

AISTATS 2025poster

Retrosynthesis poses a key challenge in biopharmaceuticals, aiding chemists in finding appropriate reactant molecules for given product molecules. With reactants and products represented as 2D graphs, retrosynthesis constitutes a conditional graph-to-graph (G2G) generative task. Inspired by advancem…

Cited by 0SourceScholar
2025

ShortListing Model: A Streamlined Simplex Diffusion for Discrete Variable Generation

NeurIPS 2025poster

Generative modeling of discrete variables is challenging yet crucial for applications in natural language processing and biological sequence design. We introduce the Shortlisting Model (SLM), a novel simplex-based diffusion model inspired by progressive candidate pruning. SLM operates on simplex cen…

Cited by 0SourcecodeScholar
2025

Smooth Interpolation for Improved Discrete Graph Generative Models

ICML 2025poster

Though typically represented by the discrete node and edge attributes, the graph topological information can be sufficiently captured by the graph spectrum in a continuous space. It is believed that incorporating the continuity of graph topological information into the generative process design coul…

Cited by 0SourcePDFScholar
2025

Steering Protein Family Design through Profile Bayesian Flow

ICLR 2025oral

Protein family design emerges as a promising alternative by combining the advantages of de novo protein design and mutation-based directed evolution.In this paper, we propose ProfileBFN, the Profile Bayesian Flow Networks, for specifically generative modeling of protein families. ProfileBFN extends…

Cited by 0SourcePDFScholar
2025

Straight-Line Diffusion Model for Efficient 3D Molecular Generation

NeurIPS 2025poster

Diffusion-based models have shown great promise in molecular generation but often require a large number of sampling steps to generate valid samples. In this paper, we introduce a novel Straight-Line Diffusion Model (SLDM) to tackle this problem, by formulating the diffusion process to follow a line…

Cited by 0SourcecodeScholar
2025

UniGEM: A Unified Approach to Generation and Property Prediction for Molecules

ICLR 2025poster

Molecular generation and molecular property prediction are both crucial for drug discovery, but they are often developed independently. Inspired by recent studies, which demonstrate that diffusion model, a prominent generative approach, can learn meaningful data representations that enhance predicti…

Cited by 2SourcePDFScholar
2024

ESM All-Atom: Multi-Scale Protein Language Model for Unified Molecular Modeling

ICML 2024poster

Protein language models have demonstrated significant potential in the field of protein engineering. However, current protein language models primarily operate at the residue scale, which limits their ability to provide information at the atom level. This limitation prevents us from fully exploiting…

2024

Mol-AE: Auto-Encoder Based Molecular Representation Learning With 3D Cloze Test Objective

ICML 2024poster

3D molecular representation learning has gained tremendous interest and achieved promising performance in various downstream tasks. A series of recent approaches follow a prevalent framework: an encoder-only model coupled with a coordinate denoising objective. However, through a series of analytical…

Cited by 7SourcePDFScholar
2024

MolCRAFT: Structure-Based Drug Design in Continuous Parameter Space

ICML 2024poster

Generative models for structure-based drug design (SBDD) have shown promising results in recent years. Existing works mainly focus on how to generate molecules with higher binding affinity, ignoring the feasibility prerequisites for generated 3D poses and resulting in *false positives*. We conduct t…

2024

Protein-ligand binding representation learning from fine-grained interactions

ICLR 2024poster

The binding between proteins and ligands plays a crucial role in the realm of drug discovery. Previous deep learning approaches have shown promising results over traditional computationally intensive methods, but resulting in poor generalization due to limited supervised data. In this paper, we prop…

Cited by 11SourcePDFScholar
2024

Rethinking Specificity in SBDD: Leveraging Delta Score and Energy-Guided Diffusion

ICML 2024poster

In the field of Structure-based Drug Design (SBDD), deep learning-based generative models have achieved outstanding performance in terms of docking score. However, further study shows that the existing molecular generative methods and docking scores both have lacked consideration in terms of specifi…

Cited by 4SourcePDFScholar
2024

Self-supervised Pocket Pretraining via Protein Fragment-Surroundings Alignment

ICLR 2024poster

Pocket representations play a vital role in various biomedical applications, such as druggability estimation, ligand affinity prediction, and de novo drug design. While existing geometric features and pretrained representations have demonstrated promising results, they usually treat pockets independ…

Cited by 12SourcePDFScholar
2024

Sliced Denoising: A Physics-Informed Molecular Pre-Training Method

ICLR 2024poster

While molecular pre-training has shown great potential in enhancing drug discovery, the lack of a solid physical interpretation in current methods raises concerns about whether the learned representation truly captures the underlying explanatory factors in observed data, ultimately resulting in limi…

Cited by 14SourcePDFScholar
2024

UniCorn: A Unified Contrastive Learning Approach for Multi-view Molecular Representation Learning

ICML 2024poster

Recently, a noticeable trend has emerged in developing pre-trained foundation models in the domains of CV and NLP. However, for molecular pre-training, there lacks a universal model capable of effectively applying to various categories of molecular tasks, since existing prevalent pre-training method…

Cited by 11SourcePDFScholar
2024

Unified Generative Modeling of 3D Molecules with Bayesian Flow Networks

ICLR 2024oral

Advanced generative model (\textit{e.g.}, diffusion model) derived from simplified continuity assumptions of data distribution, though showing promising progress, has been difficult to apply directly to geometry generation applications due to the \textit{multi-modality} and \textit{noise-sensitive}…

Cited by 26SourcePDFScholar
2023

Coarse-to-Fine: a Hierarchical Diffusion Model for Molecule Generation in 3D

ICML 2023poster

Generating desirable molecular structures in 3D is a fundamental problem for drug discovery. Despite the considerable progress we have achieved, existing methods usually generate molecules in atom resolution and ignore intrinsic local structures such as rings, which leads to poor quality in generate…

2023

DrugCLIP: Contrastive Protein-Molecule Representation Learning for Virtual Screening

NeurIPS 2023poster

Virtual screening, which identifies potential drugs from vast compound databases to bind with a particular protein pocket, is a critical step in AI-assisted drug discovery. Traditional docking methods are highly time-consuming, and can only work with a restricted search library in real-life applicat…

Cited by 51SourcePDFScholar
2023

Equivariant Flow Matching with Hybrid Probability Transport for 3D Molecule Generation

NeurIPS 2023poster

The generation of 3D molecules requires simultaneously deciding the categorical features (atom types) and continuous features (atom coordinates). Deep generative models, especially Diffusion Models (DMs), have demonstrated effectiveness in generating feature-rich geometries. However, existing DMs ty…

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
2020

Controllable Person Image Synthesis With Attribute-Decomposed GAN

CVPR 2020oral

This paper introduces the Attribute-Decomposed GAN, a novel generative model for controllable person image synthesis, which can produce realistic person images with desired human attributes (e.g., pose, head, upper clothes and pants) provided in various source inputs. The core idea of the proposed m…

Cited by 309PDFScholar
2019

Unified Visual-Semantic Embeddings: Bridging Vision and Language With Structured Meaning Representations

CVPR 2019oral

We propose the Unified Visual-Semantic Embeddings (Unified VSE) for learning a joint space of visual representation and textual semantics. The model unifies the embeddings of concepts at different levels: objects, attributes, relations, and full scenes. We view the sentential semantics as a combinat…

Cited by 221PDFcodeScholar
2016

Dual Learning for Machine Translation

NeurIPS 2016poster

While neural machine translation (NMT) is making good progress in the past two years, tens of millions of bilingual sentence pairs are needed for its training. However, human labeling is very costly. To tackle this training data bottleneck, we develop a dual-learning mechanism, which can enable an N…