Ad

LLaDA: Diffusion Language Models

LLaDA introduces Large Language Diffusion with Masking, an 8B parameter diffusion model trained from scratch, rivaling LLaMA3 in performance. It explores a novel approach to language modeling using diffusion techniques.
Screenshot of ML-GSAI/LLaDA homepage

LLaDA introduces Large Language Diffusion with Masking (LLaDA), a novel language model approach leveraging diffusion models. LLaDA is a diffusion-based language model with an unprecedented 8B scale, trained entirely from scratch, and rivals LLaMA3 8B in performance. It addresses the limitations of autoregressive models by utilizing a different probabilistic approach to language modeling, offering a complementary path to scaling language models.

LLaDA introduces a new approach to language modeling based on diffusion techniques, achieving competitive performance with autoregressive models. It demonstrates strong in-context learning and zero-shot capabilities. The project provides comprehensive evaluation code and guidelines for training and fine-tuning. LLaDA-MoE-7B-A1B-Base is the first diffusion language model pretrained from scratch with MoE architecture.

  • Core Functionality: Generates text using a novel masked diffusion approach, challenging the dominance of autoregressive models.
  • Supported Platforms: Primarily designed for research and experimentation with Python.
  • Configuration/Extensibility: Offers straightforward guidelines for adaptation and training using standard LLM frameworks.
  • Performance/Scalability: Achieves competitive performance with LLaMA3 8B, and LLaDA-MoE-7B-A1B-Instruct demonstrates efficiency with MoE architecture.
  • Developer Experience: Provides clear documentation, evaluation code, and a Gradio demo for easy experimentation.

The project is actively developed with recent updates including new model releases (LLaDA-V, LLaDA 1.5, LLaDA-MoE-7B-A1B-Base), evaluation code support, and batch inference. The codebase is relatively new, but maintenance and community engagement show active development. Extensive documentation and a clear FAQ are available.

LLaDA benefits researchers and developers interested in exploring alternative language modeling approaches beyond traditional autoregressive models. It provides a valuable open-source resource for experimenting with diffusion-based language models, offering a pathway towards improved efficiency and scalability. It is particularly relevant to those seeking to understand and implement the theory and practice of masked language modeling.

Summarize:
Share:
Stars
3,879
Forks
269
Issues
92
Created
1 year ago
Commit
28 days ago
License
MIT
Archived
No
Updated 16 days ago

Similar Repositories