HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Continual Training of Language Models for Few-Shot Learning

Zixuan Ke Haowei Lin Yijia Shao Hu Xu Lei Shu Bing Liu

Continual Training of Language Models for Few-Shot Learning

Abstract

Recent work on applying large language models (LMs) achieves impressive performance in many NLP applications. Adapting or posttraining an LM using an unlabeled domain corpus can produce even better performance for end-tasks in the domain. This paper proposes the problem of continually extending an LM by incrementally post-train the LM with a sequence of unlabeled domain corpora to expand its knowledge without forgetting its previous skills. The goal is to improve the few-shot end-task learning in these domains. The resulting system is called CPT (Continual PostTraining), which to our knowledge, is the first continual post-training system. Experimental results verify its effectiveness.

Code Repositories

UIC-Liu-Lab/ContinualLM
pytorch
Mentioned in GitHub
uic-liu-lab/cpt
Official
pytorch
Mentioned in GitHub
zixuanke/pycontinual
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
continual-pretraining-on-ag-newsCPT
F1 - macro: 63.77

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Continual Training of Language Models for Few-Shot Learning | Papers | HyperAI