CoolFace
Modelpublic

subhashtalluri/Telco_Transformer_V1

sourceHugging Facemitupdated 3y agoView on Hugging Face
3likes90downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

TelcoTransformerV1

The objective of the 'Telco Transformer' initiative is to pre-train a language model for the telecom industry to understand complex, contextual relationships in domain specific text data. Business hypothesis hinges on non-standard natural languages, such as components of a telecom system, technical terminology, and rich knowledge from multiple subdomains. This builds a strong case for pretraining a model from scratch. It constitutes a custom tokenizer to capture telco vocabulary, large scale unsupervised pre-training that is paired with supervised fine tuning to perform well on downstream tasks. This model will be able to complete sentences, and answer questions with accuracy that is superior to RAG or a fine tuned model.

Model V1 achieves the following results on the evaluation set:

  • Loss: 3.8171

Input Token Examples

RAN, IP Network, Radio, 5G, Core Network

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 3e-05
  • trainbatchsize: 16
  • evalbatchsize: 8
  • seed: 42
  • distributedtype: sagemakerdata_parallel
  • num_devices: 8
  • totaltrainbatch_size: 128
  • totalevalbatch_size: 64
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: cosine
  • lrschedulerwarmup_steps: 300
  • num_epochs: 30
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation Loss
5.33081.575005.2874
4.93783.1410004.8655
4.59294.7215004.5568
4.43086.2920004.3593
4.27037.8625004.2217
4.19779.4330004.1222
4.098611.0135004.0477
4.079112.5840003.9904
3.962514.1545003.9470
3.938115.7250003.9114
3.939917.355003.8844
3.914618.8760003.8640
3.877920.4465003.8468
3.84422.0170003.8355
3.836423.5875003.8266
3.856625.1680003.8216
3.841126.7385003.8187
3.81528.390003.8172
3.822529.8795003.8171

Framework versions

  • Transformers 4.28.1
  • Pytorch 2.0.0
  • Datasets 2.16.1
  • Tokenizers 0.13.3