---
title: MLLM-Engaged Joint Optimization (MEJO)
url: https://www.emergentmind.com/topics/mllm-engaged-joint-optimization-mejo
type: topic
---

# MLLM-Engaged Joint Optimization (MEJO)

Searching arXiv for the cited papers and related "MLLM-Engaged Joint Optimization" usage.
MLLM-Engaged Joint Optimization (MEJO) denotes a family of multimodal learning and inference schemes in which a multimodal large language model (MLLM) is not treated as a frozen instruction parser alone, but as an actively engaged component whose language reasoning, visual understanding, retrieval behavior, sampling policy, task heads, or serving controls are optimized jointly with other modules under shared objectives. Across recent work, the term has been used in image editing, zero-shot composed image retrieval, long-form video question answering, surgical triplet recognition, autonomous driving, and MLLM serving analysis. The common thread is that MLLM-derived signals are made operational inside the optimization loop rather than appended as post hoc prompts, with joint objectives intended to reduce modality gaps, improve task alignment, and coordinate heterogeneous subproblems [2505.19149] [2505.19707] [2602.22932] [2509.12893] [2512.12799] [2512.22695].

## 1. Conceptual scope and defining properties

In current usage, MEJO is a paradigm rather than a single algorithm. Its defining structure is jointness: multiple signals or modules that would otherwise be trained separately are optimized together so that the MLLM’s internal representations become directly useful for downstream control, alignment, or prediction. In "MIND-Edit" the MLLM jointly produces an optimized instruction $T_{\text{opt}}$ and an MLLM-derived visual embedding $v_{\text{insight}}$ for diffusion-based editing [2505.19149]. In "MVFT-JI" the MLLM synthesizes supervision for two complementary retrieval tasks, while inference fuses VLM and MLLM signals through late fusion [2505.19707]. In "MSJoE" the MLLM and a lightweight sampler are jointly optimized through reinforcement learning so that query reasoning and key-frame selection co-adapt [2602.22932].

A second defining property is active engagement of the MLLM in task structure. The MLLM may clarify ambiguous instructions, generate target texts, reason out visually grounded sub-queries, provide expert-level semantic prompts, or share parameters with dense spatial heads. This differs from pipelines in which an MLLM is merely prompted once and left outside training. In surgical triplet recognition, MEJO explicitly targets both inter-task and intra-task optimization conflicts through shared-specific disentanglement and coordinated gradient learning [2509.12893]. In autonomous driving, DrivePI uses a trainable Qwen2.5-VL 0.5B backbone whose shared hidden states feed text, occupancy, flow, and planning heads in parallel under a single joint loss [2512.12799].

A third property is task-dependent coupling between the MLLM and auxiliary modules. The auxiliary component can be a diffusion model interface, a Q-Former, a sampler, prompt-integration modules, dense voxel decoders, or stage-wise serving controls. This suggests that MEJO is best understood as a design pattern for cross-module co-optimization, not as a fixed architectural template.

## 2. Recurring architectural patterns

One recurrent pattern is dual-path conditioning. In MIND-Edit, the MLLM produces two complementary outputs from the same source image $I_{\text{src}}$ and raw instruction $T_{\text{raw}}$: the language head $\mathbf{F}_t$ generates $T_{\text{opt}}$, while the vision head $\mathbf{F}_v$ maps intermediate hidden states to $v_{\text{insight}}$. These two signals then enter decoupled cross-attention branches of Stable Diffusion v1.5 through IP-Adapter, with fused output
$$
Z_{\text{new}} = Z' + Z''.
$$
The design keeps the diffusion model fixed and injects language and image guidance in parallel, rather than forcing a single textual condition to carry all editing intent [2505.19149].

A second pattern is shared latent query spaces. MVFT-JI removes pseudo-text adapters and directly fine-tunes the VLM’s Q-Former using shared learnable query tokens $\{q_j\}_{j=1..k}$. The same query-token machinery is used for composed queries, target texts, captions, and images, so that multiple tasks constrain a common representational substrate. Similarity is computed through token-level cosine similarity with max-over-target-token aggregation, and both training tasks operate on this shared token geometry [2505.19707].

A third pattern is reasoning-to-selection coupling. MSJoE first uses the MLLM to generate up to $N_q \le 4$ distinct queries from a sparse preview, then constructs a query-frame similarity matrix
$$
S_{ij}=\operatorname{cos}(t_i,v_j),
$$
and finally feeds $S$ to a 1D U-Net sampler of approximately $2$M parameters. The sampler predicts per-frame probabilities for without-replacement key-frame selection, after which the MLLM answers from the selected high-resolution frames. Query reasoning, frame selection, and answer generation therefore form a closed loop rather than a one-way preprocessing stack [2602.22932].

A fourth pattern is explicit decomposition of shared and task-specific subspaces. In surgical triplet recognition, the Shared-Specific-Disentangled (S$^2$D) scheme assigns expert-level semantic augmentation to a shared branch through GMM Prompt Integration (GPI) and prefix tuning, while a Temporal-Spatial Prompt (TSP) adapter constructs task-specific features for instrument, verb, target, and triplet heads. The MLLM contribution appears in a probabilistic prompt pool built from GPT-4o-generated instrument-anchored attributes encoded by CLIP [2509.12893].

A fifth pattern is unified multimodal tokenization for dense spatial outputs. DrivePI converts multi-view images and LiDAR into a latent BEV feature $F_{\text{bev}} \in \mathbb{R}^{H \times W \times C}$, patchifies it, performs cross-attention between pooled patch summaries and patch features, and linearly projects the result into MLLM tokens. The same fused hidden states support text QA, 3D occupancy, occupancy flow, and action diffusion in parallel. Here MEJO is realized as end-to-end joint training over a single 4D spatial-aware MLLM backbone [2512.12799].

## 3. Optimization formulations

MEJO methods span supervised multi-objective training, contrastive joint learning, reinforcement learning, gradient coordination, and serving-time constrained optimization.

In MIND-Edit, the MLLM is jointly fine-tuned with a language-generation loss and a visual-embedding alignment loss:
$$
\ell_{\text{text}}
= -\sum_{t=1}^{n} \log P\big(w_t \mid w_{<t}, I_{\text{src}}, T_{\text{raw}}\big),
$$
$$
\ell_{\text{embed}}
= 1 - \cos\!\left(v_{\text{insight}},\ \mathcal{E}_{\text{CLIP}}(I_{\text{gt}})\right),
$$
$$
\ell_{\text{total}} = \ell_{\text{text}} + \lambda\, \ell_{\text{embed}}, \quad \lambda = 2.
$$
A training design detail is that $\mathbf{F}_v$ consumes the 21st hidden layer while $\mathbf{F}_t$ consumes the final layer, which is used to reduce gradient interference [2505.19149].

In MVFT-JI, MEJO is instantiated by two InfoNCE objectives: target-text retrieval and text-to-image retrieval. The joint objective is
$$
L_{\text{MEJO}} = \lambda_1 \, L_{\text{target-text}} + \lambda_2 \, L_{\text{t2i}} + R,
$$
with $\lambda_1 = \lambda_2 = 1$ in experiments. The theoretical justification decomposes the retrieval target as
$$
P(I_t \mid I_r, t_m;\theta) = \sum_t P(I_t \mid t;\theta) \, P(t \mid I_r, t_m;\theta),
$$
and links both InfoNCE terms to mutual-information lower bounds [2505.19707].

In MSJoE, the optimization is purely reinforcement-based. The joint objective maximizes expected QA reward over the coupled MLLM-sampler policy:
$$
J(\theta_M,\theta_S)=\mathbb{E}_{\pi_{\theta_M,\theta_S}}[R(\hat{y},y)].
$$
The MLLM is updated with GRPO, the sampler with REINFORCE, and no supervised loss is added. The reward is additive, combining accuracy, format correctness, and an informativeness term that encourages peaked query-frame similarities [2602.22932].

In surgical triplet recognition, the multi-task backbone is trained with BCE losses for instrument, verb, target, and IVT prediction, while the triplet head additionally uses Coordinated Gradient Learning (CGL) to rebalance positive and negative BCE gradients for head and tail classes. The total form is
$$
\mathcal{L} = \lambda_I \mathcal{L}_I + \lambda_V \mathcal{L}_V + \lambda_T \mathcal{L}_T + \lambda_{IVT} \mathcal{L}_{IVT} + \lambda_{\text{orth}} \mathcal{L}_{\text{orth}} + \mathcal{R},
$$
with architectural disentanglement serving as the primary mechanism for inter-task conflict mitigation [2509.12893].

In DrivePI, joint optimization is explicitly multi-head:
$$
L_{\text{total}} = \lambda_1 L_{\text{llm}} + \lambda_2 L_{\text{occ}} + \lambda_3 L_{\text{flow}} + \lambda_4 L_{\text{action}},
$$
with default weights $\lambda_1 = \lambda_2 = \lambda_3 = \lambda_4 = 1$. The text head uses autoregressive next-token cross-entropy; occupancy follows FlashOcc-style multi-component losses; flow uses weighted $L_1$; planning uses $L_1$ trajectory loss [2512.12799].

A distinct systems-level formulation appears in "Modality Inflation: Energy Characterization and Optimization Opportunities for MLLM Inference," where MEJO is a joint optimization problem over encoder configuration, visual token budget, batching, and per-stage DVFS:
$$
\min_{\theta} \; E_{tot}(\theta) = \sum_{s \in \{\text{encode}, \text{prefill}, \text{decode}\}} \int P_s(t;\theta)\,dt
$$
subject to latency, accuracy, memory, and throughput constraints. Here jointness applies to inference-time control variables rather than to learned task heads [2512.22695].

## 4. Representative instantiations across domains

| Work | Domain | MEJO mechanism |
|---|---|---|
| MIND-Edit [2505.19149] | Text-guided image editing | Jointly fine-tunes one MLLM to generate $T_{\text{opt}}$ and $v_{\text{insight}}$, then injects both into Stable Diffusion v1.5 through decoupled cross-attention |
| MVFT-JI [2505.19707] | Zero-Shot Composed Image Retrieval | Uses an MLLM to synthesize $t_m$, $t^*$, and captions, then jointly optimizes target-text retrieval and text-to-image retrieval on a Q-Former VLM |
| MSJoE [2602.22932] | Long-form video QA | Jointly evolves MLLM query reasoning and a 1D U-Net key-frame sampler through GRPO and REINFORCE |
| MEJO for surgical triplet recognition [2509.12893] | Surgical scene understanding | Combines Shared-Specific-Disentangled learning, MLLM-powered probabilistic prompt pools, TSP task prompts, and Coordinated Gradient Learning |
| DrivePI [2512.12799] | Autonomous driving | Jointly trains a 4D spatial-aware MLLM for text QA, 3D occupancy, occupancy flow, and planning with shared hidden states |
| Modality Inflation guide [2512.22695] | MLLM serving and systems optimization | Jointly optimizes stage-wise energy via encoder configuration, token budget, batching, and DVFS under QoS constraints |

These instantiations show that the MLLM can engage at multiple levels: semantic clarification, latent representation shaping, query synthesis, retrieval supervision generation, dense spatial grounding, policy learning, or even serving-time control. Taken together, they suggest that MEJO is less a domain-specific method than a reusable optimization stance in which MLLM outputs are promoted from auxiliary hints to variables that materially determine the training or inference objective.

## 5. Empirical characteristics and reported gains

In image editing, MIND-Edit reports HumanEdit results of CLIP-I $0.9310$, LPIPS $0.1245$, PSNR $22.2714$, and SSIM $0.8517$. On ComplexMultistepImageEditing, it reports CLIP-I $0.6340$, LPIPS $0.6963$, PSNR $8.9245$, and SSIM $0.2069$. The ablation labeled “#6 (Ours, both joint)” exceeds the independently trained “#3 (both but independently trained)” on all listed metrics, supporting the claim that jointly training $T_{\text{opt}}$ and $v_{\text{insight}}$ improves alignment and fidelity [2505.19149].

In zero-shot composed image retrieval, MVFT-JI reports FashionIQ average $R@10 = 34.84$ and $R@50 = 56.29$, CIRR $R@1 = 39.30$, $R@5 = 69.49$, and $R@10 = 79.69$, and CIRCO $mAP@5 = 21.69$ and $mAP@50 = 26.37$. Ablations show that removing either inference component or either training task degrades performance on multiple benchmarks, with target-text retrieval identified as the most critical for compositional understanding and text-to-image retrieval described as stabilizing cross-modal grounding [2505.19707].

In long-form video QA, MSJoE reports an $8.0\%$ average accuracy gain upon the base MLLM with $64$ frames, a $6.7$ point gain with $32$ frames, and $1.1\%$ higher accuracy than the strongest baseline method. With $64$ frames, the reported scores include MLVU $75.1$, LongVideoBench $62.2$, VideoMME-Long $57.4$, VideoMME-Avg $66.2$, and LVBench $51.1$. The paper further reports that total pipeline time increases by approximately $30\%$ versus uniform sampling, although the final QA step is approximately $10\%$ faster than uniform sampling due to prefix cache hits in vLLM [2602.22932].

In surgical triplet recognition, MEJO-B reports $AP_{IVT} = 41.2\%$ on CholecT45, surpassing TERL-B $(38.9\%)$ by $2.3\%$, with component scores $AP_I = 93.9\%$, $AP_V = 72.9\%$, and $AP_T = 51.6\%$. On CholecT50, MEJO-B is reported to outperform the second-best method (CoT) by $+1.6\%$ $AP_{IVT}$. Progressive ablations on Fold 1 show MSTCN at $37.35\%$, TransFPN at $38.3\%$, $+CGL$ at $40.0\%$, $+GPI$ at $41.0\%$, and $+TSP$ at $42.3\%$, which isolates the contribution of each MEJO component [2509.12893].

In autonomous driving, DrivePI reports OccScore $49.3$, RayIoU $49.3$, and mAVE $0.509$ on OpenOcc; planning $L2$ average $0.49$ without ego status and $0.40$ with ego status on nuScenes; and a collision rate of $0.11\%$ with ego status. Relative comparisons include outperforming OpenDriveVLA-7B by $2.5\%$ mean accuracy on nuScenes-QA, reducing collision rate by $70\%$ over ORION from $0.37\%$ to $0.11\%$, exceeding FB-OCC by $10.3$ RayIoU on OpenOcc, improving occupancy-flow mAVE from $0.591$ to $0.509$, and reducing planning error from VAD’s $0.72$m to $0.49$m [2512.12799].

At the systems level, the modality-inflation study reports multimodal energy overheads ranging from $17\%$ to $94\%$ over text-only baselines on NVIDIA A100-80GB. Qwen2.5-VL is reported at $+94\%$ energy and $+179\%$ latency, while LLaVA-OneVision is reported at $+17\%$ energy despite $3{,}715$ total input tokens. Stage-level measurements identify encoder-dominated and prefill-dominated bottlenecks, and stage-wise DVFS is presented as an effective optimization that allows energy savings with only modest performance impact [2512.22695].

## 6. Limitations, misconceptions, and open problems

A common misconception is that MEJO implies full end-to-end retraining of all components. The literature does not support that interpretation. In MIND-Edit, Stable Diffusion v1.5 and IP-Adapter are pretrained and kept fixed while only the MLLM, including $\mathbf{F}_t$ and $\mathbf{F}_v$, is fine-tuned [2505.19149]. In MVFT-JI, the BLIP-2 ViT-L/14 image encoder is frozen, while the Q-Former and shared learnable query tokens are fine-tuned [2505.19707]. In DrivePI, the vision encoder is frozen during both stages, while the spatial projector, MLLM, and task heads are jointly optimized in the second stage [2512.12799].

A second misconception is that MEJO is synonymous with supervised multi-task learning. MSJoE uses reinforcement learning only, with GRPO for the MLLM and REINFORCE for the sampler, and explicitly states that no supervised loss is added. By contrast, surgical MEJO uses BCE-based multi-task supervision plus gradient gating, and MVFT-JI uses dual InfoNCE objectives. The term therefore spans heterogeneous optimization regimes rather than a unique loss family [2602.22932] [2509.12893].

Several limitations recur across domains. MIND-Edit reports that performance can be constrained by Stable Diffusion v1.5’s generative quality and IP-Adapter’s capacity, and that extremely fine-grained attribute control or highly compositional edits may remain challenging when instructions are still ambiguous or when the implicit visual embedding under-specifies local details [2505.19149]. MVFT-JI notes that subtle attribute changes may be underrepresented in synthetic $t^*$ and captions, that MLLM hallucinations can introduce label noise, and that one-to-one text-image training pairs can create distribution gaps in multi-target settings such as CIRCO [2505.19707]. MSJoE identifies missing temporal cues, overfitting to visually salient but irrelevant frames, and brittle query generation when preview frames are too sparse [2602.22932].

In knowledge-heavy domains, prompt quality and domain specificity become central failure modes. Surgical MEJO depends on GPT-4o-generated, surgeon-validated instrument attributes; noisy prompts may degrade performance, and transferring the prompt pool beyond cholecystectomy may require new knowledge extraction [2509.12893]. DrivePI uses fixed task weights and does not apply reinforcement learning or closed-loop finetuning, which the paper identifies as opportunities for improved planning in complex scenarios [2512.12799].

At the serving layer, modality inflation indicates that MLLM engagement can incur substantial energy and latency overhead because image inputs add encoder stages and expand prefill token sequences. The energy study argues that architecture-aware, input-aware, and stage-aware control is necessary, and motivates dynamic stage-aware controllers, multi-GPU or disaggregated pipelines, dynamic token routing, and online optimization of encoder configuration, visual token budget, batching, and per-stage DVFS [2512.22695].

Taken together, these works suggest that MEJO is best viewed as a general principle of coupling MLLM-derived reasoning with jointly optimized auxiliary structures under shared constraints. Its empirical successes are domain-specific, but its central proposition is stable across settings: cross-modal alignment improves when the MLLM is trained or controlled as part of the task mechanism itself rather than used only as an external prompt-conditioned interface.

Source: https://www.emergentmind.com/topics/mllm-engaged-joint-optimization-mejo