Yi: Open Foundation Models by 01.AI
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , , , , , , , , , , , , , , , , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866929682637651968 |
|---|---|
| author | AI, 01. : Young, Alex Chen, Bei Li, Chao Huang, Chengen Zhang, Ge Zhang, Guanwei Wang, Guoyin Li, Heng Zhu, Jiangcheng Chen, Jianqun Chang, Jing Yu, Kaidong Liu, Peng Liu, Qiang Yue, Shawn Yang, Senbin Yang, Shiming Xie, Wen Huang, Wenhao Hu, Xiaohui Ren, Xiaoyi Niu, Xinyao Nie, Pengcheng Li, Yanpeng Xu, Yuchi Liu, Yudong Wang, Yue Cai, Yuxuan Gu, Zhenyu Liu, Zhiyuan Dai, Zonghong |
| author_facet | AI, 01. : Young, Alex Chen, Bei Li, Chao Huang, Chengen Zhang, Ge Zhang, Guanwei Wang, Guoyin Li, Heng Zhu, Jiangcheng Chen, Jianqun Chang, Jing Yu, Kaidong Liu, Peng Liu, Qiang Yue, Shawn Yang, Senbin Yang, Shiming Xie, Wen Huang, Wenhao Hu, Xiaohui Ren, Xiaoyi Niu, Xinyao Nie, Pengcheng Li, Yanpeng Xu, Yuchi Liu, Yudong Wang, Yue Cai, Yuxuan Gu, Zhenyu Liu, Zhiyuan Dai, Zonghong |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2403_04652 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Yi: Open Foundation Models by 01.AI AI, 01. : Young, Alex Chen, Bei Li, Chao Huang, Chengen Zhang, Ge Zhang, Guanwei Wang, Guoyin Li, Heng Zhu, Jiangcheng Chen, Jianqun Chang, Jing Yu, Kaidong Liu, Peng Liu, Qiang Yue, Shawn Yang, Senbin Yang, Shiming Xie, Wen Huang, Wenhao Hu, Xiaohui Ren, Xiaoyi Niu, Xinyao Nie, Pengcheng Li, Yanpeng Xu, Yuchi Liu, Yudong Wang, Yue Cai, Yuxuan Gu, Zhenyu Liu, Zhiyuan Dai, Zonghong Computation and Language Artificial Intelligence 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. |
| title | Yi: Open Foundation Models by 01.AI |
| topic | Computation and Language Artificial Intelligence |
| url | https://arxiv.org/abs/2403.04652 |