This model essentially explores having different experts (MoE) for image encoder part of vision language model. How? š§ The authors concatenate the vision encoder output tokens together, and they apply "pre-alignment" essentially fine-tune experts with frozen text encoder.
Then they freeze both experts and the decoder and just train the projection layer, and finally, they unfreeze everything for supervised fine-tuning āØ
In the paper, they explore different fusion strategies and vision encoders, extending basic CLIP encoder, and figure out simply concatenating visual tokens works well. Rest of the architecture is quite similar to LLaVA. (see below the architecture)
Good folks at Meta has just unveiled Llama 3.2, pushing the boundaries of language models and computer vision.
Even more interesting is how they trained this cutting-edge model:
1ļøā£ Architecture: Llama 3.2 uses an optimized transformer architecture with auto-regressive capabilities. The largest models (11B and 90B) now support multimodal inputs, integrating both text and images.
2ļøā£ Training Pipeline: ā¢ Started with pretrained Llama 3.1 text models ā¢ Added image adapters and encoders ā¢ Pretrained on large-scale noisy (image, text) pair data ā¢ Fine-tuned on high-quality in-domain and knowledge-enhanced (image, text) pairs
3ļøā£ Vision Integration: ā¢ Trained adapter weights to integrate a pre-trained image encoder ā¢ Used cross-attention layers to feed image representations into the language model ā¢ Preserved text-only capabilities by not updating language model parameters during adapter training
4ļøā£ Post-Training Alignment: ā¢ Multiple rounds of supervised fine-tuning (SFT) ā¢ Rejection sampling (RS) ā¢ Direct preference optimization (DPO) ā¢ Synthetic data generation using Llama 3.1 for Q&A augmentation ā¢ Reward model ranking for high-quality fine-tuning data
5ļøā£ Lightweight Models: ā¢ Used pruning and distillation techniques for 1B and 3B models ā¢ Structured pruning from Llama 3.1 8B model ā¢ Knowledge distillation using Llama 3.1 8B and 70B as teachers
6ļøā£ Context Length: All models support an impressive 128K token context length.
7ļøā£ Safety Measures: Incorporated safety mitigation data to balance helpfulness and safety.
The result? A suite of models ranging from edge-friendly 1B parameters to powerful 90B parameter versions, capable of sophisticated reasoning across text and images. Llama 3.2 is set to revolutionize AI applications from mobile devices to enterprise-scale solutions.
What are your thoughts on these advancements? How do you see Llama 3.2 impacting your industry? Let's discuss in the comments!
just landed at Hugging Face Hub: community-led computer vision course šš¤ learn from fundamentals to details of the bleeding edge vision transformers!