File size: 3,237 Bytes
4b7256c ee372ff 4b7256c ee372ff 4b7256c ee372ff | 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 46 47 48 49 50 51 52 53 54 55 | ---
license: apache-2.0
pipeline_tag: text-generation
tags:
- lfm2
- data-use
- provenance
- sft
- lora
---
# lfm2.5-350M-datause-provenance
LoRA SFT of `LiquidAI/LFM2.5-350M` for data-mention provenance attribute extraction
(producer / year / geography / acronym, verbatim from context).
## Training
- base model: `LiquidAI/LFM2.5-350M`
- dataset: `rafmacalaba/data-use-provenance-sft`
- epochs: 3
- learning rate: 0.0002
- LoRA: r=16 alpha=32 dropout=0.05
- completion-only masking (loss on assistant JSON turn)
## Evaluation (holdout, n=3487)
Exact string match of each emitted attribute against the gold label.
| attribute | tp | fp | fn | precision | recall | f0.5 | f1 |
| --- | --- | --- | --- | --- | --- | --- | --- |
| producer | 1143 | 296 | 306 | 0.7943 | 0.7888 | 0.7932 | 0.7916 |
| year | 1085 | 207 | 265 | 0.8398 | 0.8037 | 0.8323 | 0.8213 |
| geography | 1715 | 371 | 312 | 0.8221 | 0.8461 | 0.8268 | 0.8339 |
| acronym | 1271 | 212 | 145 | 0.8570 | 0.8976 | 0.8649 | 0.8769 |
| **overall** | 5214 | 1086 | 1028 | 0.8276 | 0.8353 | 0.8291 | 0.8314 |
Verbatim rate (emitted values that are substrings of the context): 6290/6300 = 0.9984
## Sample predictions (holdout)
| mention | target | predicted | correct | verbatim |
| --- | --- | --- | --- | --- |
| plant-level data | {"geography": "Indonesian"} | {"geography": "Indonesian"} | geography | geography |
| GIDD database | {"geography": "all regions in the world"} | {"acronym": "GIDD", "geography": "all regions in the world"} | geography | geography,acronym |
| NFHS-3 data | {"acronym": "NFHS-3", "geography": "India"} | {"acronym": "NFHS-3", "geography": "India"} | geography,acronym | geography,acronym |
| data from Hallegatte ( 2012 ) | {"year": "2012"} | {"geography": "LICs", "year": "2012"} | year | year,geography |
| national accounts | {"geography": "Turkey"} | {"geography": "Turkey"} | geography | geography |
| SRF registry of firms | {"acronym": "SRF", "geography": "Brazil", "producer": "Brazilian tax authority", "year": "2012"} | {"acronym": "SRF", "geography": "Brazil", "producer": "Brazilian tax authority", "year": "2012"} | producer,year,geography,acronym | producer,year,geography,acronym |
| Human Capital Index | {"geography": "Kenya", "producer": "World Bank", "year": "2020"} | {"geography": "Kenya", "producer": "World Bank", "year": "2020"} | producer,year,geography | producer,year,geography |
| United Nations population counts by age | {"producer": "United Nations"} | {"producer": "United Nations"} | producer | producer |
| 2002 Census | {"geography": "Rwanda", "year": "2002"} | {"geography": "Rwanda", "year": "2002"} | year,geography | year,geography |
| ENCASEH 1997 | {"acronym": "ENCASEH", "year": "1997"} | {"acronym": "ENCASEH", "year": "1997"} | year,acronym | year,acronym |
| Agricultural Wages in India | {"acronym": "AWI", "geography": "India", "producer": "Ministry of Agriculture"} | {"acronym": "AWI", "geography": "India", "producer": "Ministry of Agriculture"} | producer,geography,acronym | producer,geography,acronym |
| COVID-19 Enterprise Survey | {"geography": "Chad", "year": "2020"} | {"acronym": "ES", "geography": "Chad", "year": "2020"} | year,geography | year,geography,acronym |
|