We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CL

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Computation and Language

Title: Applying SoftTriple Loss for Supervised Language Model Fine Tuning

Abstract: We introduce a new loss function TripleEntropy, to improve classification performance for fine-tuning general knowledge pre-trained language models based on cross-entropy and SoftTriple loss. This loss function can improve the robust RoBERTa baseline model fine-tuned with cross-entropy loss by about (0.02% - 2.29%). Thorough tests on popular datasets indicate a steady gain. The fewer samples in the training dataset, the higher gain -- thus, for small-sized dataset it is 0.78%, for medium-sized -- 0.86% for large -- 0.20% and for extra-large 0.04%.
Subjects: Computation and Language (cs.CL)
Journal reference: 17th Conference on Computer Science and Intelligence Systems 2022. Series: ACSIS Annals of Computer Science and Information Systems
DOI: 10.15439/2022F185
Cite as: arXiv:2112.08462 [cs.CL]
  (or arXiv:2112.08462v1 [cs.CL] for this version)

Submission history

From: Witold Sosnowski [view email]
[v1] Wed, 15 Dec 2021 20:21:20 GMT (74kb,D)

Link back to: arXiv, form interface, contact.