MétaCan
Menu
Back to cohort
Record W7034215471

Towards accurate low-bitwidth BERT

2023· dissertation· en· W7034215471 on OpenAlexaff

Bibliographic record

VenueeScholarship@McGill (McGill) · 2023
Typedissertation
Languageen
FieldSocial Sciences
TopicEducation, Innovation and Language Studies
Canadian institutionsMcGill University
Fundersnot available
KeywordsQuantization (signal processing)EncoderPruningComputational complexity theoryEdge deviceFocus (optics)Language model
DOInot available

Abstract

fetched live from OpenAlex

Transformer-based models or BERT (Bidirectional Encoder Representations from Transformers) models have seen a lot of success in natural language processing tasks in recent years.However, these networks are very difficult to deploy on edge devices where memory and compute resources are limited.To lower their computational and memory cost, pruning and quantization can be applied.However, naively applying these techniques to compress transformers, can cause a significant reduction in the accuracy of these models.This thesis will focus on improving the performance of quantization-aware training or improving the final performance or accuracy of a quantized transformer-based model without increasing the computational complexity of the forward pass of the network or slightly changing the network's architecture to increase the model capacity.In this thesis, we give one example of such and propose a more efficient quantization-aware training pipeline, called Efficient Two Stage Progressive Quantization in which the model with quantized weights is fine-tuned first, and at the same time, the bitwidth of the weights is lowered.Then, the model with quantized weights and activations will be fine-tuned.At Abstract ii the same time, we strategically choose which bitwidth to fine-tune on and to initialize from, and which bitwidth to fine-tune under augmented data.We are able to ternarize the weights of BERT, compressing it by 14.9x while outperforming the full-precision model performance across nearly all GLUE datasets, and without data augmentation, it can outperform existing BERT ternarization methods.Also, we outperform the state-of-the-art BERT binarization in performance without adding an extra module, saving 18% more memory than BinaryBERT.iii Abrg Les modles bass sur les transformateurs ou modles BERT (Bidirectional Encoder Representations from Transformers) ont connu beaucoup de succs dans les tches de traitement du langage naturel ces dernires annes.Cependant, ces rseaux sont trs difficiles dployer sur des priphriques priphriques o les ressources de mmoire et de calcul sont limites.Pour rduire leur cot de calcul et de mmoire, l'lagage et la quantification peuvent tre appliqus.Cependant, l'application nave de ces techniques pour compresser les transformateurs peut entraner une rduction significative de la prcision de ces modles.Cette thse se concentrera sur l'amlioration des performances de la formation sensible la quantification ou sur l'amlioration des performances finales ou de la prcision d'un modle bas sur un transformateur quantifi sans augmenter la complexit de calcul de la passe avant du rseau ou en modifiant lgrement l'architecture du rseau pour augmenter la capacit du modle.Dans cette thse, nous en donnons un exemple et proposons un pipeline de formation sensible la quantification plus efficace, appel Efficient Two Stage Progressive Quantization Abrg iv dans lequel le modle avec des poids quantifis est affin en premier, et en mme temps, la largeur de bit du les poids sont abaisss.Ensuite, le modle avec les poids quantifis et les activations sera affin.En mme temps, nous choisissons stratgiquement la largeur de bit affiner et partir de laquelle initialiser, et la largeur de bit affiner sous les donnes augmentes.Nous sommes en mesure de ternariser les poids de BERT, en le compressant de 14,9x tout en surpassant les performances du modle de prcision totale sur presque tous les ensembles de donnes GLUE, et sans augmentation des donnes, il peut surpasser les mthodes de ternarisation BERT existantes.De plus, nous surpassons la binarisation BERT de pointe en termes de performances sans ajouter de module supplmentaire, ce qui permet d'conomiser 18% de mmoire en plus que BinaryBERT.v List of Tables xi 3.10 Performance of using Direction Mismatch Distillation (DMD) versus using Mean Square Error (MSE) Distillation. . . . . . . . . . . . . . . . . . . . .

Fetched live from OpenAlex and de-inverted. Abstracts are not stored in this database: the inverted indexes are 8.6 GB of the frame’s 9.3 GB of text, and the host has 13 GB free.

How this classification was reachedexpand

Full frame distilled prediction

Teacher imitation

Not calibrated prevalence, not ground truth. Human validation pending. Learned from the 10,348 direct Codex labels and 10,348 direct Gemma labels. Candidate is the union of thresholded teacher heads; consensus is their intersection. These outputs are machine_predicted_unvalidated and are not human labels or direct frontier model labels.

metaresearch head score (Codex)0.002
metaresearch head score (Gemma)0.004
Version: codex-gemma-dda1882f352aValidation status: machine_predicted_unvalidated
Candidate categoriesMeta-epidemiology (narrow), Science and technology studies, Insufficient payload (model declined to judge)
Consensus categoriesInsufficient payload (model declined to judge)
DomainCandidate signal: none · Consensus signal: none
Study designCandidate signal: Not applicable · Consensus signal: none
GenreCandidate signal: Empirical · Consensus signal: Empirical
Teacher disagreement score0.864
Threshold uncertainty score1.000

Codex and Gemma teacher scores by category

CategoryCodexGemma
Metaresearch0.0020.004
Meta-epidemiology (narrow)0.0010.001
Meta-epidemiology (broad)0.0010.000
Bibliometrics0.0010.002
Science and technology studies0.0040.000
Scholarly communication0.0000.001
Open science0.0010.000
Research integrity0.0010.001
Insufficient payload (model declined to judge)0.0050.001

Machine scores (provisional)

The two teacher heads of the student model, read on this work. A score orders the frame for review; it never asserts a category, and the validation status ships verbatim with every row.

Baseline scores from an immature model (maturity gate not passed, 7 training rounds). Scores rank; they never assert a category.

Opus teacher head0.039
GPT teacher head0.341
Teacher spread0.302 · how far apart the two teachers sit on this one work
Validation statusscore_only:v0-immature-baseline · verbatim from the scoring run: score_only means the number may rank works, and no category label ships from it

Classification

machine, unvalidated

Machine predicted; both teacher heads agree on what is shown here.

Study designNot applicable
Domainnot available
GenreEmpirical

How this classification was reached, model by model and score by score, is at the end of the page under "How this classification was reached".

Quick stats

Citations0
Published2023
Admission routes1
Has abstractyes

Explore more

Same venueeScholarship@McGill (McGill)Same topicEducation, Innovation and Language StudiesFrench-language works237,207