TAGS: A Test-Time Generalist-Specialist Framework with Retrieval-Augmented Reasoning and Verification
Abstract
TAGS, a test-time framework combining generalist and specialist models with hierarchical retrieval and reliability scoring, enhances medical LLM reasoning without fine-tuning.
Recent advances such as Chain-of-Thought prompting have significantly improved large language models (LLMs) in zero-shot medical reasoning. However, prompting-based methods often remain shallow and unstable, while fine-tuned medical LLMs suffer from poor generalization under distribution shifts and limited adaptability to unseen clinical scenarios. To address these limitations, we present TAGS, a test-time framework that combines a broadly capable generalist with a domain-specific specialist to offer complementary perspectives without any model fine-tuning or parameter updates. To support this generalist-specialist reasoning process, we introduce two auxiliary modules: a hierarchical retrieval mechanism that provides multi-scale exemplars by selecting examples based on both semantic and rationale-level similarity, and a reliability scorer that evaluates reasoning consistency to guide final answer aggregation. TAGS achieves strong performance across nine MedQA benchmarks, boosting GPT-4o accuracy by 13.8%, DeepSeek-R1 by 16.8%, and improving a vanilla 7B model from 14.1% to 23.9%. These results surpass several fine-tuned medical LLMs, without any parameter updates. The code will be available at https://github.com/JianghaoWu/TAGS.
Community
TAGS introduces a parameter-efficient, test-time framework for robust medical question answering. By combining a generalist–specialist reasoning collaboration with hierarchical retrieval and uncertainty-aware verification, TAGS enables structured multi-agent inference without any model fine-tuning. The framework demonstrates strong generalization under distribution shift, achieving substantial performance gains across nine challenging MedQA benchmarks. Notably, TAGS improves the zero-shot accuracy of GPT-4o and DeepSeek-R1 by 13.8% and 16.8%, respectively—outperforming several domain-specific fine-tuned models.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Collab-RAG: Boosting Retrieval-Augmented Generation for Complex Question Answering via White-Box and Black-Box LLM Collaboration (2025)
- Question-Aware Knowledge Graph Prompting for Enhancing Large Language Models (2025)
- Talk Before You Retrieve: Agent-Led Discussions for Better RAG in Medical QA (2025)
- m1: Unleash the Potential of Test-Time Scaling for Medical Reasoning with Large Language Models (2025)
- MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphs (2025)
- MMKB-RAG: A Multi-Modal Knowledge-Based Retrieval-Augmented Generation Framework (2025)
- WixQA: A Multi-Dataset Benchmark for Enterprise Retrieval-Augmented Generation (2025)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment:
@librarian-bot
recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper