Downloads · 30 days
6
2% of all-time downloads
LoFlexMDM-Medium-BracketSAFE
dhruveshpatel/LoFlexMDM-Medium-BracketSAFE
LoFlexMDM-Medium-BracketSAFE is a machine learning model from dhruveshpatel. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Variable-length discrete diffusion model for bracket SAFE molecule generation.
All-time downloads
363
Public
Parameters
96.9M
6.2 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors387 MB · 100%
At a glance
- Access
- Public
- Created
- May 29, 2026
- Updated
- May 29, 2026
- SHA
- e6ffcff2
Model card
From the Hugging Face model README
LoFlexMDM-Medium-BracketSAFE
Variable-length discrete diffusion model for bracket SAFE molecule generation.
Paper: https://arxiv.org/pdf/2602.18695
Training
| Hyperparameter | Value |
|---|---|
| Learning rate | 0.0001 |
| Global batch size | 2048 |
| Block size | 256 |
| Training steps | 50000 |
| Weight decay | 0.01 |
| Dataset | Bracket SAFE (datamol-io/safe-gpt, ~1.17B molecules) |
| Checkpoint | EMA weights at step 50000 |
W&B run: learned-noise-icml/safe_lflexmdm_no_share_tiny_fix_b_um_a1
Unconditional generation (de novo)
1024 sampling steps, 1000 molecules per run, mean ± std over 5 seeds (from paper Table 1 / appendix).
| conf. | p | Validity (%) | Diversity | Uniqueness (%) | Quality (%) |
|---|---|---|---|---|---|
| yes | 1.0 | 99.0 ± 0.0 | 0.900 ± 0.000 | 99.300 ± 0.200 | 55.400 ± 0.300 |
| yes | 0.5 | 99.900 ± 0.000 | 0.830 ± 0.000 | 93.200 ± 0.600 | 69.300 ± 0.700 |
| no | 1.0 | 99.100 ± 0.100 | 0.900 ± 0.000 | 99.600 ± 0.100 | 55.100 ± 0.600 |
| no | 0.5 | 99.700 ± 0.100 | 0.850 ± 0.000 | 99.500 ± 0.100 | 79.500 ± 0.700 |
Conditional generation (fragment-constrained)
Means over 5 runs (from paper Table 2). Tasks: LD (linker design), ME (motif extension), SD (scaffold decoration), SG (superstructure generation).
| Task | Validity (%) | Diversity | Uniqueness (%) | Quality (%) |
|---|---|---|---|---|
| Linker design | 99.6 | 0.576 | 64.4 | 51.7 |
| Motif extension | 99.9 | 0.608 | 79.2 | 53.6 |
| Scaffold decoration | 99.8 | 0.601 | 82.6 | 40.5 |
| Superstructure generation | 100.0 | 0.593 | 72.6 | 37.0 |
Usage
See the https://github.com/dhruvdcoder/LoFlexMDM release repository for training and evaluation instructions.
- Created
- May 29, 2026
- Updated
- May 29, 2026