← Search

Rui Jiao

13 accepted papers

2026

DrugTrail: Explainable Drug Discovery via Structured Reasoning and Druggability‑Tailored Preference Optimization

ICLR 2026poster

Machine learning promises to revolutionize drug discovery, but its "black-box" nature and narrow focus limit adoption by experts. While Large Language Models (LLMs) offer a path forward with their broad knowledge and interactivity, existing methods remain data-intensive and lack transparent reasonin…

Cited by 0SourceScholar
2025

DenoiseVAE: Learning Molecule-Adaptive Noise Distributions for Denoising-based 3D Molecular Pre-training

ICLR 2025poster

Denoising learning of 3D molecules learns molecular representations by imposing noises into the equilibrium conformation and predicting the added noises to recover the equilibrium conformation, which essentially captures the information of molecular force fields. Due to the specificity of Potential…

Cited by 2SourcePDFScholar
2025

Learning 3D Anisotropic Noise Distributions Improves Molecular Force Fields

NeurIPS 2025poster

Coordinate denoising has emerged as a promising method for 3D molecular pretraining due to its theoretical connection to learning molecular force field. However, existing denoising methods rely on oversimplied molecular dynamics that assume atomic motions to be isotropic and homoscedastic. To addres…

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

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

3D Structure Prediction of Atomic Systems with Flow-based Direct Preference Optimization

NeurIPS 2024poster

Predicting high-fidelity 3D structures of atomic systems is a fundamental yet challenging problem in scientific domains. While recent work demonstrates the advantage of generative models in this realm, the exploration of different probability paths are still insufficient, and hallucinations during s…

Cited by 0SourcePDFScholar
2024

Equivariant Diffusion for Crystal Structure Prediction

ICML 2024poster

In addressing the challenge of Crystal Structure Prediction (CSP), symmetry-aware deep learning models, particularly diffusion models, have been extensively studied, which treat CSP as a conditional generation task. However, ensuring permutation, rotation, and periodic translation equivariance durin…

Cited by 14SourcePDFScholar
2024

Learning Superconductivity from Ordered and Disordered Material Structures

NeurIPS 2024poster

Superconductivity is a fascinating phenomenon observed in certain materials under certain conditions. However, some critical aspects of it, such as the relationship between superconductivity and materials' chemical/structural features, still need to be understood. Recent successes of data-driven app…

Cited by 1SourcePDFScholar
2023

Crystal Structure Prediction by Joint Equivariant Diffusion

NeurIPS 2023poster

Crystal Structure Prediction (CSP) is crucial in various scientific disciplines. While CSP can be addressed by employing currently-prevailing generative models (**e.g.** diffusion models), this task encounters unique challenges owing to the symmetric geometry of crystal structures---the invariance o…

2023

Energy-Motivated Equivariant Pretraining for 3D Molecular Graphs

AAAI 2023technical

Pretraining molecular representation models without labels is fundamental to various applications. Conventional methods mainly process 2D molecular graphs and focus solely on 2D tasks, making their pretrained models incapable of characterizing 3D geometry and thus defective for downstream 3D tasks.…