H.3.2.2. Computer vision
Mahdi Zarrin; Haniyeh Nikkhah
Abstract
Medical image analysis, crucial for disease diagnosis and treatment, often suffers from the challenge of class imbalance, where the area of normal tissue significantly outweighs that of abnormal regions. Furthermore, the varying class ratios across different images within a dataset complicate the application ...
Read More
Medical image analysis, crucial for disease diagnosis and treatment, often suffers from the challenge of class imbalance, where the area of normal tissue significantly outweighs that of abnormal regions. Furthermore, the varying class ratios across different images within a dataset complicate the application of uniform loss adjustments. To address these issues and advance automated segmentation, this study proposes a novel deep learning model integrating the strengths of YOLO Version 8's efficient feature extraction modules (SPPF and C2F) within a U-shaped architecture enhanced by a Receptive Field Enhancement (RFE) module. The RFE module, acting as an advanced skip connection, strategically fuses multi-scale features from corresponding and subsequent encoder layers processed through SPPF and C2F to enrich feature transfer and improve receptive field. To specifically tackle the class imbalance and the diversity of class distributions across images, we introduce a novel Adapt Exponential Loss function. This pixel-level loss dynamically adjusts class weights for each image based on its individual lesion-to-total-pixel ratio (k). We evaluated our proposed model and loss function on challenging skin lesion datasets: ISIC 2018, ISIC 2017, and PH2. Our method achieved significant segmentation performance with IoU scores of 86.47%, 85.67%, and 93.13%, and Dice scores of 91.63%, 90.19%, and 96.02% on ISIC 2018, ISIC 2017, and PH2, respectively, demonstrating its effectiveness in accurately delineating skin lesions despite class imbalance and varying lesion proportions. This work contributes a robust framework for medical image segmentation, facilitating more reliable diagnostic tools in dermatology.
