Literature DB >> 35789884

A Novel Distant Domain Transfer Learning Framework for Thyroid Image Classification.

Fenghe Tang1, Jianrui Ding1, Lingtao Wang1, Chunping Ning2.   

Abstract

Medical ultrasound imaging technology is currently the preferred method for early diagnosis of thyroid nodules. Radiologists' analysis of ultrasound images is highly dependent on their clinical experience and is susceptible to intra- and inter-observer variability. Although end-to-end deep learning technique can address these limitations, the difficulty of acquiring annotated medical image makes it very challenging. Transfer learning can alleviate the problems, but the large gap between source and target domain will lead to negative transfer. In this paper, a novel transfer learning method with distant domain high-level feature fusion (DHFF) model is proposed. It reduces the distribution distance between the source domain and the target domain while maintaining the characteristics of respective domains, which can avoid excessive feature fusion while enabling the model to learn more valuable transfer knowledge. The DHFF is validated by multiple public source and private target datasets in experiments. The results show that the classification accuracy of DHFF is up to 88.92% with thyroid ultrasound auxiliary source domains, which is up to 8% higher than existing transfer and distant transfer algorithms.
© The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2022.

Entities:  

Keywords:  Distant domain; Feature Fusion; Thyroid image classification; Transfer learning

Year:  2022        PMID: 35789884      PMCID: PMC9243866          DOI: 10.1007/s11063-022-10940-4

Source DB:  PubMed          Journal:  Neural Process Lett        ISSN: 1370-4621            Impact factor:   2.565


Introduction

Ultrasound imaging technology is a non-invasive, non-radiation, low cost and real time detection method. It has widely used in the detection of thyroid, fetal, mammary gland and gonadal tissue [1]. However, due to its low contrast, Manual image analysis is time-consuming and laborious. At the same time, it will be affected by subjective factors such as radiologists’ experience and mental state, which is prone to misdiagnosis. Automatic medical image analysis techniques can effectively overcome the above limitations. Generally, it can be roughly classified into two main categories: hand-crafted feature-based methods and the data-driven methods. The pipeline of hand-craft feature methods frequently involves feature extraction and classification. Despite their rapid development in recent years, handcrafted features are highly dependent on expert knowledge. Moreover, handcrafted features in some sense ignore high-level abstract information that is not visible to the human eye in ultrasound images and it can only exploit the low-level information, such as image texture [2, 3], geometry morphology [4], and statistical distributions [5]. Such methods usually require further employ classifiers to conduct classification. Therefore, only given the highly discriminative features, this method can solve the recognition problem well. On the contrary, data-driven methods, without the need of hand-crafted feature description, can greatly improve the classification performance of medical images by using the convolution neural networks (CNNs) [6]. At present, deep learning technology in data-driven approaches has become the mainstream method of image analysis and understanding [7]. It depends on large-scale labeled training data. However, compared with natural scene images, the collection and annotation of medical images is difficult and expensive, which brings great challenges to the application of deep learning technology in medical domain. Transfer learning [8] is one of the effective methods to solve small data learning, which has been widely used in medical image analysis [9]. Theoretically, it attempts to build a robust model by transferring the knowledge learned from the source domain with large-scale training data to the target domain with a small amount of data. However, low or even irrelevance between the source and target domains may lead to negative transfer [8], which causes the knowledge generated by the source domain negatively affects the target domain. How to transfer the knowledge beneficial to medical image analysis from the source domain is a challenging problem. In this paper, we propose a method to transfer valuable knowledge from distant domains which have low correlation or even seemingly unrelated with target domain, and apply it to thyroid ultrasound image classification. Summary of distant domain transfer learning method Distant domain transfer learning (DDTL) [10] is a new transfer learning method. It is inspired by the human ability to learn new things by integrating knowledge obtained from several seemingly independent things. Considering the variation between distant domain and target domain, the auxiliary domain is served as an intermediary bridge to narrow the gap between them. Most of the existing methods [10-13] apply simple auto-encoders as feature extractors. The extracted features are more representative of the low-level details of the image and lack the ability to represent the domain Therefore they are greatly affected by the variation among domains. The performance and stability are unsatisfactory. The existing distant domain transfer learning methods in the past are summarized as listed in Table 1 and we introduce the limitation of these methods. Inspired by feature-based [12] and instance-based [11] DDTL methods, we propose a DDTL method which extracts high-level semantic features from different domains and performs distant domain feature fusion. An auxiliary domain is also adopted as a bridge to reduce the gap between source domain and target domain. It can transfer valuable knowledge to the target domain, reduce the cost and effort of collecting training data in target domain, and suppress the negative effects resulting from the irrelevant parts of distant domains. The flow chart of our method is shown in Fig. 1. Firstly, crop region of interest in thyroid ultrasound dataset and augment the source and the target domain dataset. Secondly, source and target domain high-level semantic features are extracted by high-level semantic features extractor (CNN). And then utilize an auto encoder-decoder to perform high-level semantic feature fusion, and the decoder is applied to maintain the diversity of each domain. Finally, a target classifier is deployed for target domain classification. The effectiveness of this method is verified on thyroid ultrasound dataset. Our main contributions are summarized as follows:The rest of the paper is organized as follows. In Sect. 2 and Sect. 3, we briefly describe the recent work of DDTL and formulate the DDTL problem definition. In Sect. 4 we present the details of the proposed method. In Sect. 5, we present experimental results and analysis. Finally, in Sect. 6, we conclude our work.
Table 1

Summary of distant domain transfer learning method

DDTL TechniquesMethodLimitations
TTL [10]Instances-basedHighly dependent intermediate domain, which is selected by users manually
SLA [11]Instances-basedNeed to adjust the conditions for selecting instances according to different tasks
Xie et al. [21]Feature-basedRequires a large amount of labeled intermediate data, which can be too expensive to apply
DFF [12]Feature-basedKnowledge transfer only for low-level semantic features
AM-DDTL [13]Feature-basedThe feature extraction is computationally expensive
Fig. 1

The flow chart of our method

The feature extraction network is redesigned to mine the information transferred from the source domain to the target domain that is beneficial to the target task. Based on resnet50, a lightweight feature extraction network is designed to extract high-level semantic features that can better represent the source and target domains. Propose a novel distant domain high-level semantic feature fusion method. A high-level semantic feature adaption encoder with domain distance measure is designed to discover valuable knowledge across different domains. To improve the generalization performance of transfer learning, the diversity of source and target domains is preserved while narrowing the difference between source and target domains. A decoder and content loss are added to narrow the content gap between the reconstructed features and the input features, which helps maintain the invariance of the source domain and the target domain. The effectiveness of the proposed method is verified by extensive experiments on open-source datasets, thyroid and breast ultrasound images. The flow chart of our method

Related Work

Deep learning models, especially convolutional neural networks (CNN), have shown advantages over other traditional learning methods in thyroid ultrasound image tasks [6, 14]. Many scholars use CNN to achieve the task of thyroid nodule image classification and segmentation [15-17]. Unfortunately, the lack of annotated training data limits the performance of most models. The transfer learning method can effectively solve the above problems, and has been applied to thyroid ultrasound image task [14, 18, 19]. However, traditional transfer learning methods are still limited by the distribution variation between the source and target domains. Recent studies [20] have shown that fine-tuning a pretrained model on natural images may not help to significantly improve the accuracy of medical image classification. Unlike fine-tuning, DDTL focuses on making use of the knowledge in distant domains to improve the performance of the task in the target domain. Moreover, in the previous research of DDTL algorithm, Tan [10, 11] introduced two instance-based algorithms: Transitive Transfer learning (TTL) and Selective Learning Algorithm (SLA). TTL transfers knowledge between text data in the source domain and the image data in the target domain by using annotated image data as a bridge. SLA select helpful instances from several unrelated auxiliary domains to expand the source domain’s volume. However, the limitations of these two methods are that they are unstable and can only be used for binary classification. Xie [21] proposed a feature-based method to predict poverty using satellite imagery. This method exploits nighttime light intensity as an intermediate domain to transfer the knowledge learned from natural images during the day to high-resolution satellite images, but this method requires a large amount of annotated data in auxiliary domain. Niu [12] proposed another feature-based method (DFF) to the classification and diagnosis of COVID-19 images. DFF does not require labeled data in source and intermediate domains, it can solve multi-classification problems as well. However, DFF cannot perform high-level semantic features fusion, and its model is only used to process low-noise lung CT images. Qin [13] added an attention mechanism into DDTL for extraction more effective information from satellite images, however the processing of this method feature extraction is computationally expensive and it still unable to extract sufficient high-level semantic features. Recent work [20] shows that low-level image features are more likely to be reused in transfer learning. However, these low-level image features in different domains are very similar, which cannot provide additional valuable domain knowledge to be transferred. On the contrary, high-level semantic features can better represent the essential characteristics of the domain. Therefore, extracting and fusing the high-level semantic features of the domain can maximize the learning of additional valuable knowledge in the distant domain. In this paper, we aim to implement efficient transfer of high-level semantic information in distant domain through DDTL. The layer that can extract high-level semantic features is retained in CNN model, and a lightweight network is designed to realize efficient high-level semantic feature fusion between distant domain and target domain. Then an encoder-decoder network is utilized to retain the high-level features of each domain, increase the diversity and improve the generalization of the model. Finally, multiple loss functions are integrated to make a trade-off among classification, transferring knowledge and preserving the diversity.

Preliminary

Problem Definition

The goal of DDTL is to transfer the valuable information in distant source domains to target domain. An auxiliary domain is also used as a bridge between the source domain and the target domain. According to the DDTL problem, we assume that a small amount of target domain data is not enough to train a robust model. The target domains T with fewer annotated data are denoted as:where and represent the number of samples in ith target domain and the number of target domains, is the jth sample of the ith target domain, is corresponding label. Then we denote the unlabeled source domain S as:where and represent the number of samples in ith source domain and the number of source domains, furthermore, we denote the unlabeled auxiliary domain A as:where and represent the number of samples in ith auxiliary domain and the number of auxiliary domains. Let the marginal distribution and conditional distribution of source domain data be denoted as , the distribution of target domain data be denoted as ,, and the distribution of auxiliary domain be denoted as , we have the following assumptions:

Maximum Mean Discrepancy

Maximum Mean Discrepancy (MMD) [22] is an important statistical indicator to measure the discrepancy between the source and target domain distributions ( and ). Given the distributions s and t over two domains, MMD is presented as:where represents the kernel function that maps the original data to the Reproduced Kernel Hilbert Space (RKHS) [22]. The empirical estimate of MMD is defined as:where M and N are the number of instances in the source and target domains.

Distant High-Level Feature Fusion Model (DHFF)

In this section, we present the proposed DHFF. Firstly, we introduce the redesigned high-level semantic feature extractor. After that, we present the implementation details of the encoder-decoder network and define the content loss for maintaining different domains invariance. And then, we denote high-level semantic feature adaption methods. Finally, the classification loss and the overall loss of DHFF are defined. Our proposed model DHFF is shown in Fig. 2. It mainly consists of four parts: High-level Semantic Feature Extraction, High-level Semantic Feature Encoder-Decoder, High-level Semantic Feature Adaption and Target domain Classification. The last three parts correspond to three types of losses: content loss, domain loss, and classification loss.
Fig. 2

DHFF model

DHFF model The specific structure of the encoder-decoder

High-level Semantic Feature Extraction

Inspired by a recent study [20], we use hybrid approaches to transfer learning where a subset of pretrained weights are used, and the high-level parts of the network are redesigned and made more lightweight. In this paper, we design a lightweight feature extraction network based on ResNet50 [23]. In the structure of ResNet50, the closer to the output, the more the features can reflect the high-level semantic information of the input, which is more conducive to the classification task, and the greater the correlation with the domain. In DDTL, we want to transfer features from the source domain to the target domain to improve the performance of tasks in the target domain. Therefore, the correlation between the transferred features and the source domain should be reduced. In the structure of ResNet50, the features closer to the input can reflect the low-level features of the image, such as edge, texture and other information. These features cannot represent the domain well. As a trade-off, we retain the first three layers of ResNet50, extract the information that can both reflect the source domain and target domain. The redesigned feature extractor can also prevent the large fluctuation of weight when trained on different domain data and accelerate the convergence.

Encoder-Decoder Network and Content Loss

In order to maintain the diversity of features, avoid excessively narrowing the inter-domain distribution distance. In [24], it exploits autoencoder pair to minimize the reconstruction error on all the training instances. Autoencoder is an unsupervised feedforward neural network capable of efficient feature extraction and dimensionality reduction. However, using the it directly may not be able to effectively learn valuable high-level semantic information, and its performance is unstable. Therefore, an encoder-decoder network is applied to our study to extract and preserve inter-domain high-level semantic features. Generally, a convolutional autoencoder pair includes an input layer, an output layer, an up-sampling layer and multiple convolutional layers. The output from feature extractor in 4.1 is encoded with encoder and decoded by the decoder to reconstruct the high-level semantic features. The specific structure of the encoder-decoder is shown in Fig. 3. The encoder includes two pooling layers and two convolutional layers, conv layer4 and conv layer5 respectively. Specifically, a 3x3 convolution kernel with pad of 1 and stride of 1 is applied, and a 2x2 max pooling layer is used for down-sampling. The decoder consists of two convolutional layers and two up-sampling layers. We adopt 2x2 for up-sampling to maintain the same quality of reconstructed feature maps. The standard process of encode-decode can be demonstrated as:where f is the output of redesigned ResNet50 model, is a higher-level abstract feature of f, is the high-level semantic feature after reconstruction. Our motivation is that, if the knowledge transferred from the source domain is helpful to the target domain, the encoder and decoder can be learned to maintain the integrity of the respective domain features while minimizing the difference between the high-level semantic features of the source and target domains. The auxiliary domain is also used to prevent excessive fusion of high-level semantic features between different domains. Inspired by style transfer, we adopt content loss to measure the difference between reconstruction features and input features. It is defined as:when adding a auxiliary domain, the content loss objective function to be minimized is formulated as:where , and represent the high-level semantic features of a sample in source domain, target domains and auxiliary domains respectively, , and are the reconstructed features of the sample in the corresponding domains, ,,are the number of samples in source domain, target domain and auxiliary domain.
Fig. 3

The specific structure of the encoder-decoder

High-Level Semantic Feature Adaption

Minimizing the content loss can preserve the integrity of the respective domain features and ensure the diversity. However, in order to use the knowledge transferred from distant domain to help the target domain task, it is necessary to narrow the distance from the distant domain to the target domain. We introduce the Maximum Mean Discrepancy (MMD) mentioned in 3.2 as domain loss to measure the distribution distance between distant domain and target domain. It is defined as:when adding auxiliary domains, it is formulated as:where , and represent the high-level semantic features extracted by encoder in 4.2.

Target Domain Classification

Two fully-connected layers are added after the encoder in Fig. 2 to build a target classifier. They can find the best combination of high-level semantic abstraction features for the target task. The cross-entropy loss is adopted as classification loss. It is defined as follows:where is the prediction result of sample ith in target domain, is the label corresponding to the sample ith. Finally, the overall loss of DHFF can be expressed as:where ,,, are the parameters of the high-level semantic feature extractor, encoder, decoder and the classifier. Finally, all parameters in the network are optimized by minimizing the objective function L.

Experimentation

Datasets

As shown in Table 2, We evaluate the DHFF model with five datasets, three of which are open source datasets: Catech-256 [25], Office31 [26] and Stanford Open Source Thyroid Nodule Ultrasound Image Dataset (Thyroid Ultrasound Cine-clip) [27], private datasets: thyroid and breast ultrasound datasets acquired from different ultrasound machines at the Affiliated Hospital of Qingdao University. Catech-256 is a natural image dataset that includes labeled data of 256 different class. Then, Office-31 includes 31 different common office items and is composed of a collection of three different domains: “Webcam”, “Dslr” and “Amazon”. Furthermore, the Stanford open-source thyroid nodule dataset consists of 17,412 thyroid ultrasound images provided by 167 patients. Moreover, the thyroid ultrasound dataset provided by Affiliated Hospital of Qingdao University consists of 3,003 images collected by different ultrasound machines, 2268 malignant samples and 735 benign samples. And the breast ultrasound dataset consists of 725 images.
Table 2

dataset

Data SetClasses numSamples numLabelMask
Catech-25625630670YesNo
Office31314110YesNo
Thyroid Ultrasound Cine-clip217412YesYes
Thyroid Ultrasound23003YesYes
Breast Ultrasound2725YesYes
Some ultrasound thyroid images are shown in Fig. 4. Fig. 4a shows the benign nodules, and most of them have irregular shapes, smooth regions, and boundaries. Fig. 4b shows the malignant nodules, and most of them have irregular shapes, coarse regions, and boundaries. As can be seen from Fig. 4, there is some overlap in the image characteristics between benign and malignant thyroid nodule. Therefore, it is difficult to differentiate thyroid nodules based on low-level features such as gray scale and shape.
Fig. 4

Illustration of thyroid nodules: (a) Benign nodules; (b) Malignant nodules

dataset Illustration of thyroid nodules: (a) Benign nodules; (b) Malignant nodules

Baseline Model and Experiment Setups

As shown in Fig. 5, the modified ResNet50 model (ResNet50-baseline) is used as a baseline to verify the rationality and effectiveness of our proposed method (DHFF). It exploits ResNet50 as a feature extractor and implements domain-adaptive transfer learning by minimizing the marginal distribution of the two domains.
Fig. 5

baseline model based on Resnet50

In single distant source domain transfer learning category, we choose six transfer learning algorithms including Deep Transfer Learning (DTL) [28], Co-Tuning [29], Selective Learning Algorithm (SLA) [11], feature adaption baseline model (ResNet50-baseline), Distant Feature Fusion (DFF) [12] and AM-DDTL [13]. The ResNet50-baseline and DHFF are initialized with parameters pretrained on the ImageNet dataset [30], while the other methods are trained from scratch respectively. For DTL, we first train a deep model in the source domain, and then train another deep model for the target domain by reusing the first several layers of the source model. And for Co-Tuning, we use ResNet50 as a source domain pretrain model. For SLA, DFF, DHFF, ResNet50-baseline and AM-DDTL, we use all the source domain, target domain data to learn a model. Meanwhile, in multiple source domains transfer learning category, we choose four transfer learning algorithms including SLA, ResNet50-baseline, DFF and AM-DDTL. We use all the source domain, target domain and auxiliary source domains data to learn a model. baseline model based on Resnet50 In all experiments, as shown in Fig. 6, according to the ground truth boundary of the thyroid nodule delineated by the radiologist, the minimum circumscribed rectangle containing the nodule area is cropped from the image as the input. Meanwhile, in experiments 1-3, we use the full thyroid ultrasound dataset for experiments. In experiment 4 we conduct experiments with a small thyroid ultrasound dataset, which has 716 images. Specifically, it consists of 360 malignant samples and 356 benign samples. Besides, in our 1-3 experiments, the thyroid dataset is split by 6:2:2 for training, validating and testing respectively. Meanwhile, in our 4 experiments, the thyroid dataset is split by 7:2:1 for training, validating and testing respectively. Moreover, the random cropping and horizontal flipping are used for natural images, and horizontal flipping is used for medical ultrasound images data augmentation. Furthermore, each experiment was run eight times and the accuracy was averaged to remove performance fluctuations due to parameter initialization.
Fig. 6

thyroid ultrasound datasets preprocessing

thyroid ultrasound datasets preprocessing In Experiment 1, six datasets, Catech-256, Amazon, Dslr, Webcam, Thyroid Ultrasound Cine-clip, and breast ultrasound datasets, are served as unlabeled source domains to explore the correlation of natural scene images and ultrasound images with targets domains. The experimental results are shown in Table 3. Then in Experiment 2, four datasets, Catech-256, Amazon, Dslr, Webcam are used as unlabeled source domains and breast ultrasound images is served as auxiliary domain, which can be regarded as related to target domain (thyroid ultrasound images). This experiment is utilized to verify whether the auxiliary domain close to the target domain is helpful for distant domain transfer. The experimental results are shown in Table 4. Finally, in Experiment 3, five datasets, Catech-256, Amazon, Dslr, Webcam, and breast ultrasound datasets are used as unlabeled source domain and Stanford thyroid ultrasound datasets is served as auxiliary domain. The experimental results are shown in Table 5.
Table 3

Accuracies (%) of models with single source domain

Source domainCatech256AmazonWebcamDslrBreastTucc
DTL82.9977.3773.6374.7280.3481.43
Co-Tuning83.9380.8879.0978.9382.1886.27
DFF75.3276.9676.9976.4673.1376.02
SLA79.4080.4979.8779.2580.0379.71
AM-DDTL80.49 80.4080.1979.4081.1283.46
Resnet5074.7074.8776.0477.4880.0585.58
DHFF78.5376.1479.5482.5084.1487.20
Table 4

Accuracies (%) of models with breast ultrasound auxiliary source domain

Source domainCatech256AmazonWebcamDslr
Auxiliary Source DomainBreast ultrasound
DFF76.7077.2076.1777.17
Resnet5077.9876.5277.0082.93
SLA81.7482.2182.2180.03
AM-DDTL82.8382.5281.2781.43
DHFF82.6883.4484.3986.66
Table 5

Accuracies (%) of models with thyroid ultrasound auxiliary source domain

Source domainCatech256AmazonWebcamDslrBreast
Auxiliary Source DomainThyroid Ultrasound Cine-clip
DFF76.3675.7876.5978.7574.76
Resnet5083.0679.1975.2385.5087.33
SLA82.8382.0582.6881.1280.49
AM-DDTL83.6182.5281.5982.0583.93
DHFF84.3982.7484.7386.7388.92
Accuracies (%) of models with single source domain Accuracies (%) of models with breast ultrasound auxiliary source domain Accuracies (%) of models with thyroid ultrasound auxiliary source domain

Experimental Result Analysis

For single-source domain experiment, as shown in Table 3, we can see that the transfer learning methods such as DTL, DFF and ResNet50-baseline achieve worse performance than SLA, AM-DDTL and DHFF by using the Office 31 (distant domains) as the source domain, because the source domain and the target domain have huge distribution gap, which leads to negative transfer. SLA performs well because it selects knowledge valuable for distant transfer learning based on instance and AM-DDTL benefits from the CBAM (Convolutional Block Attention Module) attention mechanism which making use of the channel and the spatial attention for better feature extraction. In addition, our proposed DHFF model performs well on different datasets, DHFF can not only extract appropriate high-level semantic features, but also use content loss to avoid negative transfer caused by excessive feature fusion between two different domains. It achieves the highest accuracy (87.20%) using the Thyroid Ultrasound Cine-clip dataset (TUCC) as the source domain, which has the highest correlation with the target domain. Moreover, compared with DTL, Co-Tuning, applying the collaboratively supervise the fine-tuning process, can ameliorate the negative transfer effect of traditional fine-tuning on distant domains, however the method is still not enough to obtain more valuable transfer knowledge on the source domain (Breast and Tucc) which has high correlation and the same categories with the target domain. And when the target domain dataset is small, the Co-Tuning is easy to overfit to the limited labeled train data. Furthermore, compared with the DHFF, DTL and ResNet50-baseline, the feature extraction structure of DFF, SLA is so simple and cannot extract enough high-level semantic information in ultrasound imaging dataset, so its results are not satisfactory. Although AM-DDTL alleviates this problem, its feature extraction relying on attention mechanism is still insufficient. It is proved that high-level semantic information plays an important role in distant domain transfer learning. When the correlation between the source domain and the target domain is low, the transferred knowledge is not helpful for the task of the target domain, and may even cause negative transfer. In experiment 2, we adopt breast ultrasound images as the auxiliary domain and natural images as the source domain. The results in Table 4 shows that the auxiliary domain which closer to the target domain may help to narrow the gap between the source and target domain and is beneficial to improve the performance of the target task. In this experiment, compared with ResNet50-baseline, the SLA, DFF and AM-DDTL which belongs to DDTL methods can alleviate distant domain negative transfer. But they cannot sufficiently extract high-level semantic features and the performance is lower than that of the method proposed in this paper. In contrast, DHFF achieves the highest accuracy (86.66%) when breast ultrasound images are served as auxiliary domain. It further proves DHFF can effectively transfer valuable knowledge compared with other models. In addition, we also experiment with thyroid ultrasound images as the auxiliary domain in experiment 3. The results shown in Table 5 prove that using the thyroid ultrasound images as the auxiliary domain which share the most useful knowledge with the target domain can maximize the model classification accuracy. This shows that the auxiliary domain which is more relevant to the target domain can help distant transfer learning. Surprisingly, when the breast dataset and thyroid datasets are adopted as source domain and auxiliary domain respectively, the highest classification accuracy (88.92%) is achieved. This shows that when the thyroid dataset is used as a bridge, the breast dataset with relatively strong correlation can provide the most valuable knowledge to the target domain. On the contrary, SLA is not satisfactory in the high correlation source domains, which proves that the instances-based method may ignore the potentially valuable transfer knowledge. Moreover, although highly correlation auxiliary domain facilitates the valuable information transfer, DFF and AM-DDTL which extract low level features to fusion cannot provide effective information for the classifier. Finally, we conduct experiments with a small thyroid ultrasound dataset (716 images). The results are shown in Table 6. Our method DDHF can still achieve best results (86.27%), which benefits from the transfer of valuable knowledge from the source and auxiliary domains to the target domain. However, negative transfer occurs on all five source domains based on the DFF, SLA and AM-DDTL model. The reason is that due to the simple feature extractor and few samples, DFF and SLA cannot effectively learn high-level semantic features from different domains, making the features of different domains excessive overlap, ultimately lead to the model underfitting the target domain data. In addition, the attention mechanism model AM-DDTL which relies on a large amount of data cannot achieve good results.
Table 6

Accuracies (%) of models with few samples

Source domainCatech256AmazonWebcamDslrBreast
Auxiliary Source DomainThyroid Ultrasound Cine-clip
DFF70.2469.2671.3069.2270.33
Resnet5080.7579.4683.6385.0585.11
SLA76.4775.4975.4978.4377.45
AM-DDTL77.5677.0778.5379.0281.46
DHFF85.2380.9785.0985.2786.27
Accuracies (%) of models with few samples Further, we analyzed the samples which misclassified by the model. Fig. 7 shows that benign samples are misclassified as malignant, and Fig. 8 shows that malignant samples are misclassified as benign.
Fig. 7

benign samples misclassified as malignant

Fig. 8

malignant samples misclassified as benign

benign samples misclassified as malignant malignant samples misclassified as benign In Fig. 7, each sample shows some malignant features, such as no cyst, hypoechoic, rough edge. They are classified as Ti-rads4 by radiologist. In Fig. 8(a), the internal echo attenuation is too great, which is close to cyst. In Fig. 8(b), there is some wall structure, which is benign feature. From the analysis, it can be seen that overlap in the image characteristics between benign and malignant thyroid nodule is one of the causes of classification errors. The limited number of training samples also affects the generalization performance of the model. In addition, the static image only reflects one section of the nodule, and cannot reflect the whole picture of the nodule. Using the dynamic image or multiple static images of a nodule can help to improve the accuracy.

Conclusions

This paper proposes a novel method to solve the problem of distant domain transfer learning. The purpose is to transfer valuable information in the distant domain to improve the task performance in target domain. A lightweight feature extractor is proposed to extract high-level semantic features in both source domain and target domain. The MMD loss is to narrow the distribution distance of source domain and target domain. Then, an encoder-decoder network and content loss are proposed to maintain the diversity of the source domain and the target domain. Experiments show that this method can effectively transfer knowledge from distant domain and improve the accuracy of thyroid ultrasound image classification. When the samples in the target domain are difficult to collect and annotate, our method can effectively transfer the knowledge in the distant domain, which is conducive to improving the task in the target domain. This makes the deep learning model more applicable to small sample scenarios. Generally, we think that training data and test data are independent and identically distributed (IID), but this is not the case in practical applications. For example, the thyroid ultrasound images collected by different machines usually are not IID, which brings difficulties to the learning task. The proposed method can partly solve the problem by transfer the knowledge of the distant domain, to improve the generalization of the model. For example, in the experiment, using breast ultrasound image as the source domain can improve the classification performance of thyroid ultrasound image. Due to the high cost of collecting and labeling data in medical imaging, algorithms in this field usually use small data for training. To improve performance, transfer the model trained in big data to the target task is a simple and effective method. But if the source and target domain are not similar enough, it will be difficult to use transfer learning directly. From the perspective of target task, we hope to shorten the distance between source domain and target domain; from the perspective of generalization, we hope to retain the differences between the source domain and the target domain, and transfer high-level semantic features which is helpful to the target task. This is a trade-off. The method proposed in this paper makes a useful exploration to solve this problem. Although some good results have been achieved, there are still several limitations: 1) Lack of means to guide the selection of the source domain to maximize the performance of the target domain. 2) The auxiliary domain needs to use unlabeled ultrasound images with high similarity to the target domain, which is relatively difficult to collect. In future research, we will further analyze which knowledge transferred by the model is helpful to improve performance of target domain tasks and which knowledge will have an adverse impact on target tasks. So as to further improve the performance of the model, reveal the internal working principle of the model and improve the interpretability of the model.
  12 in total

1.  Integrating structured biological data by Kernel Maximum Mean Discrepancy.

Authors:  Karsten M Borgwardt; Arthur Gretton; Malte J Rasch; Hans-Peter Kriegel; Bernhard Schölkopf; Alex J Smola
Journal:  Bioinformatics       Date:  2006-07-15       Impact factor: 6.937

2.  Morphological and wavelet features towards sonographic thyroid nodules evaluation.

Authors:  Stavros Tsantis; Nikos Dimitropoulos; Dionisis Cavouras; George Nikiforidis
Journal:  Comput Med Imaging Graph       Date:  2008-12-25       Impact factor: 4.790

Review 3.  Deep learning.

Authors:  Yann LeCun; Yoshua Bengio; Geoffrey Hinton
Journal:  Nature       Date:  2015-05-28       Impact factor: 49.962

4.  Multitask Cascade Convolution Neural Networks for Automatic Thyroid Nodule Detection and Recognition.

Authors:  Wenfeng Song; Shuai Li; Ji Liu; Hong Qin; Bo Zhang; Shuyang Zhang; Aimin Hao
Journal:  IEEE J Biomed Health Inform       Date:  2018-07-03       Impact factor: 5.772

5.  Knowledge-guided synthetic medical image adversarial augmentation for ultrasonography thyroid nodule classification.

Authors:  Guohua Shi; Jiawen Wang; Yan Qiang; Xiaotang Yang; Juanjuan Zhao; Rui Hao; Wenkai Yang; Qianqian Du; Ntikurako Guy-Fernand Kazihise
Journal:  Comput Methods Programs Biomed       Date:  2020-06-30       Impact factor: 5.428

6.  Diagnosis of Benign and Malignant Thyroid Nodules Using Combined Conventional Ultrasound and Ultrasound Elasticity Imaging.

Authors:  Pinle Qin; Kuan Wu; Yishan Hu; Jianchao Zeng; Xiangfei Chai
Journal:  IEEE J Biomed Health Inform       Date:  2019-11-04       Impact factor: 5.772

7.  Online Transfer Learning for Differential Diagnosis of Benign and Malignant Thyroid Nodules With Ultrasound Images.

Authors:  Hui Zhou; Kun Wang; Jie Tian
Journal:  IEEE Trans Biomed Eng       Date:  2020-02-03       Impact factor: 4.538

8.  Distant Domain Transfer Learning for Medical Imaging.

Authors:  Shuteng Niu; Meryl Liu; Yongxin Liu; Jian Wang; Houbing Song
Journal:  IEEE J Biomed Health Inform       Date:  2021-10-05       Impact factor: 5.772

9.  Thyroid Nodule Classification in Ultrasound Images by Fine-Tuning Deep Convolutional Neural Network.

Authors:  Jianning Chi; Ekta Walia; Paul Babyn; Jimmy Wang; Gary Groot; Mark Eramian
Journal:  J Digit Imaging       Date:  2017-08       Impact factor: 4.056

Review 10.  Machine Learning in Ultrasound Computer-Aided Diagnostic Systems: A Survey.

Authors:  Qinghua Huang; Fan Zhang; Xuelong Li
Journal:  Biomed Res Int       Date:  2018-03-04       Impact factor: 3.411

View more

北京卡尤迪生物科技股份有限公司 © 2022-2023.