提出任务原型框架,统一处理多种医学病灶分割任务。
TP-Seg: Task-Prototype Framework for Unified Medical Lesion Segmentation
- 用双路专家结构分离共享与特定任务特征,避免信息混淆。
- 引入可学习的任务原型,精准建模不同任务的前景背景语义。
- 在8个不同任务上超越专用和通用模型,适合临床部署。
构建一个参数统一、能高效处理多种医学病灶分割任务的AI模型已成为辅助诊断的关键目标。现有统一分割方法通常在异构任务与模态间共享编码器,易导致特征纠缠、梯度干扰和病灶区分能力下降。本文提出TP-Seg任务原型框架:一方面,任务条件适配器通过双路专家结构,自适应地平衡共享与任务特异性表示,实现跨多模态、多病灶类型的特征提取;另一方面,原型引导的任务解码器引入可学习的任务原型作为语义锚点,利用交叉注意力机制精细建模任务特异的前景与背景语义。无需复杂组件,TP-Seg在涵盖多个成像模态的8项医学病灶分割任务中持续优于专用、通用及统一分割方法,展现出强泛化性、可扩展性与临床适用性。
原文摘要 · Abstract (English)
Building a unified model with a single set of parameters to efficiently handle diverse types of medical lesion segmentation has become a crucial objective for AI-assisted diagnosis. Existing unified segmentation approaches typically rely on shared encoders across heterogeneous tasks and modalities, which often leads to feature entanglement, gradient interference, and suboptimal lesion discrimination. In this work, we propose TP-Seg, a task-prototype framework for unified medical lesion segmentation. On one hand, the task-conditioned adapter effectively balances shared and task-specific representations through a dual-path expert structure, enabling adaptive feature extraction across diverse medical imaging modalities and lesion types. On the other hand, the prototype-guided task decoder introduces learnable task prototypes as semantic anchors and employs a cross-attention mechanism to achieve fine-grained modeling of task-specific foreground and background semantics. Without bells and whistles, TP-Seg consistently outperforms specialized, general and unified segmentation methods across 8 different medical lesion segmentation tasks covering multiple imaging modalities, demonstrating strong generalization, scalability and clinical applicability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。