arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.01092cs.CV

通过统一且定制化的指令微调学习用于胸片解读的多任务 Transformer

Learning A Multi-Task Transformer Via Unified And Customized Instruction Tuning For Chest Radiograph Interpretation

Centre for Perceptual and Interactive Intelligence(感知与交互智能中心);the Chinese University of Hong Kong(香港中文大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);Sensetime Research(商汤科技研究院);Department of Electronic Engineering(电子工程系)

Lijian Xu, Ziyu Ni, Xinglong Liu, Xiaosong Wang, Hongsheng Li, Shaoting Zhang

AI总结:

本文提出结合定制化指令微调的统一多任务 Transformer,利用1340万图文指令对训练,并在胸片多任务基准上验证其性能与报告可解释性。

摘要:

在过去十年中,多模态深度学习模型的出现对临床应用产生了重大影响。然而,大多数模型仅限于单任务处理,而没有考虑疾病诊断实际上是一个多任务过程。在这里,我们展示了一个专为多模态临床任务设计的统一 Transformer 模型,该模型结合了定制化指令微调。我们首先构建了一个多任务训练数据集,包含 1340 万个指令与真值对(约 100 万张放射影像),用于定制化微调,涵盖图像级和像素级任务。因此,我们能够在单一训练框架中统一各类视觉密集型任务,并采用同质的模型输入与输出,以提升一次阅片过程中的临床可解释性。最后,我们证明,在直接推理和微调两种设置下,相较于先前方法,本模型在跨多任务的各类胸部 X 线基准上均表现出整体更优的性能。三名放射科医生进一步将生成报告与记录报告进行评估,结果也表明我们的多任务模型具有更强的可解释性。

英文摘要:

The emergence of multi-modal deep learning models has made significant impacts on clinical applications in the last decade. However, the majority of models are limited to single-tasking, without considering disease diagnosis is indeed a multi-task procedure. Here, we demonstrate a unified transformer model specifically designed for multi-modal clinical tasks by incorporating customized instruction tuning. We first compose a multi-task training dataset comprising 13.4 million instruction and ground-truth pairs (with approximately one million radiographs) for the customized tuning, involving both image- and pixel-level tasks. Thus, we can unify the various vision-intensive tasks in a single training framework with homogeneous model inputs and outputs to increase clinical interpretability in one reading. Finally, we demonstrate the overall superior performance of our model compared to prior arts on various chest X-ray benchmarks across multi-tasks in both direct inference and finetuning settings. Three radiologists further evaluate the generated reports against the recorded ones, which also exhibit the enhanced explainability of our multi-task model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑