A small calibration model trained on non-toxic text is aligned and fused into the final layer of LLaMA-2-based LLMs, modestly reducing toxicity on RealToxicityPrompts but with mixed perplexity results.
Ziya2: Data-centric Learning is All LLMs Need
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Various large language models (LLMs) have been proposed in recent years, including closed- and open-source ones, continually setting new records on multiple benchmarks. However, the development of LLMs still faces several issues, such as high cost of training models from scratch, and continual pre-training leading to catastrophic forgetting, etc. Although many such issues are addressed along the line of research on LLMs, an important yet practical limitation is that many studies overly pursue enlarging model sizes without comprehensively analyzing and optimizing the use of pre-training data in their learning process, as well as appropriate organization and leveraging of such data in training LLMs under cost-effective settings. In this work, we propose Ziya2, a model with 13 billion parameters adopting LLaMA2 as the foundation model, and further pre-trained on 700 billion tokens, where we focus on pre-training techniques and use data-centric optimization to enhance the learning process of Ziya2 on different stages. We define three data attributes and firstly establish data-centric scaling laws to illustrate how different data impacts LLMs. Experiments show that Ziya2 significantly outperforms other models in multiple benchmarks especially with promising results compared to representative open-source ones. Ziya2 (Base) is released at https://huggingface.co/IDEA-CCNL/Ziya2-13B-Base and https://modelscope.cn/models/Fengshenbang/Ziya2-13B-Base/summary.
fields
cs.CL 1years
2025 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Detoxification of Large Language Models through Output-layer Fusion with a Calibration Model
A small calibration model trained on non-toxic text is aligned and fused into the final layer of LLaMA-2-based LLMs, modestly reducing toxicity on RealToxicityPrompts but with mixed perplexity results.