arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2410.21969cs.CV

BenchX:面向胸部X光医学视觉语言预训练的统一基准框架

BenchX: A Unified Benchmark Framework for Medical Vision-Language Pretraining on Chest X-Rays

National University of Singapore(新加坡国立大学);Shandong University(山东大学);Nanyang Technological University(南洋理工大学);Microsoft Research Asia(微软亚洲研究院)

Yang Zhou, Tan Li Hui Faith, Yanyu Xu, Sicong Leng, Xinxing Xu, Yong Liu, Rick Siow Mong Goh

AI总结:

针对现有医学视觉语言预训练方法缺乏统一评估基准的问题,提出包含综合数据集、标准化基准套件和统一微调协议的BenchX框架,为9种先进方法建立基线并发现早期方法性能可超越近年方法。

摘要:

医学视觉语言预训练(MedVLP)在从配对及非配对的医学图像与报告中学习具备泛化性和可迁移性的视觉表征方面展现出潜力。MedVLP能够为下游任务提供有用的特征,并助力用更少的样本将特定任务模型适配到新场景中。然而,现有MedVLP方法在数据集、预处理和微调实现方面往往存在差异。由于缺乏统一、标准化且全面的基准,评估MedVLP方法对各类临床相关任务的泛化能力面临极大挑战。为填补这一空白,我们提出BenchX,这是一个统一的基准框架,可利用公开胸部X光数据集实现MedVLP方法之间的正面对比与系统分析。具体而言,BenchX由三部分组成:1)涵盖9个数据集、4项医学任务的综合数据集;2)用于标准化数据预处理、训练-测试划分和参数选择的基准套件;3)适配异构MedVLP方法的统一微调协议,分别用于分类、分割和报告生成任务中的一致性任务适配。借助BenchX,我们为9种最先进的MedVLP方法建立了基线,并且发现部分早期MedVLP方法的性能可得到提升,甚至超越更新的方法,这促使人们重新审视MedVLP领域先前工作的进展与结论。我们的代码可在https://github.com/yangzhou12/BenchX 获取。

英文摘要:

Medical Vision-Language Pretraining (MedVLP) shows promise in learning generalizable and transferable visual representations from paired and unpaired medical images and reports. MedVLP can provide useful features to downstream tasks and facilitate adapting task-specific models to new setups using fewer examples. However, existing MedVLP methods often differ in terms of datasets, preprocessing, and finetuning implementations. This pose great challenges in evaluating how well a MedVLP method generalizes to various clinically-relevant tasks due to the lack of unified, standardized, and comprehensive benchmark. To fill this gap, we propose BenchX, a unified benchmark framework that enables head-to-head comparison and systematical analysis between MedVLP methods using public chest X-ray datasets. Specifically, BenchX is composed of three components: 1) Comprehensive datasets covering nine datasets and four medical tasks; 2) Benchmark suites to standardize data preprocessing, train-test splits, and parameter selection; 3) Unified finetuning protocols that accommodate heterogeneous MedVLP methods for consistent task adaptation in classification, segmentation, and report generation, respectively. Utilizing BenchX, we establish baselines for nine state-of-the-art MedVLP methods and found that the performance of some early MedVLP methods can be enhanced to surpass more recent ones, prompting a revisiting of the developments and conclusions from prior works in MedVLP. Our code are available at https://github.com/yangzhou12/BenchX.

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

↑