01 AI logo
01 AI

yi-lightning

Released Oct 2024

Yi-Lightning is a flagship large language model developed by 01.AI, the artificial intelligence company founded by Kai-Fu Lee. Utilizing an enhanced Mixture-of-Experts (MoE) architecture, the model is designed to provide high-performance reasoning and linguistic capabilities while optimizing for inference speed and cost-efficiency. It serves as the top-tier model in the Yi series, following previous iterations like Yi-Large and Yi-1.5.\n\nUpon its release in October 2024, Yi-Lightning achieved notable recognition on the LMSYS Chatbot Arena leaderboard, where it ranked among the top global models. It demonstrated competitive performance against frontier models such as GPT-4o and Claude 3.5 Sonnet, particularly excelling in specialized categories including Chinese language processing, mathematics, and coding. The model's training process involved multi-stage supervised fine-tuning (SFT) and reinforcement learning from human feedback (RLHF) to align its outputs with human preferences.\n\nArchitecturally, Yi-Lightning features advanced expert segmentation and routing mechanisms alongside optimized KV-caching techniques to reduce latency. According to 01.AI, the model offers a significant improvement in generation speed compared to its predecessor, Yi-Large, while substantially lowering the cost for developers. It also incorporates the RAISE (Responsible AI Safety Engine) framework to address safety across its pre-training and serving phases.

Rankings & Comparison