通过迭代协作提升医学图像超分辨率质量
Iterative Collaboration Network Guided By Reconstruction Prior for Medical Image Super-Resolution
- 设计双分支网络,让重建与超分任务反复交互
- 多阶段逐步放大,实现2倍逐级超分
- 新模块融合重建先验,提升特征完整性
高分辨率医学图像可提供更精细的诊断信息。传统方法先提取特征再上采样,特征可能不完整。近期多任务学习虽结合重建与超分,但两任务交互不足,导致深层特征不充分且相关性弱。为此,提出迭代协作网络(ICONet),通过渐进式引入重建先验,实现重建与超分任务的持续协作。网络包含重建分支、超分分支和SR-Rec融合模块。重建分支生成无伪影图像作为先验,供超分分支参考;设计新型残差空间-通道特征学习(RSCFL)模块,分别建模空间与通道维度特征关系,避免传统CNN或Transformer的局限。同时,自适应融合模块在各阶段动态整合重建先验与超分特征。整体采用多阶段结构,以2倍步长逐级上采样,并在每阶段施加双重监督,显著提升细节恢复能力。
原文摘要 · Abstract (English)
High-resolution medical images can provide more detailed information for better diagnosis. Conventional medical image super-resolution relies on a single task which first performs the extraction of the features and then upscaling based on the features. The features extracted may not be complete for super-resolution. Recent multi-task learning,including reconstruction and super-resolution, is a good solution to obtain additional relevant information. The interaction between the two tasks is often insufficient, which still leads to incomplete and less relevant deep features. To address above limitations, we propose an iterative collaboration network (ICONet) to improve communications between tasks by progressively incorporating reconstruction prior to the super-resolution learning procedure in an iterative collaboration way. It consists of a reconstruction branch, a super-resolution branch, and a SR-Rec fusion module. The reconstruction branch generates the artifact-free image as prior, which is followed by a super-resolution branch for prior knowledge-guided super-resolution. Unlike the widely-used convolutional neural networks for extracting local features and Transformers with quadratic computational complexity for modeling long-range dependencies, we develop a new residual spatial-channel feature learning (RSCFL) module of two branches to efficiently establish feature relationships in spatial and channel dimensions. Moreover, the designed SR-Rec fusion module fuses the reconstruction prior and super-resolution features with each other in an adaptive manner. Our ICONet is built with multi-stage models to iteratively upscale the low-resolution images using steps of 2x and simultaneously interact between two branches in multi-stage supervisions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。