Abstract:Background: Medical images have already become an essential tool for the diagnosis of many diseases. Thus a large number of medical images are being generated due to the daily routine inspection. An efficient image-based disease retrieval system will not only make full use of existing data, but also help physicians to prognosis the diseases. Medical image retrieval is represented by the classification and localization of common thorax diseases in x-ray images. Although extensive efforts have been put into this field, there are still many challenges. Purpose: Most of the existing fine-grained image research methods just apply existing deep learning frameworks in extracting the image features. However, these high-level features mainly focus on the global representations of the object, rather than simultaneously considering the local ones. It requires fine-grained details to classify the images with similar lesion areas. Thus, it is necessary to combine the global features and local ones to make the features more discriminative. On the other hand, training CNN models based on current existing strategies have a high time complexity, and is hard to get the discriminative features mentioned above. In addition, the visual retrieval method of fine-grained medical images still has the problem of insufficient sample data with accurate annotation information. Methods: To address above challenges, we introduced a novel fine-grained medical images retrieval method. First, a centralized contrastive loss (CCLoss) is proposed as our metric learning loss function. Parameters are updated by using the center point, which not only improves the distinguishing performance of features, but also effectively reduces the time complexity of the algorithm. In addition, a weakly supervised progressive feature extraction method is proposed to gradually extract the combined features. And the attention mechanism module is applied to screen the target information after the initial positioning for fine refinement, so as to separate the features with a high degree of discrimination. The retrieval of 14 different chest diseases is evaluated on the chest x-ray datasets. Results: Compared with the existing research methods, the proposed method shows a better retrieval result for Recall@8 by 2.26 % ∼ 4.6 % $\%{\sim }4.6\%$ and achieves a very efficient training speed which is 100 times faster than the pair-wise loss-based training strategy. We also assessed the effects of Recall@k (k = 2, 4, 6, 8) for progressive features extracted from different steps to obtain a model with the best retrieval performance. Conclusions: The proposed model is capable of learning discriminative representations from chest x-ray datasets, and it achieves better performance compared with other state-of-the-art methods. Therefore, the developed model would be useful in the diagnosis of common thorax disease or unknown chest disease.

Modeling Long-Range Dependencies for Weakly Supervised Disease Classification and Localization on Chest X-ray

Weakly Supervised Deep Learning for Thoracic Disease Classification and Localization on Chest X-rays

Research and implementation of multi-disease diagnosis on chest X-ray based on vision transformer

Automatic Localization and Identification of Thoracic Diseases from Chest X-rays with Deep Learning.

A Weakly Supervised Adaptive DenseNet for Classifying Thoracic Diseases and Identifying Abnormalities

Contralaterally Enhanced Networks for Thoracic Disease Detection

Attention-Guided Curriculum Learning for Weakly Supervised Classification and Localization of Thoracic Diseases on Chest Radiographs

Thoracic Disease Identification and Localization using Distance Learning and Region Verification

Spatial Non-Local Attention for Thoracic Disease Diagnosis and Visualisation in Weakly Supervised Learning.

Weighing features of lung and heart regions for thoracic disease classification

Tuberculosis Diagnostics and Localization in Chest X-Rays via Deep Learning Models

Many-to-One Distribution Learning and K-Nearest Neighbor Smoothing for Thoracic Disease Identification

Centralized contrastive loss with weakly supervised progressive feature extraction for fine-grained common thorax disease retrieval in chest x-ray

Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification

Learning Hierarchical Attention for Weakly-Supervised Chest X-Ray Abnormality Localization and Diagnosis.

High-Resolution Network with Dynamic Convolution and Coordinate Attention for Classification of Chest X-ray Images.

Abnormality Detection and Localization in Chest X-Rays using Deep Convolutional Neural Networks

Iterative Attention Mining for Weakly Supervised Thoracic Disease Pattern Localization in Chest X-Rays

Data-Efficient Vision Transformers for Multi-Label Disease Classification on Chest Radiographs

Three-Stage Framework for Accurate Pediatric Chest X-ray Diagnosis Using Self-Supervision and Transfer Learning on Small Datasets

Lesion Detection of Chest X-Ray Based on Scalable Attention Residual CNN