Continuous Diffusion Transformers for Designing Synthetic Regulatory Elements
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 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 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
| Model | Mode | GM12878 | HepG2 | K562 | hESCT0 (DNAse) |
|---|---|---|---|---|---|
| DNA-Diffusion | In-Situ | 0.53399 | 0.06957 | 0.05587 | – |
| CNN-DiT | In-Situ | 0.19025 | 0.04854 | 0.59428 | 0.20012 |
| CNN-DiT | Ex-Situ | 0.07316 | 0.04656 | 0.03551 | 0.03540 |
| CNN-DiT-DDPO | In-Situ | 4.19501 | 4.11424 | 4.76197 | 1.86090 |
| CNN-DiT-DDPO | Ex-Situ | 1.16401 | 1.17669 | 0.40109 | 0.28727 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Cell Line | ENCODE Accession |
|---|---|
| hESCT0 | ENCLB449ZZZ |
| K562 | ENCLB843GMH |
| HepG2 | ENCLB029COU |
| GM12878 | ENCLB441ZZZ |
| Parameter | U-Net | DiT CNN2D |
|---|---|---|
| Backbone | 2D Conv | Transformer |
| Input encoding | Direct | CNN2D (5x4 kernel) |
| Positional embedding | – | Learned |
| Hidden dim | 200 | 320 |
| Depth / Layers | 3 (mults ) | 6 |
| Attention heads | 4 | 8 (dim_head 48) |