szymonrucinski/Curie-7B-v1

Curie-7B-v1

Introduction

This research demonstrates the potential of fine-tuning English Large Language Models (LLMs) for Polish text generation. By employing Language Adaptive Pre-training (LAPT) on a high-quality dataset of 3.11 GB (276 million Polish tokens) and subsequent fine-tuning on the KLEJ challenges, the Curie-7B-v1 model achieves remarkable performance. It not only generates Polish text with the lowest perplexity of 3.02 among decoder-based models but also rivals the best Polish encoder-decoder models closely, with a minimal performance gap on 8 out of 9 tasks. This was accomplished using about 2-3% of the dataset size typically required, showcasing the method's efficiency. The model is now open-source, contributing to the community's collaborative progress.

Language Adaptive Pre-training Dataset

The LAPT phase utilized the SpeakLeash dataset, a comprehensive collection of Polish texts, focusing on the highest quality extract of approximately 2 GB from the original 1TB.

Hardware and Software Stack

Experiments were conducted on a server featuring an NVIDIA RTX A6000 ADA GPU with 48GB of VRAM, AMD Epyc 7742 processor, and running Ubuntu with Pytorch 2.0 and CUDA 12.2.

The Adaptive Pre-training

The model was trained using AdamW optimizer, with specific hyperparameters aimed at optimizing performance. Training completed in one epoch, taking a total of 106 hours, demonstrating the onset of overfitting beyond this point.

Hyperparameters

• lora_rank: 32
• lora_dropout: 0.05
• lora_alpha: 16
• warmup_steps: 0.1
• learning_rate: 2.5 x 10^-5
• neftune_noise_alpha: 2
• batch_size: 128
• max_seq_len: 128

Fine-tuning for KLEJ Downstream Tasks

Curie-7B-v1 was exceptionally close to the best baseline models on 8 of 9 KLEJ tasks by using significantly less data, showcasing its efficiency and capability in handling a variety of NLP tasks in Polish.

Performance Highlights

• NKJP-NER: 93.4
• CDSC-E: 92.2
• CDSC-R: 94.9
• CBD: 49.0 (Demonstrating room for improvement)
• PolEmo2.0-IN: 92.7
• PolEmo2.0-OUT: 80.0
• DYK: 76.2
• PSC: 98.6
• AR: 86.8

Conclusions

The Curie-7B-v1 model, through LAPT, matches foundational models on eight downstream tasks with significantly less data. Its versatility in generating Polish text and the ability to be transformed into classifiers, regressors, and AI assistants highlights the method's effectiveness. This open-source Polish LLM provides a foundation for developing efficient business solutions.

Research Paper

Work and details regarding this model are described in the reserach paper Efficient Language Adaptive Pre-training: Extending State-of-the-Art Large Language Models for Polish by Szymon Ruciński.