Search papers, labs, and topics across Lattice.
This paper introduces a gender-aware two-stage framework for lung disease classification in 3D CT scans to address class imbalance issues. The framework first predicts patient gender from CT scans and then routes the CT image to a gender-specific disease classifier. Results show improved recognition performance for minority disease categories like squamous cell carcinoma.
Accounting for gender-specific imaging characteristics in lung disease classification significantly boosts performance on minority disease categories often missed due to class imbalance.
Accurate classification of lung diseases from chest CT scans plays an important role in computer-aided diagnosis systems. However, medical imaging datasets often suffer from severe class imbalance, which may significantly degrade the performance of deep learning models, especially for minority disease categories. To address this issue, we propose a gender-aware two-stage lung disease classification framework. The proposed approach explicitly incorporates gender information into the disease recognition pipeline. In the first stage, a gender classifier is trained to predict the patient's gender from CT scans. In the second stage, the input CT image is routed to a corresponding gender-specific disease classifier to perform final disease prediction. This design enables the model to better capture gender-related imaging characteristics and alleviate the influence of imbalanced data distribution. Experimental results demonstrate that the proposed method improves the recognition performance for minority disease categories, particularly squamous cell carcinoma, while maintaining competitive performance on other classes.