4 个月前

自动对焦层在语义分割中的应用

自动对焦层在语义分割中的应用

摘要

我们提出了一种用于语义分割的自动聚焦卷积层,旨在增强神经网络在多尺度处理方面的能力。自动聚焦层根据处理的上下文自适应地调整有效感受野的大小,以生成更强大的特征。这是通过并行使用具有不同膨胀率的多个卷积层,并结合一种注意力机制来实现的,该机制能够学习到由上下文驱动的最佳尺度。通过共享并行卷积层的权重,我们使网络具备了尺度不变性,同时仅适度增加了参数数量。所提出的自动聚焦层可以轻松集成到现有网络中,以提升模型的表示能力。我们在具有挑战性的盆腔CT多器官分割和脑肿瘤MRI分割任务上评估了我们的模型,并取得了非常有前景的性能。

代码仓库

luvgold/auotofoucus3D-Brats
tf
GitHub 中提及
perslev/Autofocus-Layer-TF
tf
GitHub 中提及
yaq007/Autofocus-Layer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
brain-tumor-segmentation-on-brats-2015AFN-6
Dice Score: 84%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
自动对焦层在语义分割中的应用 | 论文 | HyperAI超神经