3 个月前

正则化Frank-Wolfe用于密集CRF:推广均场方法及其他

正则化Frank-Wolfe用于密集CRF:推广均场方法及其他

摘要

我们提出了一种正则化Frank-Wolfe算法,该算法是一种通用且高效的密集条件随机场(CRF)推理与学习方法。该算法通过使用原始Frank-Wolfe方法结合近似更新,对CRF推理问题的非凸连续松弛形式进行优化,而这些近似更新等价于最小化一个正则化的能量函数。所提出的算法是对现有方法(如平均场方法或凹凸过程,CCCP)的推广。这一视角不仅为这些算法提供了统一的理论分析框架,还为探索多种潜在性能更优的变体提供了简便途径。我们在标准语义分割数据集上的实验结果验证了这一点,其中我们提出的正则化Frank-Wolfe的多个实例在作为独立组件使用或作为神经网络中可端到端训练的层时,均优于平均场推理方法。此外,我们还表明,结合我们新提出的算法,密集CRF相较于强大的CNN基线模型,能够实现显著的性能提升。

代码仓库

netw0rkf10w/crf
官方
pytorch

基准测试

基准方法指标
semantic-segmentation-on-cityscapesEuclidean Frank-Wolfe CRFs (backbone: DeepLabv3+)(coarse)
Mean IoU (class): 83.6%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
正则化Frank-Wolfe用于密集CRF:推广均场方法及其他 | 论文 | HyperAI超神经