Decoupling Structure and Property in Discrete Protein Diffusion: Alignment Dynamics and Collapse Mechanisms
Junqi WuiD, Liyan DongiD, Nuo JiaiD, Longyu LiiD, Hao ZhangiD
Abstract
Direct Preference Optimization (DPO) has emerged as a powerful paradigm for aligning generative models, yet its temporal optimization dynamics in the discrete diffusion space of proteins remain poorly understood. Existing approaches often assume that maintaining structural integrity while optimizing physicochemical properties requires simultaneous, tightly coupled reinforcement learning constraints. In this work, we challenge this assumption by uncovering a fundamental temporal decoupling between structural and functional alignment. Using antibody design as a testbed, extensive trajectory analysis reveals two distinct regimes: (1) Instant Structural Alignment, where the strong generative prior of discrete diffusion rapidly eliminates structural hallucinations via denoising within the first few epochs; and (2) Slow Property Adaptation, where physicochemical attributes improve gradually over a prolonged optimization window. We further identify a critical transition point around Epoch 50, which empirically defines a Pareto-optimal boundary between property improvement and structural stability. Beyond this point, continued optimization induces a sharp phase transition into a Structural Collapse regime. To isolate the physical driver underlying this collapse, we introduce a counterfactual preference experiment targeting negative charge. We observe a striking symmetrical collapse: while hydrophilicity optimization induces a Poly-Arginine (+) degeneration, negative charge optimization drives a Poly-Aspartate (-) degeneration. Despite opposite physicochemical trajectories, including extreme shifts in isoelectric point (> 11 vs. < 4.5), both regimes converge to the same structural failure. This symmetry demonstrates that generic Coulombic repulsion, rather than residue-specific bias, constitutes the fundamental physical constraint being violated. Our findings reveal that discrete diffusion models possess strong intrinsic structural robustness, enabling minimalist alignment strategies provided optimization halts before this physical boundary. More broadly, this work offers a mechanistic warning against unchecked reward optimization in biological generation, illustrating a concrete manifestation of Goodharts Law in protein design. Code and data are available at https://github.com/Wu-Junqi/DPO-Protein-Diffusion.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 0
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Uncertainty-Aware Discrete Diffusion Improves Protein Design
Sazan Mahbub, Christoph Feinauer, … Eric P. Xing·bioRxiv·Jun 30, 2025
- Aligning protein generative models with experimental fitness via Direct Preference Optimization
Talal Widatalla, Ashir A. Borah, … Brian Hie·Nature MethodsPeer-reviewed·Aug 14, 2026
- Constrained Diffusion for Protein Design with Hard Structural Constraints
Jacob K. Christopher, Austin Seamann, … Ferdinando Fioretto·bioRxiv·Oct 15, 2025
- Joint Generation of Protein Sequence and Structure with RoseTTAFold Sequence Space Diffusion
S. L. Lisanza, J. M. Gershon, … D. Baker·bioRxiv·May 8, 2023
- Unified Sampling and Ranking for Protein Docking with DFMDock
Lee‐Shin Chu, Sudeep Sarma, … Jeffrey J. Gray·Proteins Structure Function and BioinformaticsPeer-reviewed·Aug 6, 2026
- Improving diffusion-based protein backbone generation with global-geometry-aware latent encoding
Yuyang Zhang, Yuhang Liu, … Haipeng Gong·Nature Machine IntelligencePeer-reviewed·Jun 18, 2025