Tracking the Rareness of Diseases: Improving Long-Tail Medical Detection with a Calibrated Diffusion Model
Year: 2024
Type: article
Abstract: Motivation: Chest X-ray (CXR) is a routine diagnostic X-ray examination for checking and screening various diseases. Automatically localizing and classifying diseases from CXR as a detection task is of much significance for subsequent diagnosis and treatment. Due to the fact that samples of some diseases are difficult to acquire, CXR detection datasets often present a long-tail distribution over different diseases. Objective: The detection performance of tail classes is very poor due to the limited number and diversity of samples in the training dataset and should be improved. Method: In this paper, motivated by a correspondence-based tracking system, we build a pipeline named RaTrack, leveraging a diffusion model to alleviate the tail class degradation problem by aligning the generation process of the tail to the head class. Then, the samples of rare classes are generated to extend the number and diversity of rare samples. In addition, we propose a filtering strategy to control the quality of the generated samples. Results: Extensive experiments on public datasets, Vindr-CXR and RSNA, demonstrate the effectiveness of the proposed method, especially for rare diseases.
Source: Electronics
Authors: Tianjiao Zhang, Chaofan Ma, Yanfeng Wang
Language: English
Topic: COVID-19 diagnosis using AI
Subfield: Radiology, Nuclear Medicine and Imaging
Field: Medicine
Domain: Health Sciences
Open Access status: gold
Estimated APC paid: $2,728
OpenAlex ID: W4404756196