Yi: Open Foundation Models by 01.AI

AI-generated keywords: Yi model family language and multimodal models data-engineering efforts vision-language modeling optimized data processing techniques

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.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

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

License: CC BY-SA 4.0

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.

Submitted to arXiv on 07 Mar. 2024

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2403.04652v1

The , consisting of a series of language and multimodal models, has demonstrated impressive multi-dimensional capabilities. These models are built upon 6B and 34B pretrained language models and have been extended to include chat models, 200K long context models, depth-upscaled models, and vision-language models. The performance of these base models has been exceptional on various benchmarks such as MMLU. In fact, the finetuned chat models have achieved high human preference rates on evaluation platforms like AlpacaEval and Chatbot Arena. This success can be attributed to the high-quality data that is a result of extensive . To construct this massive corpus of 3.1 trillion tokens in English and Chinese for pretraining purposes, a cascaded data deduplication and quality filtering pipeline was used. During finetuning, a small-scale instruction dataset was meticulously polished by machine learning engineers through multiple iterations. In terms of , the chat language model was combined with a vision transformer encoder to align visual representations with the semantic space of the language model. Additionally, through lightweight continual pretraining, the context length was extended to 200K resulting in strong needle-in-a-haystack retrieval performance. Continual pretraining also proved beneficial in enhancing performance by extending the depth of pretrained checkpoints. Moving forward, it is evident that scaling up model parameters using will lead to even stronger frontier models. To further improve the capabilities of Yi base models in long-context modeling, visual understanding, and depth upscaling post-training methods are being explored. To ensure high-quality bilingual pretraining data for Yi models, a cascaded data-processing pipeline was meticulously designed. This involved heuristic rule filters to remove low-quality text content from web documents sourced from Common Crawl. This process guarantees that only relevant and diverse data is included in the training corpus, ensuring the best possible results for Yi models. Overall, our research showcases the potential for further advancements in language and multimodal modeling by leveraging scalable infrastructure and optimized data processing techniques.
Created on 06 Jul. 2024

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while 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.