← Search

Yuyan Ni

11 accepted papers

2026

Towards a Universally Transferable Acceleration Method for Density Functional Theory

ICLR 2026poster

Recently, sophisticated deep learning-based approaches have been developed for generating efficient initial guesses to accelerate the convergence of density functional theory (DFT) calculations. While the actual initial guesses are often density matrices (DM), quantities that can convert into densit…

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

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

Beyond Efficiency: Molecular Data Pruning for Enhanced Generalization

NeurIPS 2024poster

With the emergence of various molecular tasks and massive datasets, how to perform efficient training has become an urgent yet under-explored issue in the area. Data pruning (DP), as an oft-stated approach to saving training burdens, filters out less influential samples to form a coreset for trainin…

Cited by 5SourcePDFScholar
2024

Multimodal Molecular Pretraining via Modality Blending

ICLR 2024poster

Self-supervised learning has recently gained growing interest in molecular modeling for scientific tasks such as AI-assisted drug discovery. Current studies consider leveraging both 2D and 3D molecular structures for representation learning. However, relying on straightforward alignment strategies t…

Cited by 18SourcePDFScholar
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
2023

Fractional Denoising for 3D Molecular Pre-training

ICML 2023poster

Coordinate denoising is a promising 3D molecular pre-training method, which has achieved remarkable performance in various downstream drug discovery tasks. Theoretically, the objective is equivalent to learning the force field, which is revealed helpful for downstream tasks. Nevertheless, there are…