Research map / Vision-language and multimodal / Cross-modal learning
UNIALIGN: Scaling Multimodal Alignment within One Unified Model
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes UNIALIGN, a single modality-aware MoE+LoRA encoder initialized from CLIP that aligns six modalities in one training phase using unpaired cross-dataset class-label similarity and distillation from anchor modalities and partial-modal foundation models.
Problem
Multimodal alignment typically needs a separate encoder and separate training per extended modality, so parameters and training cost grow with modality count and cross-modal knowledge transfer is limited.
Approach
A unified modality-aware mixture-of-experts with LoRA experts on a frozen CLIP ViT-B, trained jointly on unpaired data from all datasets via soft modality binding (class-label similarity as soft targets) and knowledge distillation from anchor modalities and partial-modal teacher models.
Key results
- Achieves 85.3% top-1 on ImageNet-1K with 7.8M trainable parameters, comparable to Hiera's 85.2% with 70.0M.
- Reaches 71.4% top-1 on NYU-D depth scene classification, surpassing ImageBind (54.0) and VIT-LENS (68.5) in the table.
- Scores 97.4% on Speech Commands V2 versus 97.0% for MetaFormer (86.3M trainable), and 68.5% zero-shot on UCF101 outperforming ImageBind's 64.8.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| modality-aware expert grouping instead of global routing in the MoE (ablation) | UNIALIGN MoE layers · SUN-D and ModelNet40 | Top-1 Accuracy: +0.8 SUN-D, +0.6 ModelNet40 | Helped |
| soft modality binding loss (L_soft bind) added (ablation) | UNIALIGN training objective · SUN-D and ModelNet40 | Top-1 Accuracy: +1.4 SUN-D, +1.5 ModelNet40 | Helped |
| anchor-modality knowledge distillation (L_anch distil) added (ablation) | UNIALIGN training objective · SUN-D and ModelNet40 | Top-1 Accuracy: +0.5 SUN-D, +0.8 ModelNet40 | Helped |
| partial-modal foundation model distillation (L_model distil) added (ablation) | UNIALIGN training objective · SUN-D and ModelNet40 | Top-1 Accuracy: +3.5 SUN-D, +15.0 ModelNet40 | Helped |
| adding more modalities with only L_NCE (image, text, depth, video, audio, point cloud) (ablation) | UNIALIGN with plain InfoNCE training · ImageNet-1K, SUN-D, ModelNet40 | Top-1 Accuracy: SUN-D 34.4 to 40.4; ImageNet-1K 84.7 to 85.1; ModelNet40 31.6 to 38.6 with I+T+D+P+V+A | Mixed |
| full training objectives (soft binding + distillation) on all six modalities (ablation) | UNIALIGN with all modalities · ImageNet-1K, SUN-D, ModelNet40 | Top-1 Accuracy: SUN-D 40.4 to 45.8; ModelNet40 38.6 to 55.9; ImageNet-1K 85.1 to 85.3 | Helped |
| text-only anchor instead of image+text for depth alignment | UNIALIGN depth alignment · NYU-D | Top-1 Accuracy: 71.4 to 46.3 | Hurt |
| UNIALIGN with MoE and LoRA, shared across 6 modalities, trained once with 7.8M trainable params | UNIALIGN vs. single-task SOTA on ImageNet-1K · ImageNet-1K val | Top-1 Accuracy: 85.3 vs 85.2 (Hiera) and 85.4 (MetaFormer, 86.6M params) | No clear effect |
| zero-shot 3D shape classification with unified model vs. modality-specific point cloud models | UNIALIGN vs ULIP and VIT-LENS · ModelNet40 and ScanObjectNN | Top-1 Accuracy: 55.9 vs 60.4 (ULIP) and 65.4 (VIT-LENS) on ModelNet40 | Hurt |
Surprising findings
- Naively adding modalities with only the contrastive loss degrades ImageNet-1K-independent performance on depth and point cloud tasks (SUN-D 40.4 with 6 modalities vs 34.4 with 3 modalities on ablation rows, showing mixed effects of scaling).
- Aligning depth only to text yields much lower accuracy (46.3 vs 71.4 with image+text anchors on NYU-D), suggesting depth aligns more closely with images.
Limitations
- Performance trails specialized SOTA on several tasks (e.g., zero-shot 3D classification, audio classification, audio-text retrieval) partly due to less training data (0.5M AudioSet videos).
- Experiments limited to two experts per modality and top-2 routing due to GPU memory constraints.
Experimental setup
Datasets: ImageNet-1K ShapeNet SUN RGB-D AudioSet VGGSound ESC-50 AudioCaps MSR-VTT UCF101 ModelNet40 ScanObjectNN NYU-D
Benchmarks: ImageNet-1K ModelNet40 ScanObjectNN NYU-D SUN RGB-D AudioSet ESC-50 AudioCaps Speech Commands V2 MSR-VTT UCF101
Metrics: top-1 accuracy top-5 accuracy mAP Recall@k trainable parameters
Baselines: ImageBind VIT-LENS LanguageBind ULIP MetaFormer Hiera AudioCLIP UniBind
Builds on: CLIP ImageBind LoRA Mixture-of-Experts ULIP
Closest papers by meaning
- With Limited Data for Multimodal Alignment, Let the STRUCTURE Guide You (NeurIPS 2025 · similarity 0.90)
- Strong Helps Weak: Directional Cross-Modal Alignment Transfer in Multi-modal LLMs (NeurIPS 2026 · similarity 0.90)
- DeepAlign: Mitigating Modality Conflict through Modality-Specific Alignment (CVPR 2026 · similarity 0.90)
- MSP: Modality Self-Play for Training Multimodal LLMs Without Paired Cross-Modal Data (NeurIPS 2026 · similarity 0.89)
- OneLLM: One Framework to Align All Modalities with Language (CVPR 2024 · similarity 0.89)
- AlignMamba: Enhancing Multimodal Mamba with Local and Global Cross-modal Alignment (CVPR 2025 · similarity 0.89)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
