/ Модель

Характеристики

Контекст
512 токенов
Параметры
137 млн
Лицензия
Apache 2.0
Архитектура
electra · dense
Базовая модель
electra-base-discriminator
Опубликована
7 дек 2023

electra-base-discriminator-finetuned-ner-cadec

Разметка токеновТекст → label3 профессии

Дообученная ELECTRA для разметки в медицинских текстах лекарств, побочных реакций, заболеваний, симптомов и находок.

16скачиваний за 30 дней0лайковвеса обновлены13 дек 2023

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Модель csNoHug/electra-base-discriminator-finetuned-ner-cadec — дообученный для распознавания сущностей медицинского назначения энкодер ELECTRA, созданный на основе google/electra-base-discriminator. Она размечает токены в тексте, выделяя побочные реакции, заболевания, лекарства, находки и симптомы. Модель насчитывает 136,7 млн параметров, принимает текст с контекстом до 512 токенов и распространяется в формате safetensors.

В карточке указаны метрики оценки: общий F1 — 0,7097, а F1 для лекарств — 0,9538; для заболеваний — 0,1633. Это указывает на заметно неоднородное качество распознавания разных классов, особенно слабые результаты для заболеваний. Модель выпущена под лицензией Apache-2.0; язык и состав обучающего набора в карточке не раскрыты.

Сценарии применения

  • выделение названий лекарств в медицинских текстах
  • поиск упоминаний побочных реакций на препараты
  • извлечение симптомов из медицинских записей
  • разметка упоминаний заболеваний и медицинских находок

Ограничения

Качество существенно различается по классам: F1 для лекарств составляет 0,9538, а для заболеваний — 0,1633. Язык и обучающий датасет в карточке модели не уточнены.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

electra-base-discriminator-finetuned-ner-cadec

This model is a fine-tuned version of google/electra-base-discriminator on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4887
  • Precision: 0.6872
  • Recall: 0.7339
  • F1: 0.7097
  • Accuracy: 0.9230
  • Adr Precision: 0.6712
  • Adr Recall: 0.7483
  • Adr F1: 0.7076
  • Disease Precision: 0.1667
  • Disease Recall: 0.16
  • Disease F1: 0.1633
  • Drug Precision: 0.9509
  • Drug Recall: 0.9568
  • Drug F1: 0.9538
  • Finding Precision: 0.4074
  • Finding Recall: 0.3188
  • Finding F1: 0.3577
  • Symptom Precision: 0.5455
  • Symptom Recall: 0.6667
  • Symptom F1: 0.6
  • B-adr Precision: 0.7798
  • B-adr Recall: 0.8336
  • B-adr F1: 0.8058
  • B-disease Precision: 0.2174
  • B-disease Recall: 0.2
  • B-disease F1: 0.2083
  • B-drug Precision: 0.9753
  • B-drug Recall: 0.9753
  • B-drug F1: 0.9753
  • B-finding Precision: 0.5238
  • B-finding Recall: 0.3333
  • B-finding F1: 0.4074
  • B-symptom Precision: 0.6071
  • B-symptom Recall: 0.6296
  • B-symptom F1: 0.6182
  • I-adr Precision: 0.6587
  • I-adr Recall: 0.7267
  • I-adr F1: 0.6910
  • I-disease Precision: 0.1765
  • I-disease Recall: 0.15
  • I-disease F1: 0.1622
  • I-drug Precision: 0.9509
  • I-drug Recall: 0.9568
  • I-drug F1: 0.9538
  • I-finding Precision: 0.4565
  • I-finding Recall: 0.4038
  • I-finding F1: 0.4286
  • I-symptom Precision: 0.35
  • I-symptom Recall: 0.5385
  • I-symptom F1: 0.4242
  • Macro Avg F1: 0.5675
  • Weighted Avg F1: 0.7497

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 40

Training results

Training LossEpochStepValidation LossPrecisionRecallF1AccuracyAdr PrecisionAdr RecallAdr F1Disease PrecisionDisease RecallDisease F1Drug PrecisionDrug RecallDrug F1Finding PrecisionFinding RecallFinding F1Symptom PrecisionSymptom RecallSymptom F1B-adr PrecisionB-adr RecallB-adr F1B-disease PrecisionB-disease RecallB-disease F1B-drug PrecisionB-drug RecallB-drug F1B-finding PrecisionB-finding RecallB-finding F1B-symptom PrecisionB-symptom RecallB-symptom F1I-adr PrecisionI-adr RecallI-adr F1I-disease PrecisionI-disease RecallI-disease F1I-drug PrecisionI-drug RecallI-drug F1I-finding PrecisionI-finding RecallI-finding F1I-symptom PrecisionI-symptom RecallI-symptom F1Macro Avg F1Weighted Avg F1
No log1.01250.30850.54560.64330.59040.90180.48470.68830.56890.00.00.00.82450.95680.88570.00.00.00.00.00.00.66180.80710.72730.00.00.00.87910.98770.93020.00.00.00.00.00.00.46660.64980.54310.00.00.00.85640.95680.90380.00.00.00.00.00.00.31040.6156
No log2.02500.26240.57600.67840.62300.91320.55570.73170.63170.02940.040.03390.88510.95060.91670.11900.07250.09010.00.00.00.71200.85310.77620.14290.040.06250.92940.97530.95180.550.16670.25580.00.00.00.56700.71050.63070.13330.20.160.90060.95060.92490.250.13460.17500.00.00.00.39370.6840
No log3.03750.26590.60900.69310.64830.91400.62310.73830.67580.01890.040.02560.90700.96300.93410.17650.17390.17520.00.00.00.76480.82300.79280.05880.040.04760.93570.98770.96100.43400.34850.38660.00.00.00.60210.71050.65180.15220.350.21210.91230.96300.93690.41380.23080.29630.00.00.00.42850.7086
0.29714.05000.26590.63130.69990.66380.92000.62590.73330.67540.02860.040.03330.92770.95060.93900.26870.26090.26470.6250.18520.28570.76250.83540.79730.07690.040.05260.96910.96910.96910.47060.36360.41030.8750.25930.40000.61730.69230.65270.16130.250.19610.93330.95060.94190.37210.30770.33680.00.00.00.47570.7208
0.29715.06250.28270.61240.71570.66010.91560.59150.7650.66720.06900.080.07410.92810.95680.94220.19570.13040.15650.50.25930.34150.72610.87260.79260.23080.120.15790.95780.98150.96950.51430.27270.35640.83330.37040.51280.59100.74290.65830.160.20.17780.93370.95680.94510.30770.15380.20510.00.00.00.47760.7180
0.29716.07500.29000.63470.69650.66410.91960.61160.72170.66210.11900.20.14930.93940.95680.94800.32350.15940.21360.550.40740.46810.75810.83190.79320.23330.280.25450.96950.98150.97550.47830.16670.24720.80.44440.57140.60070.70650.64930.1250.20.15380.94510.95680.95090.40740.21150.27850.40.30770.34780.52220.7195
0.29717.08750.29940.63000.70780.66670.91930.61480.73170.66820.13890.20.16390.93980.96300.95120.27450.20290.23330.440.40740.42310.74650.86020.79930.20.240.21820.96390.98770.97560.42110.24240.30770.70.51850.59570.62370.70450.66160.13790.20.16330.93980.96300.95120.32560.26920.29470.30770.30770.30770.52750.7283
0.09638.010000.31970.64260.70670.67310.91900.62370.72670.67130.09520.080.08700.93900.95060.94480.31750.28990.30300.50.44440.47060.75680.83190.79260.10530.080.09090.97550.98150.97850.41670.30300.35090.56520.48150.520.61510.70850.65850.07140.050.05880.94480.95060.94770.39290.42310.40740.38460.38460.38460.51900.7263
0.09639.011250.32810.65120.69760.67360.92160.63540.71170.67140.16670.240.19670.92770.95060.93900.30190.23190.26230.68420.48150.56520.77800.82480.80070.22580.280.250.96360.98150.97250.48570.25760.33660.81250.48150.60470.63860.69030.66340.120.150.13330.93330.95060.94190.35420.32690.340.55560.38460.45450.54980.7322
0.096310.012500.34050.65630.71800.68580.92220.62410.74170.67780.09520.080.08700.95630.94440.95030.40430.27540.32760.60.55560.57690.75040.85130.79770.22220.160.18600.98110.96300.97200.52630.30300.38460.60870.51850.56000.63940.72870.68120.06670.050.05710.95630.94440.95030.450.34620.39130.46150.46150.46150.54420.7385
0.096311.013750.36410.67310.71120.69160.91840.64390.72330.68130.20830.20.20410.95090.95680.95380.45240.27540.34230.50.55560.52630.77320.82650.79900.26090.240.25000.97530.97530.97530.57140.30300.39600.520.48150.50.63600.71460.67300.16670.150.15790.95680.95680.95680.48570.32690.39080.33330.46150.38710.54860.7385
0.045512.015000.36040.65670.69310.67440.91630.63870.69830.66720.22220.240.23080.91180.95680.93370.32730.26090.29030.58330.51850.54900.77700.81420.79520.240.240.240.94670.98770.96680.47620.30300.37040.71430.55560.62500.63380.67610.65430.150.150.150.92260.95680.93940.33330.30770.320.38460.38460.38460.54460.7271
0.045513.016250.39130.66850.70330.68540.92260.63580.72170.67600.14290.080.10260.94510.95680.95090.39130.26090.31300.54170.48150.50980.78200.83190.80620.14290.080.10260.97550.98150.97850.47370.27270.34620.61900.48150.54170.62820.69430.65960.10.050.06670.95090.95680.95380.41460.32690.36560.3750.46150.41380.52350.7315
0.045514.017500.36910.67390.70220.68770.92160.65090.71170.67990.21430.240.22640.95090.95680.95380.36360.28990.32260.66670.44440.53330.77520.81770.79590.280.280.280.98770.98770.98770.48890.33330.39640.82350.51850.63640.64670.70040.67250.14290.150.14630.95090.95680.95380.35290.34620.34950.6250.38460.47620.56950.7403
0.045515.018750.39130.65970.71800.68760.91970.64570.7350.68750.16670.160.16330.92260.95680.93940.32730.26090.29030.51610.59260.55170.77200.83890.80410.22730.20.21280.96390.98770.97560.45650.31820.3750.60.55560.57690.63950.71460.67500.11110.10.10530.92810.95680.94220.34780.30770.32650.28570.46150.35290.53460.7363
0.022716.020000.39380.66490.71230.68780.91950.64540.7250.68290.23330.280.25450.95630.94440.95030.32760.27540.29920.6250.55560.58820.77630.82300.79900.28570.320.30190.98740.96910.97820.43750.31820.36840.66670.51850.58330.63040.70450.66540.19050.20.19510.95630.94440.95030.34620.34620.34620.38460.38460.38460.55720.7358
0.022717.021250.40210.67270.71690.69410.92160.66520.7350.69830.20830.20.20410.93870.94440.94150.33330.26090.29270.43240.59260.50.77810.83190.80410.26090.240.25000.97520.96910.97210.47730.31820.38180.55170.59260.57140.65740.71860.68670.18750.150.16670.95030.94440.94740.3750.34620.36000.29170.53850.37840.55190.7430
0.022718.022500.41070.69170.70890.70020.92270.66920.72170.69450.19050.160.17390.94510.95680.95090.42860.30430.35590.54170.48150.50980.78380.82120.80210.250.20.22220.96950.98150.97550.50.33330.40.59090.48150.53060.65670.70850.68160.13330.10.11430.94510.95680.95090.44190.36540.40.33330.38460.35710.54340.7415
0.022719.023750.42060.65510.70330.67830.91990.62050.71670.66510.14290.080.10260.93410.96300.94830.41510.31880.36070.52380.40740.45830.75850.83360.79430.15380.080.10530.95810.98770.97260.45830.33330.38600.61900.48150.54170.61910.69430.65460.10.050.06670.93980.96300.95120.46510.38460.42110.38460.38460.38460.52780.7281
0.012820.025000.44550.64580.72480.68300.91710.60800.73670.66620.11760.080.09520.95150.96910.96020.42110.34780.38100.60.55560.57690.73930.85310.79210.20.120.150.96970.98770.97860.52080.37880.43860.63640.51850.57140.60790.71860.65860.09090.050.06450.95150.96910.96020.46810.42310.44440.40.46150.42860.54870.7346
0.012821.026250.42080.67980.72140.70.92100.65870.73330.69400.22730.20.21280.93980.96300.95120.39290.31880.35200.560.51850.53850.77250.83540.80270.27270.240.25530.96360.98150.97250.46670.31820.37840.60870.51850.56000.65010.71460.68080.15380.10.12120.93980.96300.95120.41670.38460.40.38460.38460.38460.55070.7416
0.012822.027500.44100.65310.72480.68710.91820.62980.740.68050.19230.20.19610.93980.96300.95120.36360.28990.32260.53570.55560.54550.75360.84420.79630.250.240.24490.96950.98150.97550.50.31820.38890.60.55560.57690.62770.71660.66920.15790.150.15380.93980.96300.95120.39130.34620.36730.33330.38460.35710.54810.7357
0.012823.028750.45600.67900.72820.70270.92130.65440.74170.69530.21050.160.18180.93980.96300.95120.40.31880.35480.59260.59260.59260.76220.82830.79390.33330.240.27910.96360.98150.97250.50.34850.41070.66670.59260.62750.64610.72060.68130.10.050.06670.93980.96300.95120.43480.38460.40820.35710.38460.37040.55610.7410
0.008224.030000.46420.66460.72030.69130.92020.64210.72670.68180.24140.280.25930.93410.96300.94830.420.30430.35290.50.59260.54240.76510.83010.79630.25930.280.26920.96360.98150.97250.52380.33330.40740.62960.62960.62960.63090.70240.66480.18180.20.19050.93410.96300.94830.43900.34620.38710.27780.38460.32260.55880.7366
0.008225.031250.46060.67120.72140.69540.91870.64850.7350.68910.18520.20.19230.95680.95680.95680.37040.28990.32520.61540.59260.60380.76870.81770.79250.240.240.240.97530.97530.97530.51160.33330.40370.6250.55560.58820.62630.71260.66670.15790.150.15380.95680.95680.95680.38640.32690.35420.38460.38460.38460.55160.7346
0.008226.032500.44700.67670.73730.70570.92060.65560.7550.70180.18750.120.14630.93410.96300.94830.39340.34780.36920.55560.55560.55560.75650.83010.79160.20.120.150.96390.98770.97560.48080.37880.42370.60.55560.57690.64300.73280.68500.10.050.06670.93410.96300.94830.41180.40380.40780.38460.38460.38460.54100.7390
0.008227.033750.46740.67270.71230.69200.92150.64730.7250.68400.22220.240.23080.93940.95680.94800.39130.26090.31300.60.55560.57690.76550.83190.79730.280.280.280.96950.98150.97550.51430.27270.35640.65220.55560.60.64960.69430.67120.20.20.20000.94510.95680.95090.40.30770.34780.50.46150.48000.56590.7372
0.006228.035000.46930.67360.73610.70350.92030.64550.76170.69880.18180.160.17020.95090.95680.95380.41670.28990.34190.58330.51850.54900.75640.84070.79630.19050.160.17390.97550.98150.97850.46340.28790.35510.6250.55560.58820.63540.74090.68410.1250.10.11110.95090.95680.95380.45240.36540.40430.38460.38460.38460.54300.7395
0.006229.036250.46810.69050.73270.71100.92250.66820.74830.70600.17390.160.16670.94550.96300.95410.46300.36230.40650.56520.48150.520.76480.82300.79280.19050.160.17390.96950.98150.97550.54550.36360.43640.60870.51850.56000.65380.71860.68470.11760.10.10810.94550.96300.95410.46810.42310.44440.42860.46150.44440.55740.7428
0.006230.037500.47710.67650.72480.69980.92010.65620.7350.69340.22730.20.21280.94510.95680.95090.37290.31880.34380.58620.62960.60710.76860.82300.79490.250.20.22220.96950.98150.97550.45830.33330.38600.66670.66670.66670.63830.71460.67430.13330.10.11430.94510.95680.95090.39220.38460.38830.3750.46150.41380.55870.7384
0.006231.038750.47890.68740.72710.70670.92130.67020.7450.70560.160.160.160.93410.96300.94830.46940.33330.38980.46150.44440.45280.76730.82300.79420.22730.20.21280.96390.98770.97560.54760.34850.42590.58330.51850.54900.65440.72060.68590.16670.150.15790.93410.96300.94830.46510.38460.42110.40.46150.42860.55990.7428
0.004332.040000.48450.69690.73160.71380.92250.67320.74170.70580.19230.20.19610.95710.96300.96000.48980.34780.40680.57140.59260.58180.78190.82480.80280.250.240.24490.97550.98150.97850.53490.34850.42200.61540.59260.60380.65680.71660.68540.16670.150.15790.95710.96300.96000.51160.42310.46320.40.46150.42860.57470.7498
0.004333.041250.46830.68090.72480.70210.92350.65770.73670.69500.150.120.13330.93940.95680.94800.42110.34780.38100.61540.59260.60380.78290.84250.81160.15790.120.13640.96950.98150.97550.52170.36360.42860.640.59260.61540.64870.70650.67640.13330.10.11430.94510.95680.95090.44900.42310.43560.50.53850.51850.56630.7470
0.004334.042500.48600.68960.73730.71260.92130.66570.75330.70680.150.120.13330.93370.95680.94510.48150.37680.42280.60.55560.57690.76340.83360.79700.15790.120.13640.96340.97530.96930.52080.37880.43860.640.59260.61540.64680.72670.68450.18750.150.16670.93370.95680.94510.51110.44230.47420.41670.38460.40.56270.7445
0.004335.043750.48570.69700.73730.71660.92200.67360.75330.71130.20.160.17780.95090.95680.95380.48080.36230.41320.53570.55560.54550.77610.82830.80140.20.160.17780.97530.97530.97530.52170.36360.42860.59260.59260.59260.65280.73080.68960.18750.150.16670.95090.95680.95380.48890.42310.45360.40.46150.42860.56680.7485
0.003736.045000.48160.69120.73500.71240.92280.66570.750.70530.19050.160.17390.95090.95680.95380.47060.34780.40000.57140.59260.58180.77450.83890.80540.19050.160.17390.96930.97530.97230.51110.34850.41440.61540.59260.60380.65570.72870.69030.18750.150.16670.95090.95680.95380.50.42310.45830.46670.53850.50.57390.7502
0.003737.046250.48780.69190.73500.71280.92290.67310.75170.71020.16670.160.16330.95680.95680.95680.45280.34780.39340.51720.55560.53570.77760.83540.80550.21740.20.20830.97530.97530.97530.53490.34850.42200.59260.59260.59260.66120.73080.69420.17650.150.16220.95680.95680.95680.47830.42310.44900.38890.53850.45160.57170.7519
0.003738.047500.48900.68760.73050.70840.92350.66520.7450.70280.14290.120.13040.95680.95680.95680.44440.34780.39020.55170.59260.57140.77590.83360.80380.15790.120.13640.97530.97530.97530.52270.34850.41820.59260.59260.59260.65450.72470.68780.17650.150.16220.95680.95680.95680.44680.40380.42420.41180.53850.46670.56240.7473
0.003739.048750.48710.68690.73050.70800.92260.67220.7450.70670.16670.160.16330.95090.95680.95380.40740.31880.35770.51520.62960.56670.77890.83540.80610.21740.20.20830.96930.97530.97230.52380.33330.40740.60710.62960.61820.65930.72060.68860.17650.150.16220.95090.95680.95380.43480.38460.40820.350.53850.42420.56490.7481
0.00340.050000.48870.68720.73390.70970.92300.67120.74830.70760.16670.160.16330.95090.95680.95380.40740.31880.35770.54550.66670.60.77980.83360.80580.21740.20.20830.97530.97530.97530.52380.33330.40740.60710.62960.61820.65870.72670.69100.17650.150.16220.95090.95680.95380.45650.40380.42860.350.53850.42420.56750.7497

Framework versions

  • Transformers 4.35.2
  • Pytorch 2.1.0+cu118
  • Datasets 2.15.0
  • Tokenizers 0.15.0

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · token-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
ElectraForTokenClassification
Ревизия весов
44cf894

0 моделей