arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.03379cs.CVeess.IV
收藏

MTVNet: Mapping using Transformers for Volumes——面向长程交互的超分辨率网络

MTVNet: Mapping using Transformers for Volumes -- Network for Super-Resolution with Long-Range Interactions

Technical University of Denmark(丹麦技术大学);Institution2(机构2)

August Leander Høeg, Sophia W. Bardenfleth, Hans Martin Kjer, Tim B. Dyrby, Vedrana Andersen Dahl, Anders Dahl

AI总结:

针对三维体数据超分辨率中Transformer自注意力内存受限、无法利用长程交互的问题,提出结合分层注意力模块与多尺度载体token的多尺度Transformer模型MTVNet,通过由粗到细建模扩大感受野,在五个三维数据集上取得优于现有模型的效果,大尺寸图像优势更显著。

摘要:

迄今为止,体数据超分辨率一直难以利用二维超分辨率领域中基于Transformer的模型所取得的最新进展。三维体数据中自注意力所需的内存限制了感受野,因此三维场景中无法像二维那样充分利用长程交互,Transformer的优势也未能得到发挥。我们提出了一种基于多尺度Transformer的模型,该模型结合了分层注意力模块与多尺度载体token,以克服这一问题。在该模型中,来自粗分辨率更大区域的信息会被依次传递到更细分辨率的区域,从而预测超分辨率图像。通过在每个分辨率层级使用Transformer层,我们的由粗到细建模方式限制了每个尺度下的token数量,并且能够实现比以往方法更大范围的注意力区域。我们在五个三维数据集上,将所提方法MTVNet与当前最优的体数据超分辨率模型进行了实验对比,证明了增大感受野的优势。这一优势在尺寸大于常用三维数据集图像的样本上表现得尤为明显。我们的代码可在https://github.com/AugustHoeg/MTVNet获取。

英文摘要:

Until now, it has been difficult for volumetric super-resolution to utilize the recent advances in transformer-based models seen in 2D super-resolution. The memory required for self-attention in 3D volumes limits the receptive field. Therefore, long-range interactions are not used in 3D to the extent done in 2D and the strength of transformers is not realized. We propose a multi-scale transformer-based model based on hierarchical attention blocks combined with carrier tokens at multiple scales to overcome this. Here information from larger regions at coarse resolution is sequentially carried on to finer-resolution regions to predict the super-resolved image. Using transformer layers at each resolution, our coarse-to-fine modeling limits the number of tokens at each scale and enables attention over larger regions than what has previously been possible. We experimentally compare our method, MTVNet, against state-of-the-art volumetric super-resolution models on five 3D datasets demonstrating the advantage of an increased receptive field. This advantage is especially pronounced for images that are larger than what is seen in popularly used 3D datasets. Our code is available at https://github.com/AugustHoeg/MTVNet

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

↑