Multimodal Lego: Model Merging and Fine-Tuning Across Topologies and Modalities in Biomedicine

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hemker, Konstantin, Simidjievski, Nikola, Jamnik, Mateja
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908321316864000
author Hemker, Konstantin
Simidjievski, Nikola
Jamnik, Mateja
author_facet Hemker, Konstantin
Simidjievski, Nikola
Jamnik, Mateja
contents Learning holistic computational representations in physical, chemical or biological systems requires the ability to process information from different distributions and modalities within the same model. Thus, the demand for multimodal machine learning models has sharply risen for modalities that go beyond vision and language, such as sequences, graphs, time series, or tabular data. While there are many available multimodal fusion and alignment approaches, most of them require end-to-end training, scale quadratically with the number of modalities, cannot handle cases of high modality imbalance in the training set, or are highly topology-specific, making them too restrictive for many biomedical learning tasks. This paper presents Multimodal Lego (MM-Lego), a general-purpose fusion framework to turn any set of encoders into a competitive multimodal model with no or minimal fine-tuning. We achieve this by introducing a wrapper for any unimodal encoder that enforces shape consistency between modality representations. It harmonises these representations by learning features in the frequency domain to enable model merging with little signal interference. We show that MM-Lego 1) can be used as a model merging method which achieves competitive performance with end-to-end fusion models without any fine-tuning, 2) can operate on any unimodal encoder, and 3) is a model fusion method that, with minimal fine-tuning, surpasses all benchmarks in five out of seven datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2405_19950
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Multimodal Lego: Model Merging and Fine-Tuning Across Topologies and Modalities in Biomedicine
Hemker, Konstantin
Simidjievski, Nikola
Jamnik, Mateja
Machine Learning
Artificial Intelligence
Learning holistic computational representations in physical, chemical or biological systems requires the ability to process information from different distributions and modalities within the same model. Thus, the demand for multimodal machine learning models has sharply risen for modalities that go beyond vision and language, such as sequences, graphs, time series, or tabular data. While there are many available multimodal fusion and alignment approaches, most of them require end-to-end training, scale quadratically with the number of modalities, cannot handle cases of high modality imbalance in the training set, or are highly topology-specific, making them too restrictive for many biomedical learning tasks. This paper presents Multimodal Lego (MM-Lego), a general-purpose fusion framework to turn any set of encoders into a competitive multimodal model with no or minimal fine-tuning. We achieve this by introducing a wrapper for any unimodal encoder that enforces shape consistency between modality representations. It harmonises these representations by learning features in the frequency domain to enable model merging with little signal interference. We show that MM-Lego 1) can be used as a model merging method which achieves competitive performance with end-to-end fusion models without any fine-tuning, 2) can operate on any unimodal encoder, and 3) is a model fusion method that, with minimal fine-tuning, surpasses all benchmarks in five out of seven datasets.
title Multimodal Lego: Model Merging and Fine-Tuning Across Topologies and Modalities in Biomedicine
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2405.19950