Generalist embedding models are better at short-context clinical semantic search than specialized embedding models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Excoffier, Jean-Baptiste, Roehr, Tom, Figueroa, Alexei, Papaioannou, Jens-Michalis, Bressem, Keno, Ortala, Matthieu
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911749289017344
author Excoffier, Jean-Baptiste
Roehr, Tom
Figueroa, Alexei
Papaioannou, Jens-Michalis
Bressem, Keno
Ortala, Matthieu
author_facet Excoffier, Jean-Baptiste
Roehr, Tom
Figueroa, Alexei
Papaioannou, Jens-Michalis
Bressem, Keno
Ortala, Matthieu
contents The increasing use of tools and solutions based on Large Language Models (LLMs) for various tasks in the medical domain has become a prominent trend. Their use in this highly critical and sensitive domain has thus raised important questions about their robustness, especially in response to variations in input, and the reliability of the generated outputs. This study addresses these questions by constructing a textual dataset based on the ICD-10-CM code descriptions, widely used in US hospitals and containing many clinical terms, and their easily reproducible rephrasing. We then benchmarked existing embedding models, either generalist or specialized in the clinical domain, in a semantic search task where the goal was to correctly match the rephrased text to the original description. Our results showed that generalist models performed better than clinical models, suggesting that existing clinical specialized models are more sensitive to small changes in input that confuse them. The highlighted problem of specialized models may be due to the fact that they have not been trained on sufficient data, and in particular on datasets that are not diverse enough to have a reliable global language understanding, which is still necessary for accurate handling of medical documents.
format Preprint
id arxiv_https___arxiv_org_abs_2401_01943
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generalist embedding models are better at short-context clinical semantic search than specialized embedding models
Excoffier, Jean-Baptiste
Roehr, Tom
Figueroa, Alexei
Papaioannou, Jens-Michalis
Bressem, Keno
Ortala, Matthieu
Computation and Language
Artificial Intelligence
The increasing use of tools and solutions based on Large Language Models (LLMs) for various tasks in the medical domain has become a prominent trend. Their use in this highly critical and sensitive domain has thus raised important questions about their robustness, especially in response to variations in input, and the reliability of the generated outputs. This study addresses these questions by constructing a textual dataset based on the ICD-10-CM code descriptions, widely used in US hospitals and containing many clinical terms, and their easily reproducible rephrasing. We then benchmarked existing embedding models, either generalist or specialized in the clinical domain, in a semantic search task where the goal was to correctly match the rephrased text to the original description. Our results showed that generalist models performed better than clinical models, suggesting that existing clinical specialized models are more sensitive to small changes in input that confuse them. The highlighted problem of specialized models may be due to the fact that they have not been trained on sufficient data, and in particular on datasets that are not diverse enough to have a reliable global language understanding, which is still necessary for accurate handling of medical documents.
title Generalist embedding models are better at short-context clinical semantic search than specialized embedding models
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2401.01943