Add retrained PolyEdit checkpoint and audit artifacts
Browse files
README.md
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: pytorch
|
| 3 |
+
license: apache-2.0
|
| 4 |
+
tags:
|
| 5 |
+
- polymers
|
| 6 |
+
- molecular-optimization
|
| 7 |
+
- transformer
|
| 8 |
+
- polyedit
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# PolyEdit Molecular Optimization Transformer (polymer-retrained)
|
| 12 |
+
|
| 13 |
+
This checkpoint adapts MolecularAI's released Molecular Optimization Transformer to
|
| 14 |
+
two-anchor PSMILES and the eight PolyEdit DFT properties. It is not directly comparable
|
| 15 |
+
to the authors' original LogD/Solubility/Clint model without this adaptation.
|
| 16 |
+
|
| 17 |
+
The original transformer body is retained: 6 encoder/decoder layers, hidden size 256,
|
| 18 |
+
8 heads, and feed-forward size 2,048. All 258 shape-compatible upstream tensors were
|
| 19 |
+
loaded. Four vocabulary-dependent embedding/generator tensors were initialized for a
|
| 20 |
+
new vocabulary in which padding and the `*` polymer anchor have distinct IDs. Training
|
| 21 |
+
used 152,036 component-held-out examples; the best validation checkpoint was epoch 7
|
| 22 |
+
of 10.
|
| 23 |
+
|
| 24 |
+
On the balanced 8,176-request test set, the checkpoint obtains 99.217% RDKit+TDC
|
| 25 |
+
validity, 99.168% two-anchor polymer validity, 99.083% changed outputs, 8.892% strict
|
| 26 |
+
MIPS-retrained full-edit hits, and 10.127% strict DFT full-edit hits at 95.034% DFT
|
| 27 |
+
coverage. Only 383 unique raw outputs were generated, so the strong validity comes with
|
| 28 |
+
substantial mode collapse and must not be reported without the diversity result.
|
| 29 |
+
|
| 30 |
+
Upstream code: <https://github.com/MolecularAI/deep-molecular-optimization>
|
| 31 |
+
|
| 32 |
+
Upstream checkpoint: <https://doi.org/10.5281/zenodo.5571965>
|
| 33 |
+
|
| 34 |
+
PolyEdit implementation and record-level evaluation: <https://github.com/promotion-kim/POLYEDIT/tree/tsyou/balanced-polymer-baseline-eval>
|