NExT-GPT: Any-to-Any Multimodal LLM

Shengqiong Wu, Hao Fei, Leigang Qu, Wei Ji, Tat-Seng Chua

2023 · ICML

NExT-GPT: Any-to-Any Multimodal LLM

Problem

Framing

Multimodal LLMs mostly stop at understanding or call external generators in a brittle cascade. NExT-GPT closes this gap by aligning one LLM to both multimodal encoders and diffusion decoders, enabling any-to-any input and output with only about 1% trainable parameters.

Currently Used Methods

Direct antecedents

Proposed Method

Architecture

NExT-GPT connects modality encoders, lightweight projection layers, a Vicuna-7B LLM, modality-specific output projectors, and frozen diffusion decoders. Image, audio, and video inputs are encoded by ImageBind; outputs are generated by Stable Diffusion, AudioLDM, and Zeroscope. Only the grouping/projection modules and LoRA adapters are updated.

System diagram: image, audio, and video encoders feed trainable input projections into a central LLM, whose trainable output projections drive frozen image, audio, and video diffusion decoders.

Loss / Objective

Training uses cross-entropy captioning and instruction-following losses across three stages.

L=t=1Tlogpθ(yty<t,x)\mathcal{L} = - \sum_{t=1}^{T} \log p_{\theta}(y_t \mid y_{<t}, x)

Sampling Rule / Algorithm

Inference maps LLM output representations into modality-specific signal tokens, then invokes the matched diffusion decoder.

z(m)=Pout(m) ⁣(hLLM),x^(m)D(m) ⁣(z(m)),m{img,aud,vid}\mathbf{z}^{(m)} = P_{\mathrm{out}}^{(m)}\!\left(h_{\mathrm{LLM}}\right), \qquad \hat{x}^{(m)} \sim D^{(m)}\!\left(\mathbf{z}^{(m)}\right), \quad m \in \{\mathrm{img}, \mathrm{aud}, \mathrm{vid}\}

Training Procedure

Evaluation

Datasets

Metrics

Headline results

Table 4: Zero-shot multimodal understanding results across image captioning, question answering, and evaluation-only benchmarks.

MethodLLMNoCapsFlickr30KCOCOVQAv2VizWizOKVQAMMBSEED
NExT-GPTVicuna-7B123.784.5124.966.748.452.158.057.5

Ablations

Qualitative results: multi-turn conversations where NExT-GPT answers about video content, generates audio, and returns illustrative images.

Method Strengths and Weaknesses

Strengths

Weaknesses

Suggestions from the authors

Links

Prior Papers

Further Papers