Yi: Open Foundation Models by 01.AI
Authors: 01. AI, :, Alex Young, Bei Chen, Chao Li, Chengen Huang, Ge Zhang, Guanwei Zhang, Heng Li, Jiangcheng Zhu, Jianqun Chen, Jing Chang, Kaidong Yu, Peng Liu, Qiang Liu, Shawn Yue, Senbin Yang, Shiming Yang, Tao Yu, Wen Xie, Wenhao Huang, Xiaohui Hu, Xiaoyi Ren, Xinyao Niu, Pengcheng Nie, Yuchi Xu, Yudong Liu, Yue Wang, Yuxuan Cai, Zhenyu Gu, Zhiyuan Liu, Zonghong Dai
AI-generated Key Points
- The Yi base models consist of a series of language and multimodal models with impressive multi-dimensional capabilities.
- These models are built upon 6B and 34B pretrained language models and have been extended to include various specialized models such as chat models, long context models, depth-upscaled models, and vision-language models.
- The performance of these base models has been exceptional on benchmarks like MMLU, with finetuned chat models achieving high human preference rates on evaluation platforms like AlpacaEval and Chatbot Arena.
- High-quality data is attributed to the success of these models, achieved through constructing a massive corpus of 3.1 trillion tokens in English and Chinese for pretraining purposes using cascaded data deduplication and quality filtering pipeline.
- Continual pretraining techniques have been utilized to extend context length to 200K, align visual representations with the semantic space of the language model, and enhance performance by extending the depth of pretrained checkpoints.
- Scaling up model parameters using continual pretraining will lead to even stronger frontier models in the future.
- Post-training methods are being explored to further improve capabilities in long-context modeling, visual understanding, and depth upscaling for Yi base models.
- A meticulous cascaded data-processing pipeline was designed to ensure high-quality bilingual pretraining data for Yi models by removing low-quality text content from web documents sourced from Common Crawl.
Abstract: We introduce the Yi model family, a series of language and multimodal models that demonstrate strong multi-dimensional capabilities. The Yi model family is based on 6B and 34B pretrained language models, then we extend them to chat models, 200K long context models, depth-upscaled models, and vision-language models. Our base models achieve strong performance on a wide range of benchmarks like MMLU, and our finetuned chat models deliver strong human preference rate on major evaluation platforms like AlpacaEval and Chatbot Arena. Building upon our scalable super-computing infrastructure and the classical transformer architecture, we attribute the performance of Yi models primarily to its data quality resulting from our data-engineering efforts. For pretraining, we construct 3.1 trillion tokens of English and Chinese corpora using a cascaded data deduplication and quality filtering pipeline. For finetuning, we polish a small scale (less than 10K) instruction dataset over multiple iterations such that every single instance has been verified directly by our machine learning engineers. For vision-language, we combine the chat language model with a vision transformer encoder and train the model to align visual representations to the semantic space of the language model. We further extend the context length to 200K through lightweight continual pretraining and demonstrate strong needle-in-a-haystack retrieval performance. We show that extending the depth of the pretrained checkpoint through continual pretraining further improves performance. We believe that given our current results, continuing to scale up model parameters using thoroughly optimized data will lead to even stronger frontier models.
Tools for this paper
Ask questions about this paper to our AI assistant
You can also chat with multiple papers at once here.
The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.
Similar papers summarized with our AI tools
Navigate through even more similar papers through a
tree representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.