CoMMIT: Coordinated Multimodal Instruction Tuning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Xintong, Wu, Junda, Yu, Tong, Wang, Yu, Chen, Xiang, Gu, Jiuxiang, Yao, Lina, McAuley, Julian, Shang, Jingbo
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909777209065472
author Li, Xintong
Wu, Junda
Yu, Tong
Wang, Yu
Chen, Xiang
Gu, Jiuxiang
Yao, Lina
McAuley, Julian
Shang, Jingbo
author_facet Li, Xintong
Wu, Junda
Yu, Tong
Wang, Yu
Chen, Xiang
Gu, Jiuxiang
Yao, Lina
McAuley, Julian
Shang, Jingbo
contents Instruction tuning in multimodal large language models (MLLMs) generally involves cooperative learning between a backbone LLM and a feature encoder of non-text input modalities. The major challenge is how to efficiently find the synergy between the two modules so that LLMs can adapt their reasoning abilities to downstream tasks while feature encoders can adjust to provide more task-specific information about its modality. In this paper, we analyze the MLLM instruction tuning from both theoretical and empirical perspectives, where we find the unbalanced learning between the feature encoder and the LLM can cause problems of oscillation and biased learning that lead to sub-optimal convergence. Inspired by our findings, we propose a Multimodal Balance Coefficient that enables quantitative measurement of the balance of learning. Based on this, we further design a dynamic learning scheduler that better coordinates the learning between the LLM and feature encoder, alleviating the problems of oscillation and biased learning. In addition, we introduce an auxiliary regularization on the gradient to promote updating with larger step sizes, which potentially allows for a more accurate estimation of the proposed MultiModal Balance Coefficient and further improves the training sufficiency. Our proposed approach is agnostic to the architecture of LLM and feature encoder, so it can be generically integrated with various MLLMs. We conduct experiments on multiple downstream tasks with various MLLMs, demonstrating that the proposed method is more effective than the baselines in MLLM instruction tuning.
format Preprint
id arxiv_https___arxiv_org_abs_2407_20454
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CoMMIT: Coordinated Multimodal Instruction Tuning
Li, Xintong
Wu, Junda
Yu, Tong
Wang, Yu
Chen, Xiang
Gu, Jiuxiang
Yao, Lina
McAuley, Julian
Shang, Jingbo
Machine Learning
Computation and Language
Instruction tuning in multimodal large language models (MLLMs) generally involves cooperative learning between a backbone LLM and a feature encoder of non-text input modalities. The major challenge is how to efficiently find the synergy between the two modules so that LLMs can adapt their reasoning abilities to downstream tasks while feature encoders can adjust to provide more task-specific information about its modality. In this paper, we analyze the MLLM instruction tuning from both theoretical and empirical perspectives, where we find the unbalanced learning between the feature encoder and the LLM can cause problems of oscillation and biased learning that lead to sub-optimal convergence. Inspired by our findings, we propose a Multimodal Balance Coefficient that enables quantitative measurement of the balance of learning. Based on this, we further design a dynamic learning scheduler that better coordinates the learning between the LLM and feature encoder, alleviating the problems of oscillation and biased learning. In addition, we introduce an auxiliary regularization on the gradient to promote updating with larger step sizes, which potentially allows for a more accurate estimation of the proposed MultiModal Balance Coefficient and further improves the training sufficiency. Our proposed approach is agnostic to the architecture of LLM and feature encoder, so it can be generically integrated with various MLLMs. We conduct experiments on multiple downstream tasks with various MLLMs, demonstrating that the proposed method is more effective than the baselines in MLLM instruction tuning.
title CoMMIT: Coordinated Multimodal Instruction Tuning
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2407.20454