arXiv:2605.01667cs.CV2026-05

将费舍尔向量编码引入混合CNN+ViT模型,提升小样本医学图像分类性能

Deep neural networks with Fisher vector encoding for medical image classification

论文配图:Deep neural networks with Fisher vector encoding for medical image classification
图 1 · 摘自论文原文
  • 用费舍尔向量对CNN+ViT特征进行无序编码,增强表示能力
  • 在MedMNIST(v2)上超越基准,在其他数据集达文献领先水平
  • 提出可控计算成本的GMM估计方法,适合大中小规模数据

无序编码方法在数据有限时能提升卷积神经网络(CNN)的图像分类表现。近期提出的混合CNN+视觉变换器(ViT)模型可缓解CNN的局部性偏差,优于纯CNN方法。然而,将此类混合模型与更复杂的特征表示结合仍鲜有研究。本文提出将费舍尔向量编码引入混合CNN+ViT架构,以适应小规模和大规模数据。该编码基于图像特征的高斯混合模型(GMM)估计,但在大数据集下计算成本过高。为此,我们设计方法控制GMM估计成本随数据增长的增速。通过在MedMNIST(v2)、Clean-CC-CCII和ISIC2018上的实验验证,本方法在所有MedMNIST(v2)数据集上超越基准,在Clean-CC-CCII和ISIC2018上达到文献可比结果。

原文摘要 · Abstract (English)

Orderless encoding methods have shown to improve Convolutional Neural Networks (CNNs) for image classification in the context of limited availability of data. Additionally, hybrid CNN + Vision Transformers (ViT) models have been recently proposed to address CNN locality bias issues. These models outperformed CNN-only approaches. Despite that, the integration of such hybrid models with more elaborated feature representation can be highly beneficial and remains large unexplored in the literature. In this context, we propose the introduction of an orderless encoding method, Fisher Vectors, to hybrid CNN + ViT architectures, aiming at achieving a model suitable for both small and large datasets. Such enconding method relies on estimating a Gaussian Mixture Model (GMM) on image features. In large datasets, computational costs of the GMM estimation is a limiting factor for the application of Fisher Vectors. Thus, we propose a method to limit the growth of GMM estimation costs as we increase the size of the dataset. We explore the feasibility of our method in the context of medical image classification by appling it to MedMNIST (v2), Clean-CC-CCII and ISIC2018. This collection of datasets contains a wide variety of data scales and modalities. We outperform benchmark results in all MedMNIST (v2) datasets and obtain literature-competitive results in Clean-CC-CCII and ISIC2018.

医学图像特征编码混合模型小样本学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。