3 个月前

ResViT:用于多模态医学图像合成的残差视觉Transformer

ResViT:用于多模态医学图像合成的残差视觉Transformer

摘要

基于卷积神经网络(CNN)主干的生成对抗模型近年来已在众多医学图像合成任务中达到最先进水平。然而,CNN 旨在通过紧凑的局部滤波器进行局部处理,其固有的归纳偏置限制了对上下文特征的学习能力。为此,本文提出一种新型生成对抗医学图像合成方法——ResViT,该方法结合了视觉变换器(Vision Transformer)对上下文信息的敏感性、卷积算子的精确性以及对抗学习带来的真实感。ResViT 的生成器采用一个中心瓶颈结构,其中包含一种新型的聚合残差变换器(Aggregated Residual Transformer, ART)模块,该模块协同整合了残差卷积与变换器模块的优势。ART 模块中的残差连接有助于提升所捕捉表征的多样性,而通道压缩模块则有效提炼与任务相关的关键信息。为降低计算负担,ART 模块之间引入了权重共享策略。此外,本文提出一种统一的实现框架,避免了针对不同源-目标模态配置重复构建独立的合成模型。我们在多对比度 MRI 中缺失序列的生成以及从 MRI 生成 CT 图像等任务中进行了全面验证。实验结果表明,ResViT 在定性观察与定量指标上均显著优于现有的基于 CNN 和变换器的方法。

代码仓库

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ResViT:用于多模态医学图像合成的残差视觉Transformer | 论文 | HyperAI超神经