A self-supervised feature-standardization-block for cross-domain lung disease classification
•We propose a self-supervised FSB, which can be easily extended to any CNN architectures.•We design an GLCM loss to minimize the differences among input images.•Our FSB achieves about 6% accuracy improvement for testing datasets. With the advance of deep learning technology, convolutional neural net...
Gespeichert in:
Veröffentlicht in: | Methods (San Diego, Calif.) Calif.), 2022-06, Vol.202, p.70-77 |
---|---|
Hauptverfasser: | , , , , , , , , , , |
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | •We propose a self-supervised FSB, which can be easily extended to any CNN architectures.•We design an GLCM loss to minimize the differences among input images.•Our FSB achieves about 6% accuracy improvement for testing datasets.
With the advance of deep learning technology, convolutional neural network (CNN) has been wildly used and achieved the state-of-the-art performances in the area of medical image classification. However, most existing medical image classification methods conduct their experiments on only one public dataset. When applying a well-trained model to a different dataset selected from different sources, the model usually shows large performance degradation and needs to be fine-tuned before it can be applied to the new dataset. The goal of this work is trying to solve the cross-domain image classification problem without using data from target domain. In this work, we designed a self-supervised plug-and-play feature-standardization-block (FSB) which consisting of image normalization (INB), contrast enhancement (CEB) and boundary detection blocks (BDB), to extract cross-domain robust feature maps for deep learning framework, and applied the network for chest x-ray-based lung diseases classification. Three classic deep networks, i.e. VGG, Xception and DenseNet and four chest x-ray lung diseases datasets were employed for evaluating the performance. The experimental result showed that when employing feature-standardization-block, all three networks showed better domain adaption performance. The image normalization, contrast enhancement and boundary detection blocks achieved in average 2%, 2% and 5% accuracy improvement, respectively. By combining all three blocks, feature-standardization-block achieved in average 6% accuracy improvement. |
---|---|
ISSN: | 1046-2023 1095-9130 |
DOI: | 10.1016/j.ymeth.2021.05.007 |