Imbalance multiclass Problem: A robust Feature Enhancement-based Framework for Liver Lesion Classification
Rui Xin Hu, Yuqing Song, Yi Liu, Yan Zhu, Nuo Feng, Chengjian Qiu, Kai Han, Qiaoying Teng, Haq Imran Ul, Zhe Liu · Research Square · 2023
Abstract The classification of liver lesions in CT images is essential for the diagnosis and treatment of liver diseases. Since the characteristics of different classes of lesions are similar and the degree of differentiation is not obvious, it is difficult to accurately classify different classes of liver lesions, especially imbalanced multiclass distribution. To this end, we propose a novel feature enhancement-based framework for imbalanced liver lesion classification. Specifically, a liver lesion processing method is introduced to expand data based on Augmentor. Besides, dual feature enhancement based on feature refinement extraction and feature global correlation is designed to enhance feature extraction ability. To further alleviate the problem of class imbalance, the improved loss function based on standard Cross-Entropy (CE) is also adopted to make the network pay more attention to the class with fewer samples. Experiments on 551 liver lesions in 120 patients showed that: 1) Our proposed framework improved the classification performance of liver tumors under imbalanced multiclass data distribution; 2) The dual feature enhancement was lightweight and efficient which enhanced the semantic expression of overall features without introducing additional parameters.