Robust-U1
Collection
5 items • Updated • 1
How to use Jiaqi-hkust/Robust-U1 with Transformers:
# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("Jiaqi-hkust/Robust-U1", trust_remote_code=True, dtype="auto", device_map="auto")Robust-U1 is a unified Multimodal Large Language Model (MLLM) that can self-recover corrupted visual content and perform multimodal reasoning over it. This enables robust visual understanding even under real-world image degradations.
| Checkpoint | Link | Note |
|---|---|---|
| BAGEL-7B-MoT | ByteDance-Seed/BAGEL-7B-MoT | Used as initial weights for training. |
| Robust-U1 | Jiaqi-hkust/Robust-U1 | Final model for visual self-recovery and multimodal reasoning. |
| Robust-U1-RL | Jiaqi-hkust/Robust-U1-RL | Fine-tuned with reinforcement learning. |
| Robust-U1-SFT | Jiaqi-hkust/Robust-U1-SFT | Fine-tuned with supervised learning. |
If you find this repository useful, please cite the paper:
@inproceedings{tang2026robustu1,
title={Robust-U1: Can MLLMs Self-Recover Corrupted Visual Content for Robust Understanding?},
author={Tang, Jiaqi and Chen, Jianmin and Zhai, Youyang and Wei, Wei and Liu, Runtao and Zhao, Mengjie and Wu, Xiangyu and Xiao, Qingfa and Chen, Qifeng},
booktitle={Proceedings of the 43rd International Conference on Machine Learning (ICML)},
year={2026},
}