cs.LGMar 11, 2026

Continuous Diffusion Transformers for Designing Synthetic Regulatory Elements

Authors: Jonathan Liu, Kia Ghods

Organizations: Department of Computer Science Princeton University

Abstract

We present a parameter-efficient Diffusion Transformer (DiT) for generating 200bp cell-type-specific regulatory DNA sequences. By replacing the U-Net backbone of DNA-Diffusion with a transformer denoiser equipped with a 2D CNN input encoder, our model matches the U-Net's best validation loss in 13 epochs (60×\times fewer) and converges 39% lower, while reducing memorization from 5.3% to 1.7% of generated sequences aligning to training data via BLAT. Ablations show the CNN encoder is essential: without it, validation loss increases 70% regardless of positional embedding choice. We further apply DDPO finetuning using Enformer as a reward model, achieving a 38×\times improvement in predicted regulatory activity. Cross-validation against DRAKES on an independent prediction task confirms that improvements reflect genuine regulatory signal rather than reward model overfitting.

Figures & tables

Appendix figures & tables2 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. D3LM: A Discrete DNA Diffusion Language Model for Bidirectional DNA Understanding and Generation

    Mar 2, 2026Zhao Yang, Hengchang Liu, Chuan Cao +1Language ModelingMolecular Generation

  2. Biological Reasoning-Informed Regression for Interpretable Regulatory DNA Activity Prediction

    Jun 6, 2026Yi Duan, Zhao Yang, Jiwei Zhu +3Interpretable ML