arXiv Computer Vision· Martial Guidez, Stefan Duffner, Christophe Garcia·· 2 小时前AI 评分22
CoViG:用图粗化降低 Vision GNN 图像分类计算量
CoViG: Coarsened Vision GNN for Image Classification
AI 导读
CoViG 将图粗化与 Vision GNN(ViG)结合,在中间 ViG 块的 Grapher 组件中引入额外的 MinCUT pooling 步骤,通过简化中间图、减少特定块的节点数来降低推理计算成本。在 ImageNet 上对多种 ViG 架构的评测显示,CoViG 达到与原 ViG 模型相当的准确率,FLOPs 仅约为原来的 51%。
来源:arXiv Computer Vision · arxiv.org