论文
使用整流流缩放胸部放射成像生成基础模型 Transformer
Scaling Generative Foundation Models for Chest Radiography with Rectified Flow Transformers
摘要
我们引入了第一个用于胸片合成的生成基础模型,该模型是在十亿参数规模上从头开始训练的。现有的放射摄影人工智能模型通常在患者亚群、机构和采集环境中普遍性较差,导致现实世界的临床效用有限。胸片的受控、高保真合成是实现临床数据集多样化和评估诊断模型稳健性的一条有前途的途径。因此,我们提出了迄今为止最大的胸部放射线照片专家生成基础模型,具有超过 1.3B 个参数,在包含 1.20 万张放射线照片和临床专家指导元数据的精选异构数据集上针对 1.6T 标记进行了训练。我们的模型支持跨多个人口亚组、采集视图和十几种病理的可控放射线照片生成和编辑。此外,我们显着提高了放射线照片合成保真度的最先进水平,生成的图像与临床专家无法区分的真实放射线照片。