跳到正文
原文
arXiv cs.CV· Martial Guidez, Stefan Duffner, Christophe Garcia·· 3 小时前AI 评分35

CoViG:面向图像分类的粗化 Vision GNN

CoViG: Coarsened Vision GNN for Image Classification

AI 导读

CoViG 将图粗化与 Vision GNN(ViG)结合,在中间 ViG 块的 Grapher 组件中引入 MinCUT pooling,通过简化中间图、减少节点数来降低推理阶段的计算开销。在 ImageNet 上用多个 ViG 架构评测,其精度与原 ViG 模型相当,FLOPs 仅约 51%。

来源:arXiv cs.CV · arxiv.org