Picking the Right Specialist: Attentive Neural Process-based Selection of Task-Specialized Models as Tools for Agentic Healthcare Systems
Organizations: Department of Engineering Science, University of Oxford, United Kingdom · Department of Computer Science, Khalifa University, Abu Dhabi, United Arab Emirates
Abstract
Task-specialized models form the backbone of agentic healthcare systems, enabling the agents to answer clinical queries across tasks such as disease diagnosis, localization, and report generation. Yet, for a given task, a single "best" model rarely exists. In practice, each task is better served by multiple competing specialist models where different models excel on different data samples. As a result, for any given query, agents must reliably select the right specialist model from a heterogeneous pool of tool candidates. To this end, we introduce ToolSelect, which adaptively learns model selection for tools by minimizing a population risk over sampled specialist tool candidates using a consistent surrogate of the task-conditional selection loss. Concretely, we propose an Attentive Neural Process-based selector conditioned on the query and per-model behavioral summaries to choose among the specialist models. Motivated by the absence of any established testbed, we, for the first time, introduce an agentic Chest X-ray environment equipped with a diverse suite of task-specialized models (17 disease detection, 19 report generation, 6 visual grounding, and 13 VQA) and develop ToolSelectBench, a benchmark of 1448 queries. Our results demonstrate that ToolSelect consistently outperforms 10 SOTA methods across four different task families.
Figures & tables
| Method | Open-I | VinDr | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Acc | F1 | P | R | AUC | Acc | F1 | P | R | AUC | |
| Tool Candidate Models | ||||||||||
| D-PC | 60.22 | 13.35 | 9.72 | 42.39 | 38.74 | 49.32 | 22.12 | 20.06 | 46.86 | 41.16 |
| D-NIH | 53.88 | 12.54 | 9.85 | 42.93 | 36.91 | 35.50 | 23.90 | 17.86 | 51.70 | 37.21 |
| D-CX | 42.02 | 10.59 | 9.12 | 29.71 | 36.24 | 40.02 | 23.99 | 18.19 | 49.06 | 39.60 |
| D-M-Ch | 20.69 | 11.95 | 7.88 | 48.44 | 33.50 | 26.25 | 24.72 | 16.84 | 62.31 | 36.62 |
| Model | SemB | F1-RG | R-L | MR | Rate |
|---|---|---|---|---|---|
| Tool Candidate Models | |||||
| CheXAgent-8B | 41.91 | 16.33 | 21.97 | 25.00 | 55.06 |
| CheXagent-2-3B | 34.44 | 11.47 | 15.26 | 23.64 | 54.24 |
| CheXpert Plus (C) | 42.86 | 15.71 | 19.48 | 30.55 | 56.59 |
| CheXpert Plus (C+M) | 41.81 | 16.10 | 20.35 | 30.88 | 57.05 |
| CheXpert Plus (M) | 43.32 | 10.71 | 19.91 | 28.99 | 54.31 |
| Model | Aortic enl. | Atel. | Calc. | Cardio. | Cons. | ILD | Infil. | Lung Opac. | Nodule/Mass | Other les. | Pulm. fib. | [email protected] | [email protected] | Mean IoU |
| Tool Candidate Models | ||||||||||||||
| AG-KD | 66.25 | 33.41 | 17.33 | 70.12 | 39.45 | 22.90 | 31.48 | 28.71 | 7.58 | 12.70 | 22.55 | 45.84 | 32.96 | 40.66 |
| MDETR-AGPT | 25.84 | 6.80 | 1.06 | 40.52 | 6.19 | 14.11 | 5.07 | 5.18 | 2.91 | 6.29 | 3.10 | 18.96 | 1.76 | 16.52 |
| TransVG-AGPT | 14.37 | 5.98 | 1.74 | 41.52 | 13.12 | 13.24 | 11.57 | 5.92 | 2.23 | 4.08 | 1.73 | 16.83 | 3.22 | 14.06 |
| ChEX | 53.00 | 49.44 | 18.86 | 71.50 | 54.31 | 14.21 | 35.99 | 32.65 | 13.71 | 27.72 | 20.22 | 59.69 | 39.42 | 40.25 |
| MAIRA-2 | 16.34 | 13.19 | 6.11 | 38.32 | 14.00 | 17.18 | 13.18 | 11.32 | 1.81 | 3.04 | 5.81 | 24.18 | 4.60 | 16.67 |
| Tool Candidates | Q30B | LOV | Q4B | Q8B | L7B | L13B | Q2B | L3.2V | BCXR | BMIM | CXA | RDlg | NV |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| VQA Acc (%) | 60.63 | 58.91 | 57.76 | 56.32 | 54.60 | 53.74 | 53.16 | 33.05 | 25.00 | 23.60 | 47.41 | 38.51 | 63.22 |
| Selection Methods | Random | KNN | SVM | MLP | MF | Elo | Graph | RouterDC | AutoMix | Hybrid LLM | Causal LLM | ToolSelect | Oracle |
| VQA Acc (%) | 46.49 | 41.52 | 47.37 | 47.08 | 36.55 | 63.74 | 47.37 | 45.61 | 24.27 | 42.69 | 47.37 | 72.01 | 96.49 |
Tool Selection
Tool selection studies how AI agents decide when to use external tools and which APIs, models, or physical tools best fit a task.
See all 25 papers