← Search

Yuxuan Song

23 accepted papers

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

STAC: Plug-and-Play Spatio-Temporal Aware Cache Compression for Streaming 3D Reconstruction

CVPR 2026

Online 3D reconstruction from streaming inputs requires both long-term temporal consistency and efficient memory usage. Although causal variants of VGGT address this challenge through a key-value (KV) cache mechanism, the cache grows linearly with the stream length, creating a major memory bottlenec

Cited by 0SourceScholar
2025

$f$-PO: Generalizing Preference Optimization with $f$-divergence Minimization

AISTATS 2025poster

Preference optimization has made significant progress recently, with numerous methods developed to align language models with human preferences. This paper introduces $f$-divergence Preference Optimization ($f$-PO), a novel framework that generalizes and extends existing approaches. $f$-PO minimizes…

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

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

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

Knowledge-Enhanced Hierarchical Heterogeneous Graph for Personality Identification with Limited Training Data

AAAI 2025technical

Personality identification plays important roles in understanding user behavior and offering foresight ability for downstream applications. The key challenge is how to address the scarcity of labeled personality data. Recently, some studies have adopted data augmentation and prompt learning to perfo…

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

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

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
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

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

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…

2020

Improving Maximum Likelihood Training for Text Generation with Density Ratio Estimation

AISTATS 2020poster

Autoregressive neural sequence generative models trained by Maximum Likelihood Estimation suffer the exposure bias problem in practical finite sample scenarios. The crux is that the number of training samples for Maximum Likelihood Estimation is usually limited and the input data distributions are d…

Cited by 7SourcePDFScholar
2019

Lipschitz Generative Adversarial Nets

ICML 2019oral

In this paper we show that generative adversarial networks (GANs) without restriction on the discriminative function space commonly suffer from the problem that the gradient produced by the discriminator is uninformative to guide the generator. By contrast, Wasserstein GAN (WGAN), where the discrimi…

Cited by 107SourcePDFScholar
2018

Activation Maximization Generative Adversarial Nets

ICLR 2018poster

Class labels have been empirically shown useful in improving the sample quality of generative adversarial nets (GANs). In this paper, we mathematically study the properties of the current variants of GANs that make use of class label information. With class aware gradient and cross-entropy decomposi…