Tri-correcting: Label noise correction via triple CNN ensemble for carotid plaque ultrasound image classification
Ran Zhou,Weiyan Gan,Furong Wang,Zhi Yang,Zhongwei Huang,Haitao Gan
DOI: https://doi.org/10.1016/j.bspc.2024.105981
IF: 5.1
2024-02-08
Biomedical Signal Processing and Control
Abstract:In clinical practice, classifying carotid artery plaque is vital for assessing coronary artery disease (CAD) risk. Deep supervised learning has achieved significant success in carotid artery plaque ultrasound image classification, relying on large-scale accurately annotated datasets. However, label noise can significantly degrade model performance due to the professional knowledge and experience of annotators during data collection. Previous studies tended to collaboratively train two models to alleviate the above degradation, but they did not consider how to determine the final hypothesis when the two models are inconsistent. Inspired by Tri-Training, this study proposes a label noise correction method by using a triple convolutional neural networks (CNN) ensemble to improve the robust and accuracy of carotid plaque classification. Firstly, the noise filtering strategy is used to obtain a support dataset and a noisy dataset from the original dataset. Then, three CNNs models are trained using the support dataset, and the label noise dataset is corrected using the joint voting strategy among the three CNNs. The CNN models are then retrained using the instance allocation strategy before finally integrating and classifying with the three classifier models. Evaluated on 1270 carotid ultrasound images from 844 subjects followed in Zhongnan Hospital (Wuhan, China), our method improves the classification accuracy by 4%-5%, 3%, and 2%-3% respectively in the case of 10%, 20%, and 30% label noise rates compared to other advanced methods, which may help clinical doctors diagnose carotid artery plaque categories and evaluate the risk of CAD for patients.
engineering, biomedical