arXiv:2607.17329cs.CV2026-07

通过分层聚类压缩医学图像分割模型,兼顾精度与效率。

MIS-HCC: Hierarchical Channel Clustering for Efficient Medical Image Segmentation

论文配图:MIS-HCC: Hierarchical Channel Clustering for Efficient Medical Image Segmentation
图 1 · 摘自论文原文
  • 基于通道相似性分层聚类,用Wasserstein距离衡量通道差异。
  • 在三个数据集上压缩后仍保持高精度,优于现有方法。
  • 适合资源受限的医疗设备部署,提升模型实用性。

医学图像分割模型需兼顾高精度与轻量化设计,以适应实际医疗应用。由于计算和参数开销大,这类模型在资源受限的医疗平台上的部署仍面临挑战。现有模型剪枝方法大多忽视深层神经网络内部结构间的内在联系与相似性,导致压缩后模型难以保留预训练网络的基本特征。为此,本文提出一种面向医学图像分割模型的分层聚类压缩方法(MIS-HCC)。该方法利用层次聚类对通道进行划分,并通过预训练网络各层内的通道间Wasserstein距离构建相似性矩阵,指导聚类过程。同一簇内的通道被融合以生成压缩模型。在三个医学图像数据集上的实验结果表明,MIS-HCC在精度和压缩效率方面均优于当前最优方法,为资源受限医疗平台部署医学图像分割模型提供了有效解决方案。

原文摘要 · Abstract (English)

Medical image segmentation models require both high accuracy and lightweight design to accommodate real-world medical applications. The deployment of these models on resource-limited medical platforms remains a significant challenge due to their high computational and parameter requirements. Existing pruning methods for model compression mostly overlook the intrinsic connections and similarity between the internal structures of complex deep neural networks. As a result, compressed models may not effectively retain the basic features of the pretrained network. To solve this problem, we propose a hierarchical clustering compression method for medical image segmentation models (MIS-HCC). This approach employs hierarchical clustering to partition channels and fuse their parameters efficiently. Specifically, it leverages the Wasserstein distance to represent similarity of channels within layers of pre-trained network, forming a similarity matrix that guides the clustering process. Channels within each cluster are then fused to produce a compressed network. Experimental results on three medical image datasets application demonstrate that MIS-HCC outperforms the state-of-the-art methods in both accuracy and compression efficiency, offering an effective solution for deploying medical image segmentation models on resource-limited medical platforms.

医学图像模型压缩聚类轻量化

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