MedAide: Leveraging Large Language Models for On-Premise Medical Assistance on Edge Devices

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Basit, Abdul, Hussain, Khizar, Hanif, Muhammad Abdullah, Shafique, Muhammad
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910350211809280
author Basit, Abdul
Hussain, Khizar
Hanif, Muhammad Abdullah
Shafique, Muhammad
author_facet Basit, Abdul
Hussain, Khizar
Hanif, Muhammad Abdullah
Shafique, Muhammad
contents Large language models (LLMs) are revolutionizing various domains with their remarkable natural language processing (NLP) abilities. However, deploying LLMs in resource-constrained edge computing and embedded systems presents significant challenges. Another challenge lies in delivering medical assistance in remote areas with limited healthcare facilities and infrastructure. To address this, we introduce MedAide, an on-premise healthcare chatbot. It leverages tiny-LLMs integrated with LangChain, providing efficient edge-based preliminary medical diagnostics and support. MedAide employs model optimizations for minimal memory footprint and latency on embedded edge devices without server infrastructure. The training process is optimized using low-rank adaptation (LoRA). Additionally, the model is trained on diverse medical datasets, employing reinforcement learning from human feedback (RLHF) to enhance its domain-specific capabilities. The system is implemented on various consumer GPUs and Nvidia Jetson development board. MedAide achieves 77\% accuracy in medical consultations and scores 56 in USMLE benchmark, enabling an energy-efficient healthcare assistance platform that alleviates privacy concerns due to edge-based deployment, thereby empowering the community.
format Preprint
id arxiv_https___arxiv_org_abs_2403_00830
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MedAide: Leveraging Large Language Models for On-Premise Medical Assistance on Edge Devices
Basit, Abdul
Hussain, Khizar
Hanif, Muhammad Abdullah
Shafique, Muhammad
Artificial Intelligence
Computation and Language
I.2.7
Large language models (LLMs) are revolutionizing various domains with their remarkable natural language processing (NLP) abilities. However, deploying LLMs in resource-constrained edge computing and embedded systems presents significant challenges. Another challenge lies in delivering medical assistance in remote areas with limited healthcare facilities and infrastructure. To address this, we introduce MedAide, an on-premise healthcare chatbot. It leverages tiny-LLMs integrated with LangChain, providing efficient edge-based preliminary medical diagnostics and support. MedAide employs model optimizations for minimal memory footprint and latency on embedded edge devices without server infrastructure. The training process is optimized using low-rank adaptation (LoRA). Additionally, the model is trained on diverse medical datasets, employing reinforcement learning from human feedback (RLHF) to enhance its domain-specific capabilities. The system is implemented on various consumer GPUs and Nvidia Jetson development board. MedAide achieves 77\% accuracy in medical consultations and scores 56 in USMLE benchmark, enabling an energy-efficient healthcare assistance platform that alleviates privacy concerns due to edge-based deployment, thereby empowering the community.
title MedAide: Leveraging Large Language Models for On-Premise Medical Assistance on Edge Devices
topic Artificial Intelligence
Computation and Language
I.2.7
url https://arxiv.org/abs/2403.00830