luca0621 commited on
Commit
3e206bf
·
verified ·
1 Parent(s): 3a5b7cc

Add retrained PolyEdit checkpoint and audit artifacts

Browse files
Files changed (1) hide show
  1. README.md +34 -0
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>