Token Classification
Transformers
English
ner
biomedical
transfer-learning
curriculum-learning
autoresearch
modernbert
Instructions to use osatinsky/openmed-autoresearch with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use osatinsky/openmed-autoresearch with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="osatinsky/openmed-autoresearch")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("osatinsky/openmed-autoresearch", device_map="auto") - Notebooks
- Google Colab
- Kaggle
AutoResearch Agent commited on
Commit ·
a38491d
1
Parent(s): e1949cb
add chemical NER target findings from apr8-chem-target branch
Browse files- FINDINGS_CHEM.md +60 -0
FINDINGS_CHEM.md
ADDED
|
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Cross-Dataset Transfer for Chemical NER (BC5CDR-Chem Target)
|
| 2 |
+
|
| 3 |
+
## Key Finding: Asymmetric Transfer
|
| 4 |
+
|
| 5 |
+
**Chemical NER strongly helps disease NER (+5.0%), but disease NER barely helps chemical NER (+1.3%).** This is the central discovery of this experiment set.
|
| 6 |
+
|
| 7 |
+
### Transfer Affinity: Source → BC5CDR Chemical NER
|
| 8 |
+
|
| 9 |
+
| Source Dataset | Entity Types | val_f1 (50/50 split) | Δ vs Baseline |
|
| 10 |
+
|---|---|---|---|
|
| 11 |
+
| *baseline (no transfer)* | *—* | *0.8090 ± 0.005* | *—* |
|
| 12 |
+
| ncbi_disease | Diseases | 0.8148 | +0.006 |
|
| 13 |
+
| jnlpba | DNA/RNA/Proteins/Cells | 0.8047 | -0.004 |
|
| 14 |
+
| bc2gm | Genes/Proteins | 0.8079 | -0.001 |
|
| 15 |
+
| linnaeus | Species | 0.8104 | +0.001 |
|
| 16 |
+
|
| 17 |
+
At 50/50 split, **no source dataset significantly helps chemical NER**. This is in stark contrast to disease NER, where bc5cdr_chem at 50/50 gave +4.4%.
|
| 18 |
+
|
| 19 |
+
### Optimal Curriculum (small improvement)
|
| 20 |
+
|
| 21 |
+
The best curriculum uses **minimal pretraining** — much less than the disease target needed:
|
| 22 |
+
|
| 23 |
+
| Config | Mean F1 (4 runs) | Std | Δ vs Baseline |
|
| 24 |
+
|---|---|---|---|
|
| 25 |
+
| Baseline (chem only) | 0.8090 | ±0.0053 | — |
|
| 26 |
+
| **jnlpba 10% → disease 5% → chem 85%** | **0.8195** | **±0.0024** | **+0.0105** |
|
| 27 |
+
|
| 28 |
+
The improvement is ~1.3% absolute — modest but real (lower variance too).
|
| 29 |
+
|
| 30 |
+
### Asymmetry Analysis
|
| 31 |
+
|
| 32 |
+
| Transfer Direction | Improvement | Optimal Pretrain Time |
|
| 33 |
+
|---|---|---|
|
| 34 |
+
| Chemicals → Disease NER | **+5.0%** (0.8033 → 0.8535) | 40% pretrain, 60% target |
|
| 35 |
+
| Disease → Chemical NER | **+1.3%** (0.8090 → 0.8195) | 15% pretrain, 85% target |
|
| 36 |
+
| **Ratio** | **3.8x stronger chem→disease** | **2.7x more target time needed** |
|
| 37 |
+
|
| 38 |
+
### Why the Asymmetry?
|
| 39 |
+
|
| 40 |
+
1. **BC5CDR contains BOTH chemical and disease entities.** When used as a source for disease NER, the model learns disease-relevant biomedical context alongside chemical NER. This dual-entity annotation makes it uniquely helpful.
|
| 41 |
+
|
| 42 |
+
2. **Chemical entity recognition is more self-contained.** Chemical names (e.g., "aspirin", "paracetamol") are more lexically distinctive than disease names (which overlap with symptoms, anatomy, etc.). Chemical NER relies less on contextual cues that transfer learning provides.
|
| 43 |
+
|
| 44 |
+
3. **Dataset size matters.** BC5CDR is a larger dataset (~5K train examples per entity type). NCBI Disease is smaller (~600 train examples). The smaller target benefits more from transfer because it has less training data.
|
| 45 |
+
|
| 46 |
+
4. **JNLPBA helps chemicals more than diseases.** JNLPBA (proteins, DNA, RNA) provides the best source signal for chemical NER — likely because proteins are drug targets, creating shared vocabulary in biomedical text.
|
| 47 |
+
|
| 48 |
+
### Experimental Details
|
| 49 |
+
|
| 50 |
+
- 26 experiments total (+ variance runs)
|
| 51 |
+
- Same infrastructure as disease target: ModernBERT-base, RTX 4090, 5-min time budget, batch=64
|
| 52 |
+
- Cosine scheduler confirmed better than constant_with_warmup (both targets)
|
| 53 |
+
- Hyperparameter defaults (LR=5e-5, WD=0.01) confirmed optimal (both targets)
|
| 54 |
+
|
| 55 |
+
### Conclusions for Paper
|
| 56 |
+
|
| 57 |
+
1. **Cross-dataset transfer in biomedical NER is strongly asymmetric** — this is a novel finding with practical implications for curriculum design.
|
| 58 |
+
2. **The direction of transfer matters more than the semantic similarity** — chemicals and diseases co-occur in text, but only one direction of transfer helps.
|
| 59 |
+
3. **BC5CDR's dual-entity annotations are uniquely valuable** — datasets with multiple related entity types provide richer learning signals.
|
| 60 |
+
4. **Curriculum time allocation is target-dependent** — "easy" targets (chemical NER, more distinctive entities) need minimal pretraining; "hard" targets (disease NER, contextual entities) need substantial pretraining.
|