Multimodal Deep Learning for Low-Resource Settings: A Vector Embedding Alignment Approach for Healthcare Applications

David Restrepo,Chenwei Wu,Sebastián Andrés Cajas,Luis Filipe Nakayama,Leo Anthony Celi,Diego M López
2024-06-02
Abstract:Large-scale multi-modal deep learning models have revolutionized domains such as healthcare, highlighting the importance of computational power. However, in resource-constrained regions like Low and Middle-Income Countries (LMICs), limited access to GPUs and data poses significant challenges, often leaving CPUs as the sole resource. To address this, we advocate for leveraging vector embeddings to enable flexible and efficient computational methodologies, democratizing multimodal deep learning across diverse contexts. Our paper investigates the efficiency and effectiveness of using vector embeddings from single-modal foundation models and multi-modal Vision-Language Models (VLMs) for multimodal deep learning in low-resource environments, particularly in healthcare. Additionally, we propose a simple yet effective inference-time method to enhance performance by aligning image-text embeddings. Comparing these approaches with traditional methods, we assess their impact on computational efficiency and model performance using metrics like accuracy, F1-score, inference time, training time, and memory usage across three medical modalities: BRSET (ophthalmology), HAM10000 (dermatology), and SatelliteBench (public health). Our findings show that embeddings reduce computational demands without compromising model performance. Furthermore, our alignment method improves performance in medical tasks. This research promotes sustainable AI practices by optimizing resources in constrained environments, highlighting the potential of embedding-based approaches for efficient multimodal learning. Vector embeddings democratize multimodal deep learning in LMICs, particularly in healthcare, enhancing AI adaptability in varied use cases.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The main aim of this paper is to address the issue of efficient multimodal deep learning in resource-constrained environments, particularly in low- and middle-income countries (LMICs). Specifically, the research focuses on the following aspects: 1. **Problem Background**: With the development of large-scale multimodal deep learning models and datasets, they have had a revolutionary impact on various fields, including healthcare. However, in resource-constrained environments such as LMICs, limited access to GPUs and data means that many applications have to rely on CPUs, posing a challenge for the application of multimodal deep learning. 2. **Research Objective**: The goal of the paper is to optimize multimodal deep learning methods through the use of vector embeddings, making them more efficient in resource-constrained environments. Specifically, the researchers propose a vector embedding alignment method aimed at improving the computational efficiency and effectiveness of multimodal deep learning models in healthcare applications. 3. **Technical Approaches**: The research employs two different embedding extraction methods: - Using unimodal base models (such as DINO V2 for images and LLAMA 2 for text) to extract image and text embeddings separately. - Using multimodal vision-language models (VLMs), such as CLIP, to extract image and text embeddings. 4. **Method Comparison**: The study compares these embedding-based methods with traditional multimodal deep learning methods (directly using raw data) in terms of computational efficiency and model performance. The experiments are evaluated using metrics such as accuracy, F1 score, inference time, training time, and memory usage. 5. **Key Findings**: - Embedding methods can significantly reduce computational requirements without sacrificing model performance. - The proposed embedding alignment method can further enhance model performance in healthcare tasks. - The research highlights the importance of embedding-based methods for optimizing multimodal learning in resource-constrained environments and contributes to the sustainable development of AI technology. In summary, the core of this paper lies in exploring how vector embedding techniques can overcome the challenges faced by multimodal deep learning in resource-constrained environments and proposing an effective solution to improve the computational efficiency and performance of models.