IEEE TFS 2024 | FMDNN:模糊引导的多粒度深度神经网络及其在组织病理图像分类中的应用

发布时间:2024-06-09 阅读次数:28

FMDNN: A Fuzzy-guided Multi-granular Deep Neural Network for Histopathological Image Classification

Weiping Ding*, Tianyi Zhou, Jiashuang Huang, Shu Jiang, Tao Hou and Chin-Teng Lin



MOTIVATION

  • Histopathological images commonly encompass a variety of morphological features, existing methods neglect the features of histopathological images at diverse granularity levels, limiting feature extraction to a singular scale, which leads to the incomplete capture of intrinsic feature information among cells.

  • Feature extraction is often performed at a single granularity, overlooking the multi-granular characteristics of cells.


INNOVATION

  • We introduce a multi-granular feature extraction method and employ feature visualization to investigate the semantic scales of features at different granularities, which contributes to a more comprehensive interpretation of multi-granular information embedded in histopathological images.

  • We use fuzzy logic to amalgamate key features from images through the computation of universal fuzzy features using three membership functions, which are utilized to globally guide the learning of multi-granular features. Experiments demonstrate the effectiveness of this approach in enhancing classification accuracy during model-guided training while minimizing interference from unrelated features.

  • To address information redundancy in multi-granular feature extraction, we employ fuzzy-guided cross-attention to enhance the importance of key features. Continuously guiding features at three granularities with universal fuzzy features reduces interference from irrelevant information, significantly improving classification performance, and achieves good results on five public datasets relative to other algorithms.



METHOD

Fig. 1. Overview of proposed FMDNN.

As shown in Fig. 1, FMDNN consists of three modules: theMulti-granular Feature Extraction Module conducts feature extraction on the input image at three distinct granularities, theUniversal Fuzzy Feature Module extracts the universal fuzzy features of the image, and theFuzzy-guided Cross-attention Module performs feature fusion through linear transformation and dimension alignment to obtain the final classification result.

  • Multi-granular Feature Extraction Moduleemploys a multi-granularity convolution technique to extract features from the original image at various granular levels.

  • Universal Fuzzy Feature Module,through the fusion of membership functions tailored to distinct feature points, extracts the universal fuzzy features of the original image. Extracted multi-granular features and universal fuzzy features are separately embedded with patches at different granularities, harmonizing dimensions and reducing subsequent algorithmic complexity.

  • Fuzzy-guided Cross-attention Module,which utilizes the universal fuzzy features to guide training at distinct granularities. By employing a multi-granularity classifier as an intermediary, features from different granularities and universal fuzzy features are integrated through fusion learning of feature information and are subsequently back-projected into their respective branches.


EXPERIMENTAL RESULTS

TABLE Ⅰ RESULTS OF COMPARATIVE EXPERIMENTS



Fig. 2. Heatmap and CAM map of label "colonca:" (a) original image; (b) FMDNN fine-, medium-, and coarse-grained feature heatmap and CAM map in order; (c) EfficientNet feature heatmap and CAM map in order; (d) AlexNet feature heatmap and CAM map in order.

Fig. 3. Heatmap and CAM map of label "lungaca:" (a) original image; (b) FMDNN fine-, medium-, and coarse-grained feature heatmap and CAM map in order; (c) EfficientNet feature heatmap and CAM map in order; (d) AlexNet feature heatmap and CAM map in order.


Fig. 4. Comparison of ablation experiments on different datasets: (a) LC; (b) NCT.







Link:https://ieeexplore.ieee.org/abstract/document/10552048


搜索
您想要找的