Join the conversation

Join the community of Machine Learners and AI enthusiasts.

Sign Up
singhsidhukuldeepΒ 
posted an update May 31
Post
1511
Remember Gemini, GPT-4o, all being true multimodal models 🌟.

Now we have a paper πŸ“„ describing an architecture that might achieve that!

Uni-MoE: a native multimodal, Unified Mixture of Experts (MoE) architecture πŸ—οΈ.

Uni-MoE integrates various modalities (text πŸ“, image πŸ–ΌοΈ, audio 🎡, video πŸ“Ή, speech πŸ—£οΈ) using modality-specific encoders and connectors for a cohesive multimodal understanding.

Training Strategy:
1️⃣ Training cross-modality alignment with diverse connectors πŸ”„.
2️⃣ Training modality-specific experts using cross-modality instruction data πŸ“Š.
3️⃣Tuning the Uni-MoE framework with Low-Rank Adaptation (LoRA) on mixed multimodal data πŸ”§.

Technical Details:

Modality-Specific Encoders: CLIP for images πŸ–ΌοΈ, Whisper for speech πŸ—£οΈ, BEATs for audio 🎡.

MoE-Based Blocks: Shared self-attention layers, feed-forward networks (FFN) based experts, and sparse routers for token-level expertise allocation πŸš€.

Efficient Training: Utilizes LoRA for fine-tuning pre-trained experts and self-attention layers πŸ› οΈ.

Uni-MoE outperforms traditional dense models on benchmarks like A-OKVQA, OK-VQA, VQAv2, MMBench, RACE-Audio, and English High School Listening Test πŸ†.

The code is open-sourced as well: https://github.com/HITsz-TMG/UMOE-Scaling-Unified-Multimodal-LLMs/tree/master/Uni_MoE_v2

Paper: Uni-MoE: Scaling Unified Multimodal LLMs with Mixture of Experts (2405.11273)