Search papers, labs, and topics across Lattice.
This paper introduces MIS-HCC, a hierarchical clustering method designed to enhance the efficiency of medical image segmentation models while maintaining high accuracy. By utilizing the Wasserstein distance to assess channel similarity, the method effectively clusters and fuses parameters, resulting in a compressed network that retains essential features of the original pretrained model. Experimental results across three medical image datasets indicate that MIS-HCC significantly surpasses existing state-of-the-art techniques in terms of both accuracy and compression efficiency, making it suitable for deployment on resource-constrained medical platforms.
Achieving high accuracy while reducing model size, MIS-HCC compresses medical image segmentation networks without sacrificing performance.
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.
