File size: 1,504 Bytes
16b26d4 a461d21 16b26d4 a461d21 16b26d4 69b7e55 16b26d4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | ---
library_name: pytorch
tags:
- polymers
- molecular-editing
- polybert
- supervised-learning
---
# PolyEdit DFT-consistent task augmentation
This repository contains the SFT policies used for the 369, 1,000, and
2,000-request PolyEdit task-augmentation comparison. Each condition has five
seed directories containing the policy, a self-contained policy bundle, the
corresponding fine-tuned polyBERT `Egc` verifier, and the run configuration.
## Training setup
- Property: DFT-labeled chain band gap, `Egc`
- Requests: 369 original, then 1,000 and 2,000 after train-only augmentation
- Decision steps: 1,263, 5,145, and 11,133
- Edit budget: at most three matched-pair transitions
- Policy loss: set-valued negative log-likelihood over valid next edits
- Seeds: 1004, 1005, 1006, 1007, 1008
- Evaluation: fixed held-out sources, 112 single-step and 36 multi-step requests
## Results
Five-seed means and 95% t confidence intervals:
| Requests | Single-target success | Multi-step success |
|---:|---:|---:|
| 369 | 0.889 ± 0.033 | 0.811 ± 0.089 |
| 2,000 | 0.941 ± 0.022 | 0.928 ± 0.019 |
Paired improvements were +0.052 for single-target success (`p=0.040`) and
+0.117 for multi-step success (`p=0.025`).
## Limitations
The policies were trained and evaluated only for `Egc` within the frozen
matched-pair graph. Success therefore measures navigation among polymers with
available DFT labels. It does not establish performance on arbitrary polymer
structures or other properties.
|