Saved in:
Bibliographic Details
Main Authors: Imrie, Fergus, Denner, Stefan, Brunschwig, Lucas S., Maier-Hein, Klaus, van der Schaar, Mihaela
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.18227
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916730256752640
author Imrie, Fergus
Denner, Stefan
Brunschwig, Lucas S.
Maier-Hein, Klaus
van der Schaar, Mihaela
author_facet Imrie, Fergus
Denner, Stefan
Brunschwig, Lucas S.
Maier-Hein, Klaus
van der Schaar, Mihaela
contents The application of machine learning in medicine and healthcare has led to the creation of numerous diagnostic and prognostic models. However, despite their success, current approaches generally issue predictions using data from a single modality. This stands in stark contrast with clinician decision-making which employs diverse information from multiple sources. While several multimodal machine learning approaches exist, significant challenges in developing multimodal systems remain that are hindering clinical adoption. In this paper, we introduce a multimodal framework, AutoPrognosis-M, that enables the integration of structured clinical (tabular) data and medical imaging using automated machine learning. AutoPrognosis-M incorporates 17 imaging models, including convolutional neural networks and vision transformers, and three distinct multimodal fusion strategies. In an illustrative application using a multimodal skin lesion dataset, we highlight the importance of multimodal machine learning and the power of combining multiple fusion strategies using ensemble learning. We have open-sourced our framework as a tool for the community and hope it will accelerate the uptake of multimodal machine learning in healthcare and spur further innovation.
format Preprint
id arxiv_https___arxiv_org_abs_2407_18227
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Automated Ensemble Multimodal Machine Learning for Healthcare
Imrie, Fergus
Denner, Stefan
Brunschwig, Lucas S.
Maier-Hein, Klaus
van der Schaar, Mihaela
Machine Learning
The application of machine learning in medicine and healthcare has led to the creation of numerous diagnostic and prognostic models. However, despite their success, current approaches generally issue predictions using data from a single modality. This stands in stark contrast with clinician decision-making which employs diverse information from multiple sources. While several multimodal machine learning approaches exist, significant challenges in developing multimodal systems remain that are hindering clinical adoption. In this paper, we introduce a multimodal framework, AutoPrognosis-M, that enables the integration of structured clinical (tabular) data and medical imaging using automated machine learning. AutoPrognosis-M incorporates 17 imaging models, including convolutional neural networks and vision transformers, and three distinct multimodal fusion strategies. In an illustrative application using a multimodal skin lesion dataset, we highlight the importance of multimodal machine learning and the power of combining multiple fusion strategies using ensemble learning. We have open-sourced our framework as a tool for the community and hope it will accelerate the uptake of multimodal machine learning in healthcare and spur further innovation.
title Automated Ensemble Multimodal Machine Learning for Healthcare
topic Machine Learning
url https://arxiv.org/abs/2407.18227