🚀 Akashpb13/Central_kurdish_xlsr
このモデルは、MOZILLA-FOUNDATION/COMMON_VOICE_7_0 - huデータセット上でfacebook/wav2vec2-xls-r-300mをファインチューニングしたバージョンです。評価セット(トレーニングデータセットの10%と、無効化データ、報告データ、その他のデータ、開発データセットをマージしたもの)では、以下の結果を達成しています。
- 損失: 0.348580
- 単語誤り率 (Wer): 0.401147
✨ 主な機能
このモデルは自動音声認識タスクに特化しており、中央クルド語の音声を高精度に認識することができます。
📚 ドキュメント
モデルの説明
"facebook/wav2vec2-xls-r-300m"をファインチューニングしました。
想定される用途と制限
詳細な情報が必要です。
トレーニングと評価データ
トレーニングデータ -
Common voice Central Kurdishのtrain.tsv、dev.tsv、invalidated.tsv、reported.tsv、およびother.tsv
Common voice 7.0に含まれるすべてのデータセットを連結した後、アップボートがダウンボートより多いポイントのみを考慮し、重複を削除しました。
トレーニング手順
トレーニングデータセットを作成するために、すべての可能なデータセットを追加し、90-10の分割を行いました。
トレーニングハイパーパラメータ
トレーニング中に以下のハイパーパラメータが使用されました。
- 学習率 (learning_rate): 0.000095637994662983496
- トレーニングバッチサイズ (train_batch_size): 16
- 評価バッチサイズ (eval_batch_size): 16
- シード (seed): 13
- 勾配累積ステップ (gradient_accumulation_steps): 2
- 学習率スケジューラのタイプ (lr_scheduler_type): cosine_with_restarts
- 学習率スケジューラのウォームアップステップ (lr_scheduler_warmup_steps): 200
- エポック数 (num_epochs): 100
- 混合精度トレーニング (mixed_precision_training): Native AMP
トレーニング結果
ステップ |
トレーニング損失 |
検証損失 |
単語誤り率 (Wer) |
500 |
5.097800 |
2.190326 |
1.001207 |
1000 |
0.797500 |
0.331392 |
0.576819 |
1500 |
0.405100 |
0.262009 |
0.549049 |
2000 |
0.322100 |
0.248178 |
0.479626 |
2500 |
0.264600 |
0.258866 |
0.488983 |
3000 |
0.228300 |
0.261523 |
0.469665 |
3500 |
0.201000 |
0.270135 |
0.451856 |
4000 |
0.180900 |
0.279302 |
0.448536 |
4500 |
0.163800 |
0.280921 |
0.459704 |
5000 |
0.147300 |
0.319249 |
0.471778 |
5500 |
0.137600 |
0.289546 |
0.449140 |
6000 |
0.132000 |
0.311350 |
0.458195 |
6500 |
0.117100 |
0.316726 |
0.432840 |
7000 |
0.109200 |
0.302210 |
0.439481 |
7500 |
0.104900 |
0.325913 |
0.439481 |
8000 |
0.097500 |
0.329446 |
0.431935 |
8500 |
0.088600 |
0.345259 |
0.425898 |
9000 |
0.084900 |
0.342891 |
0.428313 |
9500 |
0.080900 |
0.353081 |
0.424389 |
10000 |
0.075600 |
0.347063 |
0.424992 |
10500 |
0.072800 |
0.330086 |
0.424691 |
11000 |
0.068100 |
0.350658 |
0.421974 |
11500 |
0.064700 |
0.342949 |
0.413522 |
12000 |
0.061500 |
0.341704 |
0.415334 |
12500 |
0.059500 |
0.346279 |
0.411410 |
13000 |
0.057400 |
0.349901 |
0.407184 |
13500 |
0.056400 |
0.347733 |
0.402656 |
14000 |
0.053300 |
0.344899 |
0.405976 |
14500 |
0.052900 |
0.346708 |
0.402656 |
15000 |
0.050600 |
0.344118 |
0.400845 |
15500 |
0.050200 |
0.348396 |
0.402958 |
16000 |
0.049800 |
0.348312 |
0.401751 |
16500 |
0.051900 |
0.348372 |
0.401147 |
17000 |
0.049800 |
0.348580 |
0.401147 |
フレームワークバージョン
- Transformers 4.16.0.dev0
- Pytorch 1.10.0+cu102
- Datasets 1.18.1
- Tokenizers 0.10.3
評価コマンド
mozilla-foundation/common_voice_8_0
のtest
スプリットで評価するには
python eval.py --model_id Akashpb13/Central_kurdish_xlsr --dataset mozilla-foundation/common_voice_8_0 --config ckb --split test
📄 ライセンス
このモデルはApache-2.0ライセンスの下で提供されています。