Skip to content
CMME-LabPublic

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Latest commit

 

History

8 Commits

Folders and files

Repository files navigation

SMILE: A Structured Multi-Agent Clinical Reasoning for Liver Ultrasound Understanding

Haris Ghafoor†, Minjee Seo†, Seonaeng Cho, Keungmo Yang, Jaejun Lee, Kyungho Yoon

(† equal contribution)

Summary

SMILE interprets a liver ultrasound study with three complementary specialist agents — an image interpretation agent for morphological descriptions, a clinical-stage reasoning agent, and a VQA agent for targeted queries — and consolidates their outputs through a judge VLM using a consensus filtering prompt. On a public and an internal clinical dataset it improves ROUGE-1 by up to 19.0% and clinical F1 by 18.8% over state-of-the-art VLMs.

Method overview

Installation

1. Environment

pip install -e ".[all]"

2. Datasets

Download the datasets from link and place them under data/.

We also release our internal Dataset-2 with this work. 01_dataset_overview.ipynb walks through its schema, labels and images, and shows how to build a patient-grouped split for your own training runs. Statistics: 1,417 liver ultrasound images obtained from 286 patients (mean age: $55.18 \pm 16.04$ years; 105 males (36.7%) and 181 females (63.3%)), along with 286 corresponding clinical descriptions.

3. Models

Download the three LoRA adapters from link and place them under weights/:

weights/
├── agent_a/
├── agent_b/
└── agent_c/

4. Training

python scripts/train_agent.py --config configs/agents/agent_a.yaml \
    --records data/d1/train.json --output-dir weights/agent_a

5. Evaluation

python scripts/run_council.py --records data/d1/test.json --dataset d1 \
    --out runs/d1_council.json

python scripts/evaluate.py --results runs/d1_council.json --dataset d1 \
    --pred-col judge_final_response

Cite as

Disclaimer

If you use this codebase, the models or the dataset, please cite the paper above.

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages