Jun, 2022

轻量化思路:自适应剪枝在医学图像分割中的 Transformer 优化

TL;DR本文提出应用自适应修剪技术来改进医学图像分割的 transformer 模型。我们的自适应修剪 transformer 的方法名为 APFormer。该模型使用自监督的自我注意机制和高斯先验相对位置嵌入来改进依赖信息建立。我们的实验表明,APFormer 在可用参数更少、GFLOPs 更低的情况下,是目前最好的模型之一,并证明了自适应修剪技术可以作为插入式模块,用于改进其他混合 / 变压器方法的性能。