SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2404.15127 · 2024

GSCo: Towards Generalizable AI in Medicine via Generalist-Specialist Collaboration

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 8 functions out of this paper's own repositories and ran 7 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
sunanhe/meddr canonical 7 of 8
FunctionStatusWhere it lives
build_transform Ran sunanhe/meddr/src/dataset/transforms.py
code served (permissive licence) · get_code("3ddf5cec5e279143")
build_transform Ran sunanhe/meddr/src/train/dataset.py
code served (permissive licence) · get_code("722750f3bafc8978")
collate_fn Ran sunanhe/meddr/inference_meddr.py
code served (permissive licence) · get_code("f239b832e3dff6ea")
expand2square Ran sunanhe/meddr/src/dataset/transforms.py
code served (permissive licence) · get_code("592b3c1a88f93d7c")
get_num_layer_for_vit_and_qllama Ran sunanhe/meddr/src/train/trainer_monkey_patch.py
code served (permissive licence) · get_code("001474ccd1834ca7")
param_classification Ran sunanhe/meddr/src/train/trainer_monkey_patch.py
code served (permissive licence) · get_code("67b54ec649e579c0")
pil_loader Ran sunanhe/meddr/src/train/dataset.py
code served (permissive licence) · get_code("700e7b33548e7070")
get_conv_template Not yet run sunanhe/meddr/src/utils/conversation.py
code served (permissive licence) · get_code("bd0b36d66d5aba83")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Generalist foundation models (GFMs) are renowned for their exceptional capability and flexibility in effectively generalizing across diverse tasks and modalities. In the field of medicine, while GFMs exhibit superior generalizability based on their extensive intrinsic knowledge as well as proficiency in instruction following and in-context learning, specialist models excel in precision due to their domain knowledge. In this work, for the first time, we explore the synergy between the GFM and specialist models, to enable precise medical image analysis on a broader scope. Specifically, we propose a cooperative framework, Generalist-Specialist Collaboration (GSCo), which consists of two stages, namely the construction of GFM and specialists, and collaborative inference on downstream tasks. In the construction stage, we develop MedDr, the largest open-source GFM tailored for medicine, showcasing exceptional instruction-following and in-context learning capabilities. Meanwhile, a series of lightweight specialists are crafted for downstream tasks with low computational cost. In the collaborative inference stage, we introduce two cooperative mechanisms, Mixture-of-Expert Diagnosis and Retrieval-Augmented Diagnosis, to harvest the generalist's in-context learning abilities alongside the specialists' domain expertise. For a comprehensive evaluation, we curate a large-scale benchmark featuring 28 datasets and about 250,000 images. Extensive results demonstrate that MedDr consistently outperforms state-of-the-art GFMs on downstream datasets. Furthermore, GSCo exceeds both GFMs and specialists across all out-of-domain disease diagnosis datasets. These findings indicate a significant paradigm shift in the application of GFMs, transitioning from separate models for specific tasks to a collaborative approach between GFMs and specialists, thereby advancing the frontiers of generalizable AI in medicine.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2404.15127")
get_code_for_paper("2404.15127")
have("2404.15127")

Connect an agent — have() is free.