3 个月前

RegNet:用于图像分类的自调节网络

RegNet:用于图像分类的自调节网络

摘要

ResNet及其变体在各类计算机视觉任务中取得了显著成果。尽管其通过残差连接机制有效实现了梯度在模块间的流动,但简单的加性捷径连接方式限制了模型对新潜在互补特征的重新探索能力。为解决这一问题,本文提出引入一个调节器模块(regulator module)作为记忆机制,用于提取互补特征,并将其进一步输入至ResNet中。具体而言,该调节器模块由卷积循环神经网络(如卷积LSTM或卷积GRU)构成,这类网络被证实具备提取时空信息的优异能力。我们所提出的新型受控网络被命名为RegNet。该调节器模块具有良好的可实现性,可轻松嵌入任意ResNet架构中。此外,我们还将该模块应用于改进挤压-激励ResNet(Squeeze-and-Excitation ResNet),以验证所提方法的通用性。在三个图像分类数据集上的实验结果表明,与标准ResNet、SE-ResNet以及其他先进架构相比,所提出的RegNet展现出更具竞争力的性能表现。

基准测试

基准方法指标
image-classification-on-gashissdbRegNetY-3.2GF
Accuracy: 97.48
F1-Score: 98.70
Precision: 99.97
medical-image-classification-on-nct-crc-heRegNetY-3.2GF
Accuracy (%): 95.42
F1-Score: 97.39
Precision: 99.97
Specificity: 99.43

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
RegNet:用于图像分类的自调节网络 | 论文 | HyperAI超神经